/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators/matmul.h (507B)
#pragma once
#include
namespace torch {
namespace jit {
namespace tensorexpr {
Tensor computeMatmul(
const std::vector& inputs,
const std::vector& outputShape,
const c10::optional& outputType);
Tensor computeAddMM(
const std::vector& inputs,
const std::vector& outputShape,
const c10::optional& outputType);
} // namespace tensorexpr
} // namespace jit
} // namespace torch