Searched refs:GetSqueezeIdx (Results 1 – 3 of 3) sorted by relevance
47 std::vector<size_t> GetSqueezeIdx() const;
222 std::vector<size_t> Arrangement::GetSqueezeIdx() const { in GetSqueezeIdx() function in mindspore::parallel::Arrangement
397 std::vector<size_t> squeeze_list = tensor_shape_.GetSqueezeIdx(); in SqueezeShape()