Home
last modified time | relevance | path

Searched refs:new_outputs (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dgroup_convolution_creator.cc37 std::vector<lite::Tensor *> *new_outputs) { in FreeCurrentConv() argument
47 if (new_outputs != nullptr) { in FreeCurrentConv()
48 for (auto &out_tensor : *new_outputs) { in FreeCurrentConv()
188 std::vector<lite::Tensor *> *new_outputs, int group_id) { in GetSingleConvParam() argument
209 if (NewOutputTensor(new_outputs, output) != lite::RET_OK) { in GetSingleConvParam()
212 FreeCurrentConv(conv_param, new_inputs, new_outputs); in GetSingleConvParam()
Dgroup_convolution_creator.h56 std::vector<lite::Tensor *> *new_outputs, int group_id);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dadd_input_to_output.cc106 std::vector<AnfNodePtr> new_outputs; in Process() local
109 CreateMultipleOutputsOfAnfNode(func_graph, cnode, new_abstract_tuple->size(), &new_outputs); in Process()
110 if (new_outputs.size() != new_abstract_tuple->size()) { in Process()
114 return new_outputs[0]; in Process()
/third_party/mesa3d/src/compiler/nir/
Dnir_lower_io_to_temporaries.c42 struct exec_list new_outputs; member
97 emit_copies(&b, &state->new_outputs, &state->old_outputs); in emit_output_copies_impl()
103 emit_copies(&b, &state->old_outputs, &state->new_outputs); in emit_output_copies_impl()
111 emit_copies(&b, &state->new_outputs, &state->old_outputs); in emit_output_copies_impl()
349 exec_list_make_empty(&state.new_outputs); in nir_lower_io_to_temporaries()
356 exec_list_push_tail(&state.new_outputs, &output->node); in nir_lower_io_to_temporaries()
383 exec_list_append(&shader->variables, &state.new_outputs); in nir_lower_io_to_temporaries()
Dnir_lower_io_to_vector.c415 nir_variable *new_outputs[MAX_SLOTS][4] = {{0}}; in nir_lower_io_to_vector_impl() local
436 new_outputs, flat_outputs)) in nir_lower_io_to_vector_impl()
477 new_outputs[loc][old_frac]; in nir_lower_io_to_vector_impl()
530 nir_variable *new_var = new_outputs[loc][old_frac]; in nir_lower_io_to_vector_impl()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_inputs_for_reduce_scatter.cc48 std::vector<AnfNodePtr> new_outputs; in InsertSplitForInput() local
49 …reateMultipleOutputsOfAnfNode(func_graph, split, AnfAlgo::GetOutputTensorNum(split), &new_outputs); in InsertSplitForInput()
50 for (size_t j = 0; j < new_outputs.size(); j++) { in InsertSplitForInput()
51 split_outputs.push_back(new_outputs[j]); in InsertSplitForInput()
Dconcat_outputs_for_all_gather.cc151 std::vector<AnfNodePtr> new_outputs; in Process() local
165 (void)new_outputs.emplace_back(std::move(tuple_getitem)); in Process()
167 return InsertConcatForOutput(func_graph, node, output_info, new_outputs, rank_size); in Process()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dgroup_convolution_int8.cc63 std::vector<lite::Tensor *> new_outputs; in Init() local
64 … auto ret = group_conv_creator_->GetSingleConvParam(new_conv_param, &new_inputs, &new_outputs, i); in Init()
71 …CpuConvInt8KernelSelect(new_inputs, new_outputs, reinterpret_cast<OpParameter *>(new_conv_param), … in Init()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dgroup_convolution_fp32.cc76 std::vector<lite::Tensor *> new_outputs; in Init() local
77 … auto ret = group_conv_creator_->GetSingleConvParam(new_conv_param, &new_inputs, &new_outputs, i); in Init()
83 …nDelegateCPUKernel(reinterpret_cast<OpParameter *>(new_conv_param), new_inputs, new_outputs, ctx_); in Init()
/third_party/mindspore/tests/ut/cpp/session/
Dsession_basic_test.cc92 auto new_outputs = kernel_graph->outputs(); in TEST_F() local
93 EXPECT_EQ(new_outputs.size(), 1); in TEST_F()
94 EXPECT_EQ(AnfAlgo::GetCNodeName(new_outputs[0]), prim::kPrimMul->name()); in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dgroup_convolution_fp16.cc95 std::vector<lite::Tensor *> new_outputs; in Init() local
96 … auto ret = group_conv_creator_->GetSingleConvParam(new_conv_param, &new_inputs, &new_outputs, i); in Init()
102 reinterpret_cast<OpParameter *>(new_conv_param), new_inputs, new_outputs, ctx_)); in Init()
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dbatch_norm_grad_infer_fission_test.py68new_outputs = make_tuple(tuple_getitem(outputs, 0), tuple_getitem(outputs, 1), tuple_getitem(outpu…
69 output = tuple_getitem(new_outputs, 0)
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dconcat_outputs_for_all_gather.cc144 std::vector<AnfNodePtr> new_outputs; in Process() local
158 new_outputs.emplace_back(std::move(tuple_getitem)); in Process()
160 return InsertConcatForOutput(func_graph, node, output_info, new_outputs, rank_size); in Process()
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc202 auto new_outputs = kernel_graph->outputs(); in CreateGraphWithExecOrder() local
203 EXPECT_EQ(new_outputs.size(), 1); in CreateGraphWithExecOrder()
204 EXPECT_EQ(AnfAlgo::GetCNodeName(new_outputs[0]), prim::kPrimMul->name()); in CreateGraphWithExecOrder()
/third_party/mesa3d/src/freedreno/ir3/
Dir3_nir_lower_tess.c43 struct exec_list new_outputs; member
940 exec_list_make_empty(&state.new_outputs); in ir3_nir_lower_gs()
947 exec_list_push_tail(&state.new_outputs, &output->node); in ir3_nir_lower_gs()
989 foreach_two_lists (dest_node, &state.new_outputs, src_node, in ir3_nir_lower_gs()
999 exec_list_append(&shader->variables, &state.new_outputs); in ir3_nir_lower_gs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc539 std::vector<AnfNodePtr> new_outputs; in Process() local
540 auto lstm_input_grad = AddLSTMInputGradNode(func_graph, dynamic_rnn_grad_cnode, &new_outputs); in Process()
571 make_tuple_inputs.insert(make_tuple_inputs.end(), new_outputs.begin(), new_outputs.end()); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.cc347 auto new_outputs = outputs; in SetNewKernelInfo() local
351 new_outputs = real_outs; in SetNewKernelInfo()
354 for (size_t i = 0; i < new_outputs.size(); ++i) { in SetNewKernelInfo()
355 auto kernel_with_index = AnfAlgo::VisitKernel(new_outputs[i], 0); in SetNewKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc1217 const std::vector<AnfWithOutIndex> &new_outputs) { in UpdateGraphOutputMap() argument
1219 << ", the size of new outputs: " << new_outputs.size(); in UpdateGraphOutputMap()
1220 if (old_outputs.size() != new_outputs.size()) { in UpdateGraphOutputMap()
1226 auto new_output = new_outputs[i]; in UpdateGraphOutputMap()
Dkernel_graph.h250 const std::vector<AnfWithOutIndex> &new_outputs);