Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_fusion.h39 std::vector<AnfNodePtr> input_nodes; member
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc33 auto input_nodes = kernel_graph->inputs(); in LoadInputData() local
62 auto input_nodes = kernel_graph->inputs(); in CompileGraphImpl() local
Dgpu_inference_session.cc33 auto input_nodes = kernel_graph->inputs(); in LoadInputData() local
62 auto input_nodes = kernel_graph->inputs(); in CompileGraphImpl() local
Dcpu_session.cc169 auto &input_nodes = kernel_graph->inputs(); in LoadInputData() local
Dkernel_graph.h279 const std::vector<AnfNodePtr> &input_nodes() const { return input_nodes_; } in input_nodes() function
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc137 …ratorInfo::CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes, in CreateConcateNode()
164 …eratorInfo::CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes, in CreateReduceNode()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dunsorted_segment_op_info.cc288 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(segment_min, 1), in ComputeReplaceGraph() local
322 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(segment_max, 1), in ComputeReplaceGraph() local
Dbroadcast_to_info.cc166 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(new_broadcast_to, 1)}; in ComputeReplaceGraph() local
Dunique_info.cc151 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(sub, 1), std::make_pair(… in ComputeReplaceGraph() local
Duniform_candidate_sampler_info.cc233 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(new_sampler_op, 1)}; in ComputeReplaceGraph() local
Donehot_info.cc195 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(floor_div, 1), std::make… in ComputeReplaceGraph() local
Dslice_info.cc234 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(slice, 1)}; in ComputeReplaceGraph() local
Dmatmul_dds_info.cc250 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = { in ComputeReplaceGraph() local
Dgather_v2_p_info.cc763 std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(sub_node, 2), in ComputeReplaceGraph() local
801 …std::vector<std::pair<AnfNodePtr, int64_t>> input_nodes = {std::make_pair(sub, 2), std::make_pair(… in ComputeReplaceGraph() local
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel_util.cc38 std::vector<kernel::LiteKernel *> input_nodes; in SubgraphInputNodes() local
88 std::vector<kernel::LiteKernel *> input_nodes = SubgraphInputNodes(kernels); in SubgraphInputTensors() local
Dsub_graph_split.cc448 std::vector<uint32_t> input_nodes = tensors_.at(head_input_tensors.at(0)).out_nodes_; in OptimizeAfterFusion() local
537 std::vector<uint32_t> input_nodes = tensors_.at(head_input_tensors.at(0)).out_nodes_; in InsertNodeByMid() local
592 std::vector<uint32_t> input_nodes = tensor->out_nodes_; in InitMiddleSubgraph() local
860 std::vector<uint32_t> input_nodes = tensor->out_nodes_; in SubGraphSplitByOffLineParallel() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc207 const auto &input_nodes = graph->input_nodes(); in PrepareDataForDeviceTensorStore() local
238 const auto &input_nodes = graph->input_nodes(); in PrepareDataForHostTensorQueue() local
282 const auto &input_nodes = graph->input_nodes(); in PrepareDataForStepMode() local
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc289 std::vector<AnfNodePtr> input_nodes; in TEST_F() local
348 std::vector<AnfNodePtr> input_nodes; in TEST_F() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc224 std::vector<AnfNodePtr> input_nodes; in ConvertLayers() local
551 …elParser::ConvertBottom(const caffe::LayerParameter &layer, std::vector<AnfNodePtr> *input_nodes) { in ConvertBottom()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc285 auto &input_nodes = kernel_graph->inputs(); in CreateOutputTensors() local
309 auto &input_nodes = kernel_graph.inputs(); in BindInputTensorAddressPtr() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dfusion_tbe_json_creator.cc96 auto input_nodes = fusion_scope_info.input_nodes; in CheckInput() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc388 auto input_nodes = fusion_scope_info.input_nodes; in GetSpecDataInput() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dfunction_block.cc421 std::vector<AnfNodePtr> input_nodes; in Jump() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc665 CNodePtr CreateCNodeWithGraph(const std::vector<AnfNodePtr> &input_nodes, const BaseRef &graph) { in CreateCNodeWithGraph()
691 std::vector<AnfNodePtr> input_nodes; in HandleSexpVector() local
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc192 CNodePtr CreateCNodeWithGraph(const std::vector<AnfNodePtr> &input_nodes, const BaseRef &graph) { in CreateCNodeWithGraph()
222 std::vector<AnfNodePtr> input_nodes; in HandleSexpVector() local

12