Home
last modified time | relevance | path

Searched defs:input_nodes (Results 1 – 25 of 76) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dkernel_fusion.h41 std::vector<AnfNodePtr> input_nodes; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_inference_session.cc34 auto input_nodes = kernel_graph->inputs(); in LoadInputData() local
68 auto input_nodes = kernel_graph->inputs(); in CompileGraphImpl() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/c_api/
Dgraph_test.cc46 NodeHandle input_nodes[] = {x, y}; in TEST_F() local
Dattribute_test.cc59 NodeHandle input_nodes[] = {x, y}; in TEST_F() local
Dnode_test.cc47 NodeHandle input_nodes[] = {x, y}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dall_to_all_v_for_ge.cc143 …dePtrList CreateFlattenReshapeNodes(const FuncGraphPtr &graph, const AnfNodePtrList &input_nodes) { in CreateFlattenReshapeNodes()
157 CNodePtr CreateConcatNode(const FuncGraphPtr &graph, const AnfNodePtrList &input_nodes) { in CreateConcatNode()
248 AnfNodePtrList CreateReshapeNodes(const FuncGraphPtr &graph, const AnfNodePtrList &input_nodes, in CreateReshapeNodes()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc147 …ratorInfo::CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes, in CreateConcateNode()
176 …eratorInfo::CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes, in CreateReduceNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dunsorted_segment_op_info.cc271 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(segment_min, 1), in ComputeReplaceGraph() local
307 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(segment_max, 1), in ComputeReplaceGraph() local
Dscatter_math_ops_info.cc189 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(scatter_ops, 1), std::ma… in ComputeReplaceGraph() local
230 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(scatter_ops, 1), std::ma… in ComputeReplaceGraph() local
Dscatter_nd_ops_info.cc226 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(scatter_ops, 1), std::ma… in ComputeReplaceGraph() local
265 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(scatter_ops, 1), std::ma… in ComputeReplaceGraph() local
Dbroadcast_to_info.cc168 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(new_broadcast_to, 1)}; in ComputeReplaceGraph() local
Dunique_info.cc139 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(sub, 1), std::make_pair(… in ComputeReplaceGraph() local
Dconv3d_info.cc222 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(transpose_1, 1)}; in ComputeReplaceGraph() local
Donehot_info.cc205 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(floor_div, 1), std::make… in ComputeReplaceGraph() local
Duniform_candidate_sampler_info.cc218 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(new_sampler_op, 1)}; in ComputeReplaceGraph() local
Dmatmul_dds_info.cc274 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = { in ComputeReplaceGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dhelper.cc43 CNodePtr Helper::CreateCNodeWithGraph(const std::vector<AnfNodePtr> &input_nodes, const BaseRef &gr… in CreateCNodeWithGraph()
74 std::vector<AnfNodePtr> input_nodes; in HandleSexpVector() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/shard/
Dshard.cc87 static void GetInputNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *input_nodes) { in GetInputNodes()
224 std::vector<AnfNodePtr> input_nodes; in SetInputLayout() local
398 std::vector<AnfNodePtr> input_nodes; in SetStrategyForShard() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dload_mindir.cc77 AnfNodePtrList input_nodes{NewValueNode(new_prim)}; in ModifyOneCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dsuper_kernel_actor.cc183 auto &input_nodes = graph_->input_nodes(); in FetchInputNodePosition() local
696 auto &input_nodes = graph->input_nodes(); in CopyInputData() local
886 const auto &input_nodes = graph_->input_nodes(); in ParseInputIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dsub_graph_split.cc458 std::vector<uint32_t> input_nodes = tensors_.at(head_input_tensors.at(0)).out_nodes_; in OptimizeAfterFusion() local
547 std::vector<uint32_t> input_nodes = tensors_.at(head_input_tensors.at(0)).out_nodes_; in InsertNodeByMid() local
602 std::vector<uint32_t> input_nodes = tensor->out_nodes_; in InitMiddleSubgraph() local
872 std::vector<uint32_t> input_nodes = tensor->out_nodes_; in SubGraphSplitByOffLineParallel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/caffe/
Dcaffe_model_parser.cc177 std::vector<api::AnfNodePtr> input_nodes; in ConvertLayers() local
469 …rser::ConvertBottom(const caffe::LayerParameter &layer, std::vector<api::AnfNodePtr> *input_nodes)… in ConvertBottom()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runner.cc58 void UpdateInputTensorFromDevice(const std::vector<AnfNodePtr> &input_nodes, in UpdateInputTensorFromDevice()
131 void UpdateInputNodeDeviceAddress(const std::vector<AnfNodePtr> &input_nodes, in UpdateInputNodeDeviceAddress()
247 void CopyParameterDataToDevice(const std::vector<AnfNodePtr> &input_nodes, in CopyParameterDataToDevice()
791 const auto &input_nodes = graph->input_nodes(); in UpdateDeviceAddress() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc291 std::vector<AnfNodePtr> input_nodes; in TEST_F() local
352 std::vector<AnfNodePtr> input_nodes; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc236 std::vector<AnfNodePtr> input_nodes; in ConvertLayers() local
614 …elParser::ConvertBottom(const caffe::LayerParameter &layer, std::vector<AnfNodePtr> *input_nodes) { in ConvertBottom()

1234