Home
last modified time | relevance | path

Searched refs:real_inputs (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.cc44 std::vector<session::KernelWithIndex> real_inputs; in WeightCheck() local
45 AnfAlgo::GetRealInputs(node, &real_inputs); in WeightCheck()
48 if (index >= real_inputs.size()) { in WeightCheck()
50 << real_inputs.size(); in WeightCheck()
53 if (real_inputs[index].first->isa<Parameter>() && in WeightCheck()
54 !AnfAlgo::IsParameterWeight(real_inputs[index].first->cast<ParameterPtr>())) { in WeightCheck()
Dtrt_converter_context.cc191 std::vector<session::KernelWithIndex> real_inputs; in LoadLayerInput() local
192 AnfAlgo::GetRealInputs(node, &real_inputs); in LoadLayerInput()
193 for (auto item : real_inputs) { in LoadLayerInput()
/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_graph_param_cases.py46 real_inputs = inputs[:self.real_inputs_count]
48 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
52 real_inputs = inputs[:self.real_inputs_count]
54 return self.grad(self.network)(*real_inputs, sense_param_inputs)
Dtest_multi_grad.py44 real_inputs = inputs[:self.real_inputs_count]
46 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
50 real_inputs = inputs[:self.real_inputs_count]
52 return self.grad(self.network)(*real_inputs, sense_param_inputs)
/third_party/mindspore/tests/st/high_grad/
Dtest_highgrad_train.py46 real_inputs = inputs[:self.real_inputs_count]
49 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
50 return self.grad(self.network)(*real_inputs, sense_param_inputs)
/third_party/mindspore/tests/st/pynative/
Dtest_graph_param_transform.py44 real_inputs = inputs[:self.real_inputs_count]
46 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
50 real_inputs = inputs[:self.real_inputs_count]
52 return self.grad(self.network)(*real_inputs, sense_param_inputs)
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py41 real_inputs = inputs[:self.real_inputs_count]
43 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
46 real_inputs = inputs[:self.real_inputs_count]
48 return self.grad(self.network)(*real_inputs, sense_param_inputs)
Dtest_pynative_hook_grad.py82 real_inputs = inputs[:self.real_inputs_count]
84 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
87 real_inputs = inputs[:self.real_inputs_count]
89 return self.grad(self.network)(*real_inputs, sense_param_inputs)
/third_party/mindspore/mindspore/core/ir/
Danf.cc414 std::vector<AnfNodePtr> real_inputs; in GetVirtualNodeTargetFromInputs() local
419 std::back_inserter(real_inputs)); in GetVirtualNodeTargetFromInputs()
422 std::back_inserter(real_inputs)); in GetVirtualNodeTargetFromInputs()
426 …std::any_of(std::rbegin(real_inputs), std::rend(real_inputs), [&first_input_target](const AnfNodeP… in GetVirtualNodeTargetFromInputs()
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_gpu.py53 real_inputs = inputs[:self.real_inputs_count]
56 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
57 return self.grad(self.network)(*real_inputs, sense_param_inputs)
Dtest_auto_monad_mindtester.py49 real_inputs = inputs[:self.real_inputs_count]
52 return self.grad(self.network, self.params)(*real_inputs, sense_param_inputs)
53 return self.grad(self.network)(*real_inputs, sense_param_inputs)
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc816 const std::vector<AnfNodePtr> &real_inputs) { in ProcessNodeRetFunc() argument
845 for (auto &real_input : real_inputs) { in ProcessNodeRetFunc()
862 for (auto &input_node : real_inputs) { in ProcessNodeRetFunc()
891 std::vector<AnfNodePtr> real_inputs; in CreateCallSwitchLayerInputs() local
893 real_inputs.emplace_back(graph->GetBackendAnfByFrontAnf(cnode->input(idx))); in CreateCallSwitchLayerInputs()
920 ProcessNodeRetFunc(cnode, partial_kernel_graph.get(), real_inputs); in CreateCallSwitchLayerInputs()
923 new_partial_inputs.insert(new_partial_inputs.end(), real_inputs.begin(), real_inputs.end()); in CreateCallSwitchLayerInputs()
Dkernel_graph.cc76 std::vector<AnfNodePtr> real_inputs; in GetCallRealOutputs() local
82 std::copy(child_real_inputs.begin(), child_real_inputs.end(), std::back_inserter(real_inputs)); in GetCallRealOutputs()
84 return real_inputs; in GetCallRealOutputs()
Dsession_basic.h181 …NodeRetFunc(const CNodePtr &cnode, KernelGraph *graph, const std::vector<AnfNodePtr> &real_inputs);