Home
last modified time | relevance | path

Searched defs:return_shape (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcompare_and_bitpack.cc82 auto return_shape = out_shape; in CompareAndBitpackInferShape() local
Dnth_element.cc98 auto return_shape = out_shape; in NthElementInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/dynamic_obfuscation/
Ddynamic_obfuscation.cc737 ShapeVector return_shape = get_node_shape(last_node->cast<mindspore::CNodePtr>()); in CloneSubGraph() local
814 ShapeVector return_shape = get_node_shape(last_node->cast<mindspore::CNodePtr>()); in BuildFakeGraph() local