Searched refs:tensor_dim_apply3 (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorDimApply.h | 9 void tensor_dim_apply3(const Tensor& self, Tensor& values, Tensor& indices, int64_t dim, Function f… in tensor_dim_apply3() function
|
| D | ReduceOps.cpp | 792 …at::native::tensor_dim_apply3<scalar_t, int64_t>(self, values, indices, dim, cummax_cummin_helper<… in cummax_helper_cpu()
827 …at::native::tensor_dim_apply3<scalar_t, int64_t>(self, values, indices, dim, cummax_cummin_helper<… in cummin_helper_cpu()
|