Home
last modified time | relevance | path

Searched defs:inferSqueezeGeometry (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp3173 static inferSqueezeGeometry(const Tensor &tensor) { in inferSqueezeGeometry() function
3188 static inferSqueezeGeometry(const Tensor& tensor, int64_t dim) { in inferSqueezeGeometry() function
3202 static inferSqueezeGeometry(const Tensor &tensor, std::bitset<dim_bitset_size> dim_mask) { in inferSqueezeGeometry() function