Home
last modified time | relevance | path

Searched defs:sub_node (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/
Dgraph_compiler_test.cc64 auto sub_node = func_graph->NewCNode(sub_inputs); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dcentralization_mindir.cc100 auto sub_node = NewCNode(sub_input, func_graph); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dbatchnorm_transform.cc89 …auto sub_node = fg->NewCNode({NewValueNode(std::make_shared<Primitive>(kSubOpName)), mean, tuple_g… in CreateSubNode() local
110 const auto &sub_node = inputs[1]; in CreateMulNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dadjust_matmul_pass.cc158 auto sub_node = func_graph->NewCNode(sub_prim_c, inputs); in CreateSubCnode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dgraph_kernel_expander_lite.cc155 for (auto sub_node : sub_nodes) { in Run() local
/third_party/python/Python/
Dhamt.c734 PyHamtNode *sub_node = hamt_node_assoc( in hamt_node_bitmap_assoc() local
789 PyHamtNode *sub_node = hamt_node_new_bitmap_or_collision( in hamt_node_bitmap_assoc() local
976 PyHamtNode *sub_node = NULL; in hamt_node_bitmap_without() local
1773 PyHamtNode *sub_node = NULL; in hamt_node_array_without() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_info.cc566 …auto sub_node = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tenso… in InferReplaceGraph() local
600 …auto sub_node = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tenso… in InferReplaceGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_helper.cc350 for (auto sub_node : sub_nodes) { in IsBufferStitchNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dgraph_kernel_splitter.cc518 for (auto sub_node : area.nodes()) { in AreaExpand() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dflash_sp.cc233 auto sub_node = left_node->func_graph()->NewCNode(sub_inputs); in NewSubNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dnode.cc669 auto sub_node = mindspore::NewValueNode(res_sub_fg); in MSNewFuncCallNode() local