/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/FunctionOfAMatrixUtils.h (436B)
#pragma once
#include
#include
#include
#include
namespace at { namespace native {
using _compute_linear_combination_fn = void(*)(
TensorIterator& iter,
int64_t in_stride,
int64_t coeff_stride,
int64_t num_summations
);
DECLARE_DISPATCH(_compute_linear_combination_fn, _compute_linear_combination_stub);
}} // namespace at::native