Home
last modified time | relevance | path

Searched defs:first_input (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dreshape_shape_fusion.cc58 auto first_input = reshape->input(1); in Process() local
Dmul_reduce_fusion.cc302 auto first_input = cnode->input(1); in CheckBasicCond() local
337 auto first_input = cnode->input(1); in CheckAxisCond() local
375 auto first_input = cnode->input(1); in CheckShapeCond() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dremove_public_primitive.cc46 auto first_input = cnode->input(0); in Run() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/
Dsession_basic_test.cc81 auto first_input = inputs[0]->cast<ParameterPtr>(); in TEST_F() local
Dkernel_graph_test.cc123 auto first_input = kernel_graph->inputs()[0]; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dmerge_addn.h82 const auto &first_input = inputs.at(1); in Visit() local
Dupdatestate_eliminate.cc274 auto &first_input = make_tuple->input(kInputIndex); in EliminateUpdateStateMakeTupleWithUselessNode() local
780 const auto &first_input = cnode->input(0); in operator ()() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dredundant_op_remove_pass.cc47 auto first_input = cnode->input(kInputIndexOne); in ReplaceUpdateStateWithMonad() local
82 auto first_input = cnode->input(1); in ProcessInputIsMonad() local
Dscalar_op_pass.cc586 auto first_input = node->cast<CNodePtr>()->input(kIndexOne); in RunArithmeticCheckPass() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc194 auto first_input = inputs[0]->cast<ParameterPtr>(); in CreateGraphWithExecOrder() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/hcom/
Dinsert_depend_for_all_gather_ge.cc67 CNodePtr CreateDepend(const AnfNodePtr &first_input, const AnfNodePtr &second_input, const FuncGrap… in CreateDepend()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dtsa_atomic_add_to_first_tensor.cc68 auto first_input = tsa_node->input(1)->cast<ParameterPtr>(); in FindTsaFirstRealInputInGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dpy_interpret_to_execute.cc236 auto &first_input = cnode->input(input_index_one); in Transform() local
Dauto_monad_eliminate.cc73 const auto &first_input = cnode->input(0); in IsSpecialNode() local
/third_party/mesa3d/src/gallium/auxiliary/vl/
Dvl_idct.c432 unsigned first_input, struct ureg_dst fragment) in vl_idct_stage2_frag_shader()
Dvl_mpeg12_decoder.c1087 unsigned first_input, in mc_frag_shader_callback()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Drecompute_helper.cc585 auto first_input = new_inputs[1]; in NewRecomputedNode() local
/third_party/skia/third_party/externals/spirv-tools/source/opt/
Dfolding_rules.cpp1726 Instruction* first_input = in VectorShuffleFeedingExtract() local
/third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfolding_rules.cpp1726 Instruction* first_input = in VectorShuffleFeedingExtract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc953 auto first_input = new_node_inputs.begin(); in CreateNodeWithInterProcessEdgeOnPServer() local
/third_party/spirv-tools/source/opt/
Dfolding_rules.cpp1959 Instruction* first_input = in VectorShuffleFeedingExtract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc1291 auto first_input = new_cnode->input(kFirstDataInputIndex); in CreateNewCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc1585 auto first_input = GetRealInput(cnode->input(1)); in MatchAndMarkCNode() local
1590 auto first_input = GetRealInput(cnode->input(1)); // Unpack Depend in MatchAndMarkCNode() local