Searched defs:inferSqueezeGeometry (Results 1 – 1 of 1) sorted by relevance
3173 static inferSqueezeGeometry(const Tensor &tensor) { in inferSqueezeGeometry() function3188 static inferSqueezeGeometry(const Tensor& tensor, int64_t dim) { in inferSqueezeGeometry() function3202 static inferSqueezeGeometry(const Tensor &tensor, std::bitset<dim_bitset_size> dim_mask) { in inferSqueezeGeometry() function