Searched refs:SqueezeMapByIdxList (Results 1 – 3 of 3) sorted by relevance
44 Map SqueezeMapByIdxList(std::vector<size_t> idx_list) const;
153 Map Map::SqueezeMapByIdxList(std::vector<size_t> idx_list) const { in SqueezeMapByIdxList() function in mindspore::parallel::Map
403 out_map = tensor_map_.SqueezeMapByIdxList(squeeze_list); in SqueezeShape()