Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReduceOpsUtils.h289 inline std::vector<int64_t> get_zero_numel_tensor_size( in get_zero_numel_tensor_size() function
319 auto sizes = get_zero_numel_tensor_size(self, dim, keepdim, fn_name); in zero_numel_tensor_resize()
H A DTensorCompare.cpp606 auto sizes = get_zero_numel_tensor_size(self, dim, keepdim, "mode()"); in mode_out()