Home
last modified time | relevance | path

Searched refs:spmm_reduce_backward_input_fn (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DSpmmReduceKernel.h11 using spmm_reduce_backward_input_fn = void(*)(const Tensor&, const Tensor&, const Tensor&, const Te… variable
17 DECLARE_DISPATCH(spmm_reduce_backward_input_fn, spmm_reduce_backward_input_stub);