Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dshape_unify_mindir.cc63 auto shape_cnode = node->cast<CNodePtr>(); in Process() local
120 auto shape_cnode = anf_node->cast<CNodePtr>(); in CreateTensorShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dtensor_dot_fusion.cc166 auto shape_cnode = gather_1_cnode->input(1)->cast<CNodePtr>(); in Process() local
Dadjust_matmul_pass.cc93 auto shape_cnode = func_graph->NewCNode(shape_prim_c, inputs); in CreateShapeCNode() local
Dadjust_resize_dims_pass.cc162 auto shape_cnode = func_graph->NewCNode(shape_prim_c, inputs); in CreateShapeCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dshape_reshape_fusion.cc266 auto shape_cnode = shape_node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dscalar_op_pass.cc179 auto shape_cnode = anf_node->cast<CNodePtr>(); in GenerateTensorShape() local
412 auto shape_cnode = anf_node->cast<CNodePtr>(); in ReplaceShapeTupleGet() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_utils.cc53 auto shape_cnode = func_graph->NewCNode(shape_node_inputs); in CreateShape() local
1006 auto shape_cnode = shape_of_reshape->cast<CNodePtr>(); // MakeTuple in UpdatePartialShape() local
1115 Status UpdateReshapeShapeValue(const CNodePtr &reshape_cnode, const CNodePtr &shape_cnode, const Sh… in UpdateReshapeShapeValue()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_redistribution.cc670 auto shape_cnode = CreateShape(shape_root, func_graph, instance_name + "_get_shape"); in CreateAssembledDynamicMapping() local