/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
D | graph_partitioner.cc | 44 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()
|
D | trt_converter_context.cc | 191 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/ |
D | test_graph_param_cases.py | 46 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)
|
D | test_multi_grad.py | 44 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/ |
D | test_highgrad_train.py | 46 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/ |
D | test_graph_param_transform.py | 44 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)
|
D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 41 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)
|
D | test_pynative_hook_grad.py | 82 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/ |
D | anf.cc | 414 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/ |
D | test_auto_monad_gpu.py | 53 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)
|
D | test_auto_monad_mindtester.py | 49 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/ |
D | session_basic.cc | 816 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()
|
D | kernel_graph.cc | 76 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()
|
D | session_basic.h | 181 …NodeRetFunc(const CNodePtr &cnode, KernelGraph *graph, const std::vector<AnfNodePtr> &real_inputs);
|