Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.cc397 std::vector<size_t> squeeze_list = tensor_shape_.GetSqueezeIdx(); in SqueezeShape() local
398 if (!tensor_map_.CheckNoneByIdxList(squeeze_list)) { in SqueezeShape()
403 out_map = tensor_map_.SqueezeMapByIdxList(squeeze_list); in SqueezeShape()