Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dshape_unify_mindir.cc67 auto tensor_shape_node = CreateTensorShape(graph, node); in Process() local
124 CNodePtr tensor_shape_node = func_graph->NewCNode(inputs); in CreateTensorShape() local
Davg_pool_grad_for_ge.cc83 CNodePtr tensor_shape_node = func_graph->NewCNode(inputs); in CreateTensorShapeNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dscalar_op_pass.cc182 CNodePtr tensor_shape_node = func_graph->NewCNode(inputs); in GenerateTensorShape() local
415 auto tensor_shape_node = GenerateTensorShape(func_graph, anf_node); in ReplaceShapeTupleGet() local