Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dreduce_scatter.cc82 auto abstract_shape = input_args[kIndex0]->GetShape(); in InferShape() local
Dcollective_scatter.cc72 auto abstract_shape = input_args[kIndex0]->GetShape(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dnode_infershape.cc585 auto abstract_shape = std::make_shared<abstract::Shape>(shape); in SetCNodeAbstract() local
603 auto abstract_shape = std::make_shared<abstract::Shape>(shape); in SetCNodeAbstract() local
Dcore_infershape_pass.cc280 auto abstract_shape = std::make_shared<abstract::Shape>(shape_vec); in SetSubGraphInput() local
Dinfershape_pass.cc353 auto abstract_shape = std::make_shared<abstract::Shape>(shape_vec); in SetSubGraphInput() local
Ddecrease_transpose_algo.cc633 auto abstract_shape = std::make_shared<abstract::Shape>(shape_vec); in SetSubGraphInput() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dtensor.cc520 std::vector<abstract::AbstractBasePtr> abstract_shape; in ToAbstract() local
598 std::vector<abstract::AbstractBasePtr> abstract_shape; in ToAbstract() local
630 std::vector<abstract::AbstractBasePtr> abstract_shape; in ToAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dpynative_utils.cc2113 std::vector<abstract::AbstractBasePtr> abstract_shape; in GetSparseTensorShapeNode() local