Searched refs:max_pool3d_backward_fn (Results 1 – 1 of 1) sorted by relevance
44 using max_pool3d_backward_fn = void(*)(Tensor& grad_input, const Tensor& grad_output, const Tensor&… variable47 DECLARE_DISPATCH(max_pool3d_backward_fn, max_pool3d_backward_kernel);