text: #pragma once
  
  #include <torch/csrc/jit/tensorexpr/kernel.h>
  
  namespace torch {
  namespace jit {
  namespace tensorexpr {
  
  Tensor* computeMatmul(
      const std::vector<ArgValue>& inputs,
      const std::vector<ExprHandle>& outputShape,
      const c10::optional<ScalarType>& outputType);
  Tensor* computeAddMM(
      const std::vector<ArgValue>& inputs,
      const std::vector<ExprHandle>& outputShape,
      const c10::optional<ScalarType>& outputType);
  
  } // namespace tensorexpr
  } // namespace jit
  } // namespace torch
  