File size: 389 Bytes
7e50900 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
#pragma once
#include <ATen/native/DispatchStub.h>
#include <cstdint>
namespace at {
struct TensorIterator;
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
|