Home
last modified time | relevance | path

Searched refs:outputs_ (Results 1 – 25 of 100) sorted by relevance

1234

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dmatmul_fp32_tests.cc71 int MMTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, float *… in MMTestInit() argument
85 outputs_->push_back(out_t); in MMTestInit()
90 int MMTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, float … in MMTestInit2() argument
110 outputs_->push_back(out_t); in MMTestInit2()
117 std::vector<lite::Tensor *> outputs_; in TEST_F() local
131 int total_size = MMTestInit(&inputs_, &outputs_, a, b, a_shape, b_shape, c_shape); in TEST_F()
135 …new kernel::MatmulCPUKernel(reinterpret_cast<OpParameter *>(matmul_param), inputs_, outputs_, ctx); in TEST_F()
140 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<float *>(outputs_[0]->MutableData()), correct, tot… in TEST_F()
144 for (auto t : outputs_) delete t; in TEST_F()
149 std::vector<lite::Tensor *> outputs_; in TEST_F() local
[all …]
Dfullconnection_fp32_tests.cc35 int FcTestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in FcTestInit1() argument
61 outputs_->push_back(out_t); in FcTestInit1()
73 KernelInferShape(*inputs_, *outputs_, reinterpret_cast<OpParameter *>(matmal_param)); in FcTestInit1()
79 std::vector<lite::Tensor *> outputs_; 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()
93 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<float *>(outputs_[0]->MutableData()), correct, tot… in TEST_F()
99 for (unsigned int i = 0; i < outputs_.size(); i++) { in TEST_F()
100 delete outputs_[i]; in TEST_F()
104 int FcTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in FcTestInit2() argument
[all …]
Ddetection_post_process_test.cc30 …ionPostProcessTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in DetectionPostProcessTestInit() argument
100 outputs_->push_back(output_boxes); in DetectionPostProcessTestInit()
101 outputs_->push_back(output_classes); in DetectionPostProcessTestInit()
102 outputs_->push_back(output_scores); in DetectionPostProcessTestInit()
103 outputs_->push_back(output_num_det); in DetectionPostProcessTestInit()
121 std::vector<lite::Tensor *> outputs_; in TEST_F() local
123 DetectionPostProcessTestInit(&inputs_, &outputs_, param); in TEST_F()
128 …nel::DetectionPostProcessCPUKernel(reinterpret_cast<OpParameter *>(param), inputs_, outputs_, ctx); in TEST_F()
132 auto *output_boxes = reinterpret_cast<float *>(outputs_[0]->MutableData()); in TEST_F()
137 ASSERT_EQ(0, CompareOutputData(output_boxes, correct_boxes, outputs_[0]->ElementsNum(), 0.0001)); in TEST_F()
[all …]
Dpower_fp32_tests.cc28 int PowerTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, floa… in PowerTestInit() argument
43 outputs_->push_back(out_t); in PowerTestInit()
48 int PowerTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, flo… in PowerTestInit2() argument
57 outputs_->push_back(out_t); in PowerTestInit2()
64 std::vector<lite::Tensor *> outputs_; 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()
81 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<float *>(outputs_[0]->MutableData()), correct, tot… in TEST_F()
85 for (auto t : outputs_) delete t; in TEST_F()
90 std::vector<lite::Tensor *> outputs_; in TEST_F() local
[all …]
Dembedding_lookup_fp32_test.cc32 void ElTestInit(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, in ElTestInit() argument
54 outputs_->push_back(outputs_t); in ElTestInit()
61 std::vector<Tensor *> outputs_; 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()
75 for (unsigned int i = 0; i < outputs_.front()->shape().size(); ++i) { in TEST_F()
76 std::cout << outputs_.front()->shape()[i] << ' '; in TEST_F()
79 float *out = reinterpret_cast<float *>(outputs_.front()->MutableData()); in TEST_F()
80 for (int i = 0; i < outputs_.front()->ElementsNum(); ++i) { in TEST_F()
87 for (unsigned int i = 0; i < outputs_.size(); i++) { in TEST_F()
[all …]
Dconstant_of_shape_fp32_test.cc28 …TestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, float *a_ptr, in ConstantOfShapeTestInit() argument
41 outputs_->push_back(out_t); in ConstantOfShapeTestInit()
48 std::vector<lite::Tensor *> outputs_; 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()
64 float *output = reinterpret_cast<float *>(outputs_[0]->MutableData()); in TEST_F()
67 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<float *>(outputs_[0]->MutableData()), correct, tot… in TEST_F()
71 for (auto t : outputs_) delete t; in TEST_F()
Dskip_gram_fp32.cc34 void SkipGramTestInit(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, in SkipGramTestInit() argument
44 outputs_->push_back(output); in SkipGramTestInit()
55 std::vector<Tensor *> outputs_; in TEST_F() local
58 SkipGramTestInit(&inputs_, &outputs_, skip_gram_param_); in TEST_F()
64 …rnel::SkipGramCPUKernel(reinterpret_cast<OpParameter *>(skip_gram_param_), inputs_, outputs_, ctx); in TEST_F()
69 std::vector<StringPack> output = mindspore::lite::ParseTensorBuffer(outputs_[0]); in TEST_F()
79 for (unsigned int i = 0; i < outputs_.size(); i++) { in TEST_F()
80 delete outputs_[i]; in TEST_F()
Droi_pooling_fp32_tests.cc28 int ROIPoolingTestInit(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_,… in ROIPoolingTestInit() argument
43 outputs_->push_back(out_t); in ROIPoolingTestInit()
50 std::vector<lite::Tensor *> outputs_; in TEST_F() local
61 int total_size = ROIPoolingTestInit(&inputs_, &outputs_, a, b, a_shape, b_shape, c_shape); in TEST_F()
65 … = new kernel::ROIPoolingCPUKernel(reinterpret_cast<OpParameter *>(param), inputs_, outputs_, ctx); in TEST_F()
69 float *output = reinterpret_cast<float *>(outputs_[0]->MutableData()); in TEST_F()
72 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<float *>(outputs_[0]->MutableData()), correct, tot… in TEST_F()
76 for (auto t : outputs_) delete t; in TEST_F()
Ddeconvolution_fp32_tests.cc323 int DeConvTestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in DeConvTestInit1() argument
374 outputs_->push_back(out_t); in DeConvTestInit1()
477 std::vector<lite::Tensor *> outputs_; in TEST_F() local
484 int total_size = DeConvTestInit1(&inputs_, &outputs_, deconv_param, &correct); in TEST_F()
486 …nel::DeConvolutionCPUKernel(reinterpret_cast<OpParameter *>(deconv_param), inputs_, outputs_, ctx); in TEST_F()
491 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<float *>(outputs_[0]->MutableData()), correct, tot… in TEST_F()
495 for (auto t : outputs_) delete t; in TEST_F()
499 int DeConvTestInit2(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in DeConvTestInit2() argument
526 outputs_->push_back(out_t); in DeConvTestInit2()
546 std::vector<lite::Tensor *> outputs_; in TEST_F() local
[all …]
/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
50 outputs_->push_back(out_t); in PadInt8TestInit1()
65 std::vector<lite::Tensor *> outputs_; 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()
76 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<int8_t *>(outputs_[0]->MutableData()), correct, to… in TEST_F()
81 for (auto t : outputs_) delete t; in TEST_F()
85 int PadInt8TestInit2(std::vector<Tensor *> *inputs_, std::vector<Tensor *> *outputs_, PadParameter … in PadInt8TestInit2() argument
101 outputs_->push_back(out_t); in PadInt8TestInit2()
118 std::vector<lite::Tensor *> outputs_; in TEST_F() local
[all …]
Dconv_1x1_int8_tests.cc72 …t8TestInit1_perchannel(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in Conv1x1Int8TestInit1_perchannel() argument
104 outputs_->push_back(out_t); in Conv1x1Int8TestInit1_perchannel()
120 std::vector<lite::Tensor *> outputs_; in TEST_F() local
126 int total_size = Conv1x1Int8TestInit1_perchannel(&inputs_, &outputs_, conv_param, &correct); in TEST_F()
128 …::Convolution1x1Int8CPUKernel(reinterpret_cast<OpParameter *>(conv_param), inputs_, outputs_, ctx); in TEST_F()
132 …ASSERT_EQ(0, CompareOutputData(reinterpret_cast<int8_t *>(outputs_[0]->MutableData()), correct, to… in TEST_F()
137 for (auto t : outputs_) delete t; in TEST_F()
141 …t Conv1x1Int8TestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in Conv1x1Int8TestInit1() argument
171 outputs_->push_back(out_t); in Conv1x1Int8TestInit1()
189 std::vector<lite::Tensor *> outputs_; in TEST_F() local
[all …]
Ddeconv_int8_tests.cc338 int DeConvInt8TestInit1(std::vector<lite::Tensor *> *inputs_, std::vector<lite::Tensor *> *outputs_, in DeConvInt8TestInit1() argument
367 outputs_->push_back(out_t); in DeConvInt8TestInit1()
383 std::vector<lite::Tensor *> outputs_; in TEST_F() local
390 int total_size = DeConvInt8TestInit1(&inputs_, &outputs_, deconv_param, &correct); in TEST_F()
391 …kernel::DeConvInt8CPUKernel(reinterpret_cast<OpParameter *>(deconv_param), inputs_, outputs_, ctx); in TEST_F()
399 int8_t *out_data = reinterpret_cast<int8_t *>(outputs_[0]->data()); in TEST_F()
405 for (auto t : outputs_) delete t; in TEST_F()
/third_party/mindspore/mindspore/lite/micro/example/mnist_x86/src/
Dsession.cc35 outputs_.resize(1); in CompileGraph()
40outputs_[0] = new (std::nothrow) MTensor(String("int8toft32_Softmax-7_post0/output-0"), kNumberTyp… in CompileGraph()
41 MS_ERROR_IF_NULL(outputs_[0]); in CompileGraph()
56 void *outputs_data[outputs_.size()]; in RunGraph()
57 for (size_t i = 0; i < outputs_.size(); ++i) { in RunGraph()
58 outputs_data[i] = outputs_[i]->MutableData(); in RunGraph()
60 CopyOutputsData(outputs_data, outputs_.size()); in RunGraph()
78 for (auto &output : outputs_) { in ~LiteSession()
115 for (const auto &output : outputs_) { in GetOutputTensorNames()
122 for (const auto &output : outputs_) { in GetOutputByTensorName()
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/src/
Dsession.cc35 outputs_.resize(1); in CompileGraph()
40outputs_[0] = new (std::nothrow) MTensor(String("int8toft32_Softmax-7_post0/output-0"), kNumberTyp… in CompileGraph()
41 MS_ERROR_IF_NULL(outputs_[0]); in CompileGraph()
55 void *outputs_data[outputs_.size()]; in RunGraph()
56 for (size_t i = 0; i < outputs_.size(); ++i) { in RunGraph()
57 outputs_data[i] = outputs_[i]->MutableData(); in RunGraph()
59 CopyOutputsData(outputs_data, outputs_.size()); in RunGraph()
77 for (auto &output : outputs_) { in ~LiteSession()
114 for (const auto &output : outputs_) { in GetOutputTensorNames()
121 for (const auto &output : outputs_) { in GetOutputByTensorName()
/third_party/mindspore/tests/ut/cpp/kernel/cpu/
Dunique_cpu_kernel_test.cc36 outputs_.clear(); in SetUp()
47 outputs_.push_back(CreateKernelAddress(y_.data())); in CreateAddress()
48 outputs_.push_back(CreateKernelAddress(idx_.data())); in CreateAddress()
60 std::vector<AddressPtr> outputs_; member in mindspore::kernel::UniqueCpuKernelTest
70 unique_->Launch(inputs_, workspace_, outputs_); in TEST_F()
Dunique_with_pad_cpu_kernel_test.cc36 outputs_.clear(); in SetUp()
48 outputs_.push_back(CreateKernelAddress(out_.data())); in CreateAddress()
49 outputs_.push_back(CreateKernelAddress(idx_.data())); in CreateAddress()
62 std::vector<AddressPtr> outputs_; member in mindspore::kernel::UniqueWithPadCpuKernelTest
73 unique_with_pad_->Launch(inputs_, workspace_, outputs_); in TEST_F()
Dadam_delta_cpu_kernel_test.cc38 outputs_.clear(); in SetUp()
58 outputs_.push_back(CreateKernelAddress(delta_.data(), elem_num_)); in CreateAddress()
67 std::vector<AddressPtr> outputs_; member in mindspore::kernel::AdamDeltaCpuKernelTest
87 adam_delta_->Launch(inputs_, workspace_, outputs_); in TEST_F()
/third_party/ninja/src/
Dgraph.cc98 for (vector<Node*>::iterator o = edge->outputs_.begin(); in RecomputeDirty()
99 o != edge->outputs_.end(); ++o) { in RecomputeDirty()
151 for (vector<Node*>::iterator o = edge->outputs_.begin(); in RecomputeDirty()
152 o != edge->outputs_.end(); ++o) { in RecomputeDirty()
216 for (vector<Node*>::iterator o = edge->outputs_.begin(); in RecomputeOutputsDirty()
217 o != edge->outputs_.end(); ++o) { in RecomputeOutputsDirty()
353 int explicit_outs_count = edge_->outputs_.size() - edge_->implicit_outs_; in LookupVariable()
354 return MakePathList(&edge_->outputs_[0], explicit_outs_count, ' '); in LookupVariable()
440 for (vector<Node*>::const_iterator i = outputs_.begin(); in Dump()
441 i != outputs_.end() && *i != NULL; ++i) { in Dump()
[all …]
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()
60 for (vector<Node*>::iterator out = edge->outputs_.begin(); in AddTarget()
61 out != edge->outputs_.end(); ++out) { in AddTarget()
/third_party/mindspore/include/api/
Dkernel.h40 … : context_(ctx), inputs_(std::move(inputs)), outputs_(std::move(outputs)), primitive_(primitive) { in Kernel()
69 …virtual void set_outputs(const std::vector<mindspore::MSTensor> &out_tensors) { this->outputs_ = o… in set_outputs()
74 …virtual void set_output(mindspore::MSTensor out_tensor, int index) { this->outputs_[index] = out_t… in set_output()
82 virtual const std::vector<mindspore::MSTensor> &outputs() { return this->outputs_; } in outputs()
125 std::vector<mindspore::MSTensor> outputs_; variable
/third_party/mindspore/mindspore/lite/examples/runtime_gpu_extend/src/
Dcustom_add_kernel_gpu.cc59 auto out_shape = custom_common::GpuTensorInfo(&outputs_[0], &opencl_runtime_); in Prepare()
126 if (opencl_runtime_.SetKernelArg(kernel_, arg_idx++, outputs_[0].MutableData()) != kSuccess) { in Execute()
145 for (auto &tensor : outputs_) { in CheckSpecs()
152 if (inputs_.size() != 2 || outputs_.size() != 1) { in CheckSpecs()
153 std::cerr << "in size: " << inputs_.size() << ", out size: " << outputs_.size(); in CheckSpecs()
162 if (custom_common::CheckOutputs(outputs_) == lite::RET_OK) { in ReSize()
166 …egisterKernelInterface::GetKernelInterface({}, primitive_)->Infer(&inputs_, &outputs_, primitive_); in ReSize()
200 for (auto i = 0; i < outputs_.size(); ++i) { in PreProcess()
201 auto *output = &outputs_.at(i); in PreProcess()
/third_party/mindspore/mindspore/lite/examples/runtime_extend/src/
Dcustom_add_kernel.cc50 float *out = static_cast<float *>(outputs_[0].MutableData()); in Execute()
51 auto num = outputs_[0].ElementNum(); in Execute()
64 if (common::CheckOutputs(outputs_) != lite::RET_OK) { in PreProcess()
66 …egisterKernelInterface::GetKernelInterface({}, primitive_)->Infer(&inputs_, &outputs_, primitive_); in PreProcess()
77 for (auto &output : outputs_) { in PreProcess()
/third_party/mindspore/mindspore/lite/src/delegate/npu/
Dnpu_graph.cc50 MS_ASSERT(index < outputs_.size()); in set_output()
51 auto origin_tensor = this->outputs_[index]; in set_output()
59 this->outputs_[index] = out_tensor; in set_output()
306 if (find(outputs_.begin(), outputs_.end(), output) == outputs_.end()) { in Prepare()
321 if (find(outputs_.begin(), outputs_.end(), output) != outputs_.end()) { in Execute()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/
Dop_coder_builder.cc35 if (inputs_.empty() || outputs_.empty()) { in build()
40 MS_CHECK_PTR_RET_NULL(outputs_.at(kOutputIndex)); in build()
43 creator_func(inputs_, outputs_, node_, node_index_++, target_, schema_version); in build()
68 this->outputs_ = outputs; in outputs()
/third_party/gn/src/gn/
Daction_values.h35 SubstitutionList& outputs() { return outputs_; } in outputs()
36 const SubstitutionList& outputs() const { return outputs_; } in outputs()
61 SubstitutionList outputs_; variable

1234