Home
last modified time | relevance | path

Searched refs:inputs_ (Results 1 – 25 of 115) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/ps/
Doptimizer_info.cc31 const std::vector<AddressPtr> &OptimizerInfo::inputs() const { return inputs_; } in inputs()
60 if (ps_send_index >= lens.size() || origin_index >= inputs_.size()) { in UpdateOptimInputValue()
67 AddressPtr optim_input = inputs_[origin_index]; in UpdateOptimInputValue()
273 inputs_.push_back(weight); in MomentumOptimInfo()
274 inputs_.push_back(accumulate); in MomentumOptimInfo()
275 inputs_.push_back(learning_rate); in MomentumOptimInfo()
276 inputs_.push_back(gradient); in MomentumOptimInfo()
277 inputs_.push_back(momentum); in MomentumOptimInfo()
288 EXC_IF_VEC_IDX_OOB(inputs_, origin_grad_index); in gradient()
289 MS_EXCEPTION_IF_NULL(inputs_[origin_grad_index]); in gradient()
[all …]
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dfullconnection_fp32_tests.cc35 int FcTestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in FcTestInit1() argument
42 inputs_->push_back(in_t); in FcTestInit1()
51 inputs_->push_back(weight_t); in FcTestInit1()
57 inputs_->push_back(bias_t); in FcTestInit1()
73 KernelInferShape(*inputs_, *outputs_, reinterpret_cast<OpParameter *>(matmal_param)); in FcTestInit1()
78 std::vector<lite::Tensor *> inputs_; in TEST_F() local
82 int total_size = FcTestInit1(&inputs_, &outputs_, matmul_param, &correct); in TEST_F()
87 …el::FullconnectionCPUKernel(reinterpret_cast<OpParameter *>(matmul_param), inputs_, outputs_, ctx); in TEST_F()
96 for (unsigned int i = 0; i < inputs_.size(); i++) { in TEST_F()
97 delete inputs_[i]; in TEST_F()
[all …]
Dmatmul_fp32_tests.cc71 int MMTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, float *… in MMTestInit() argument
76 inputs_->push_back(in_t); in MMTestInit()
81 inputs_->push_back(weight_t); in MMTestInit()
90 int MMTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, float … in MMTestInit2() argument
96 inputs_->push_back(in_t); in MMTestInit2()
101 inputs_->push_back(weight_t); in MMTestInit2()
106 inputs_->push_back(bias_t); in MMTestInit2()
116 std::vector<lite::Tensor *> inputs_; in TEST_F() local
131 int total_size = MMTestInit(&inputs_, &outputs_, a, b, a_shape, b_shape, c_shape); in TEST_F()
135 …auto mm = new kernel::MatmulCPUKernel(reinterpret_cast<OpParameter *>(matmul_param), inputs_, outp… in TEST_F()
[all …]
Dpower_fp32_tests.cc28 int PowerTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, floa… in PowerTestInit() argument
34 inputs_->push_back(in_t); in PowerTestInit()
39 inputs_->push_back(weight_t); in PowerTestInit()
48 int PowerTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, flo… in PowerTestInit2() argument
53 inputs_->push_back(in_t); in PowerTestInit2()
63 std::vector<lite::Tensor *> inputs_; in TEST_F() local
73 int total_size = PowerTestInit(&inputs_, &outputs_, a, b, a_shape, b_shape, c_shape); in TEST_F()
77 …auto *op = new kernel::PowerCPUKernel(reinterpret_cast<OpParameter *>(param), inputs_, outputs_, c… in TEST_F()
84 for (auto t : inputs_) delete t; in TEST_F()
89 std::vector<lite::Tensor *> inputs_; in TEST_F() local
[all …]
Dembedding_lookup_fp32_test.cc32 void ElTestInit(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, in ElTestInit() argument
38 inputs_->push_back(in_t_first); in ElTestInit()
44 inputs_->push_back(in_t_second); in ElTestInit()
50 inputs_->push_back(ids_t); in ElTestInit()
60 std::vector<Tensor *> inputs_; in TEST_F() local
63 ElTestInit(&inputs_, &outputs_, embedding_lookup_param_); in TEST_F()
69 reinterpret_cast<OpParameter *>(embedding_lookup_param_), inputs_, outputs_, ctx); in TEST_F()
84 for (unsigned int i = 0; i < inputs_.size(); i++) { in TEST_F()
85 delete inputs_[i]; in TEST_F()
Ddeconvolution_fp32_tests.cc323 int DeConvTestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in DeConvTestInit1() argument
339 inputs_->push_back(in_t); in DeConvTestInit1()
362 inputs_->push_back(weight_t); in DeConvTestInit1()
368 inputs_->push_back(bias_t); in DeConvTestInit1()
476 std::vector<lite::Tensor *> inputs_; in TEST_F() local
484 int total_size = DeConvTestInit1(&inputs_, &outputs_, deconv_param, &correct); in TEST_F()
486 …new kernel::DeConvolutionCPUKernel(reinterpret_cast<OpParameter *>(deconv_param), inputs_, outputs… in TEST_F()
494 for (auto t : inputs_) delete t; in TEST_F()
499 int DeConvTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in DeConvTestInit2() argument
507 inputs_->push_back(in_t); in DeConvTestInit2()
[all …]
Dskip_gram_fp32.cc34 void SkipGramTestInit(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, in SkipGramTestInit() argument
41 inputs_->push_back(in_t_first); in SkipGramTestInit()
54 std::vector<Tensor *> inputs_; in TEST_F() local
58 SkipGramTestInit(&inputs_, &outputs_, skip_gram_param_); in TEST_F()
64 …new kernel::SkipGramCPUKernel(reinterpret_cast<OpParameter *>(skip_gram_param_), inputs_, outputs_… in TEST_F()
76 for (unsigned int i = 0; i < inputs_.size(); i++) { in TEST_F()
77 delete inputs_[i]; in TEST_F()
Droi_pooling_fp32_tests.cc28 int ROIPoolingTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_,… in ROIPoolingTestInit() argument
34 inputs_->push_back(in_t); in ROIPoolingTestInit()
39 inputs_->push_back(roi_t); in ROIPoolingTestInit()
49 std::vector<lite::Tensor *> inputs_; in TEST_F() local
61 int total_size = ROIPoolingTestInit(&inputs_, &outputs_, a, b, a_shape, b_shape, c_shape); in TEST_F()
65 …auto *op = new kernel::ROIPoolingCPUKernel(reinterpret_cast<OpParameter *>(param), inputs_, output… in TEST_F()
75 for (auto t : inputs_) delete t; in TEST_F()
Dconstant_of_shape_fp32_test.cc28 int ConstantOfShapeTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outp… in ConstantOfShapeTestInit() argument
33 inputs_->push_back(in_t); in ConstantOfShapeTestInit()
47 std::vector<lite::Tensor *> inputs_; in TEST_F() local
55 int total_size = ConstantOfShapeTestInit(&inputs_, &outputs_, a, a_shape); in TEST_F()
60 …new kernel::ConstantOfShapeCPUKernel(reinterpret_cast<OpParameter *>(param), inputs_, outputs_, ct… in TEST_F()
70 for (auto t : inputs_) delete t; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dnode.cc39 new_input->AddUser(this, inputs_.size()); in AddInput()
40 (void)inputs_.emplace_back(new_input); in AddInput()
45 if (i >= inputs_.size()) { in SetInput()
46 MS_LOG(EXCEPTION) << "The index " << i << " is out of the inputs range " << inputs_.size(); in SetInput()
48 auto &old_input = inputs_[i]; in SetInput()
51 inputs_[i] = new_input; in SetInput()
55 if (!inputs_.empty()) { in SetInputs()
57 for (size_t i = 0; i < inputs_.size(); i++) { in SetInputs()
58 inputs_[i]->RemoveUser(this, i); in SetInputs()
60 inputs_.clear(); in SetInputs()
[all …]
/third_party/mindspore/tests/ut/cpp/kernel/cpu/
Dadam_delta_cpu_kernel_test.cc36 inputs_.clear(); in SetUp()
49 inputs_.push_back(CreateKernelAddress(m_.data(), elem_num_)); in CreateAddress()
50 inputs_.push_back(CreateKernelAddress(v_.data(), elem_num_)); in CreateAddress()
51 inputs_.push_back(CreateKernelAddress(&beta1_power_, 1)); in CreateAddress()
52 inputs_.push_back(CreateKernelAddress(&beta2_power_, 1)); in CreateAddress()
53 inputs_.push_back(CreateKernelAddress(&lr_, 1)); in CreateAddress()
54 inputs_.push_back(CreateKernelAddress(&beta1_, 1)); in CreateAddress()
55 inputs_.push_back(CreateKernelAddress(&beta2_, 1)); in CreateAddress()
56 inputs_.push_back(CreateKernelAddress(&epsilon_, 1)); in CreateAddress()
57 inputs_.push_back(CreateKernelAddress(grad_.data(), elem_num_)); in CreateAddress()
[all …]
Dsparse_apply_lazy_adam_cpu_kernel_test.cc36 inputs_.clear(); in SetUp()
48 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
49 inputs_.push_back(CreateKernelAddress(m_.data())); in CreateInputAddress()
50 inputs_.push_back(CreateKernelAddress(v_.data())); in CreateInputAddress()
51 inputs_.push_back(CreateKernelAddress(&beta1_power_)); in CreateInputAddress()
52 inputs_.push_back(CreateKernelAddress(&beta2_power_)); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(&lr_)); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(&beta1_)); in CreateInputAddress()
55 inputs_.push_back(CreateKernelAddress(&beta2_)); in CreateInputAddress()
56 inputs_.push_back(CreateKernelAddress(&epsilon_)); in CreateInputAddress()
[all …]
Dsparse_apply_adam_cpu_kernel_test.cc36 inputs_.clear(); in SetUp()
48 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
49 inputs_.push_back(CreateKernelAddress(m_.data())); in CreateInputAddress()
50 inputs_.push_back(CreateKernelAddress(v_.data())); in CreateInputAddress()
51 inputs_.push_back(CreateKernelAddress(&beta1_power_)); in CreateInputAddress()
52 inputs_.push_back(CreateKernelAddress(&beta2_power_)); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(&lr_)); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(&beta1_)); in CreateInputAddress()
55 inputs_.push_back(CreateKernelAddress(&beta2_)); in CreateInputAddress()
56 inputs_.push_back(CreateKernelAddress(&epsilon_)); in CreateInputAddress()
[all …]
Dsparse_apply_proximal_adagrad_cpu_kernel_test.cc36 inputs_.clear(); in SetUp()
48 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
49 inputs_.push_back(CreateKernelAddress(accum_.data())); in CreateInputAddress()
50 inputs_.push_back(CreateKernelAddress(&lr_)); in CreateInputAddress()
51 inputs_.push_back(CreateKernelAddress(&l1_)); in CreateInputAddress()
52 inputs_.push_back(CreateKernelAddress(&l2_)); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(indices.data())); in CreateInputAddress()
68 std::vector<AddressPtr> inputs_; member in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest
95 sparse_proximal_adagrad_->Launch(inputs_, workspace_, outputs_); in TEST_F()
[all …]
Dsparse_apply_ftrl_cpu_kernel_test.cc40 inputs_.clear(); in SetUp()
52 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(accum_.data())); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(linear_.data())); in CreateInputAddress()
55 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
56 inputs_.push_back(CreateKernelAddress(indices.data())); in CreateInputAddress()
71 std::vector<AddressPtr> inputs_; member in mindspore::kernel::SparseApplyFtrlCpuKernelTest
96 sparse_ftrl_->Launch(inputs_, workspace_, outputs_); in TEST_F()
123 sparse_ftrl_->Launch(inputs_, workspace_, outputs_); in TEST_F()
154 sparse_ftrl_->Launch(inputs_, workspace_, outputs_); in TEST_F()
Dunique_with_pad_cpu_kernel_test.cc34 inputs_.clear(); in SetUp()
46 inputs_.push_back(CreateKernelAddress(x_.data())); in CreateAddress()
47 inputs_.push_back(CreateKernelAddress(&pad_dim_)); in CreateAddress()
60 std::vector<AddressPtr> inputs_; member in mindspore::kernel::UniqueWithPadCpuKernelTest
73 unique_with_pad_->Launch(inputs_, workspace_, outputs_); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstrategy.h41 : stage_(stage), inputs_(std::move(inputs)), internal_size_(0), internal_stragies_() {} in Strategy()
44 inputs_ = another_stra.GetInputDim(); in Strategy()
54 size_t GetInputNumber() const { return inputs_.size(); } in GetInputNumber()
55 Strategys GetInputDim() const { return inputs_; } in GetInputDim()
58 if (inputs_.size() == 1) { in ExpandInputDimFromOneToTwo()
59 inputs_.push_back(inputs_[0]); in ExpandInputDimFromOneToTwo()
62 void ResetInputs(const Strategys &input) { inputs_ = input; } in ResetInputs()
71 if ((stage_ != another_stra->GetInputStage()) || (inputs_ != another_stra->GetInputDim())) { in IsEqual()
87 Strategys inputs_; variable
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Daccumulaten_eliminate.h39 if (inputs_.empty() || node->func_graph() == nullptr) { in operator()
44 if (inputs_.size() == 2) { in operator()
45 return inputs_[1]; in operator()
49 if (inputs_.size() == 1 && args_.size() > 0) { in operator()
60 auto make_tuple = fg->NewCNode(inputs_); in operator()
73 inputs_.push_back(NewValueNode(prim::kPrimMakeTuple)); in Visit()
76 inputs_.push_back(x); in Visit()
85 inputs_.clear(); in Reset()
90 std::vector<AnfNodePtr> inputs_{}, args_{};
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dpad_int8_tests.cc34 int PadInt8TestInit1(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, PadParameter … in PadInt8TestInit1() argument
43 inputs_->push_back(in_t); in PadInt8TestInit1()
64 std::vector<lite::Tensor *> inputs_; in TEST_F() local
70 int total_size = PadInt8TestInit1(&inputs_, &outputs_, pad_param, &correct); in TEST_F()
72 … new kernel::PadInt8CPUKernel(reinterpret_cast<OpParameter *>(pad_param), inputs_, outputs_, ctx); in TEST_F()
80 for (auto t : inputs_) delete t; in TEST_F()
85 int PadInt8TestInit2(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, PadParameter … in PadInt8TestInit2() argument
94 inputs_->push_back(in_t); in PadInt8TestInit2()
117 std::vector<lite::Tensor *> inputs_; in TEST_F() local
123 int total_size = PadInt8TestInit2(&inputs_, &outputs_, pad_param, &correct); in TEST_F()
[all …]
Dconv_1x1_int8_tests.cc72 int Conv1x1Int8TestInit1_perchannel(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor … in Conv1x1Int8TestInit1_perchannel() argument
82 inputs_->push_back(in_t); in Conv1x1Int8TestInit1_perchannel()
97 inputs_->push_back(weight_t); in Conv1x1Int8TestInit1_perchannel()
119 std::vector<lite::Tensor *> inputs_; in TEST_F() local
126 int total_size = Conv1x1Int8TestInit1_perchannel(&inputs_, &outputs_, conv_param, &correct); in TEST_F()
128 …new kernel::Convolution1x1Int8CPUKernel(reinterpret_cast<OpParameter *>(conv_param), inputs_, outp… in TEST_F()
136 for (auto t : inputs_) delete t; in TEST_F()
141 int Conv1x1Int8TestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs… in Conv1x1Int8TestInit1() argument
153 inputs_->push_back(in_t); in Conv1x1Int8TestInit1()
164 inputs_->push_back(weight_t); in Conv1x1Int8TestInit1()
[all …]
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/src/
Dsession.cc26 inputs_.resize(1); in CompileGraph()
33 inputs_[0] = new (std::nothrow) MTensor(String("graph_input-0"), kNumberTypeInt8, in_shape_0); in CompileGraph()
34 MS_ERROR_IF_NULL(inputs_[0]); in CompileGraph()
47 const void *inputs_data[inputs_.size()]; in RunGraph()
48 for (size_t i = 0; i < inputs_.size(); ++i) { in RunGraph()
49 inputs_data[i] = inputs_[i]->MutableData(); in RunGraph()
51 SetInputs(inputs_data, inputs_.size()); in RunGraph()
70 for (auto &input : inputs_) { in ~LiteSession()
101 for (const auto &input : inputs_) { in GetInputs()
/third_party/mindspore/mindspore/lite/micro/example/mnist_x86/src/
Dsession.cc26 inputs_.resize(1); in CompileGraph()
33 inputs_[0] = new (std::nothrow) MTensor(String("graph_input-0"), kNumberTypeInt8, in_shape_0); in CompileGraph()
34 MS_ERROR_IF_NULL(inputs_[0]); in CompileGraph()
48 const void *inputs_data[inputs_.size()]; in RunGraph()
49 for (size_t i = 0; i < inputs_.size(); ++i) { in RunGraph()
50 inputs_data[i] = inputs_[i]->MutableData(); in RunGraph()
52 SetInputs(inputs_data, inputs_.size()); in RunGraph()
71 for (auto &input : inputs_) { in ~LiteSession()
102 for (const auto &input : inputs_) { in GetInputs()
/third_party/ninja/src/
Dgraphviz.cc51 if (edge->inputs_.size() == 1 && edge->outputs_.size() == 1) { in AddTarget()
56 edge->inputs_[0], edge->outputs_[0], edge->rule_->name().c_str()); in AddTarget()
64 for (vector<Node*>::iterator in = edge->inputs_.begin(); in AddTarget()
65 in != edge->inputs_.end(); ++in) { in AddTarget()
67 if (edge->is_order_only(in - edge->inputs_.begin())) in AddTarget()
73 for (vector<Node*>::iterator in = edge->inputs_.begin(); in AddTarget()
74 in != edge->inputs_.end(); ++in) { in AddTarget()
Dgraph_test.cc406 EXPECT_EQ(1, edge->inputs_.size()); in TEST_F()
407 EXPECT_EQ("b", edge->inputs_[0]->path()); in TEST_F()
431 EXPECT_EQ(1, edge->inputs_.size()); in TEST_F()
432 EXPECT_EQ("c", edge->inputs_[0]->path()); in TEST_F()
458 EXPECT_EQ(1, edge->inputs_.size()); in TEST_F()
459 EXPECT_EQ("c", edge->inputs_[0]->path()); in TEST_F()
504 ASSERT_EQ(2u, edge->inputs_.size()); in TEST_F()
505 EXPECT_EQ("in", edge->inputs_[0]->path()); in TEST_F()
506 EXPECT_EQ("dd", edge->inputs_[1]->path()); in TEST_F()
638 ASSERT_EQ(3u, edge1->inputs_.size()); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc139 if (inputs_.empty()) { in Predict()
140 inputs_ = inputs; in Predict()
142 if (inputs.size() != inputs_.size()) { in Predict()
146 for (size_t i = 0; i < inputs_.size(); ++i) { in Predict()
147 …auto input_tensor = MSTensor::CreateTensor(inputs_[i].Name(), inputs_[i].DataType(), inputs_[i].Sh… in Predict()
149 inputs_[i] = (*input_tensor); in Predict()
160 if (inputs_.empty()) { in SetInputs()
175 inputs_.emplace_back(*input_tensor); in SetInputs()
237 return inputs_; in GetInputs()

12345