| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | slice_int8_tests.cc | 33 lite::Tensor out_tensor(kNumberTypeInt8, {1, 2, 2, 3}); in TEST_F() local
38 out_tensor.set_data(output_data); in TEST_F()
49 out_tensor.AddQuantParam(quant_out); in TEST_F()
52 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
78 out_tensor.set_data(nullptr); in TEST_F()
84 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 2, 3}); in TEST_F() local
90 out_tensor.set_data(output_data); in TEST_F()
101 out_tensor.AddQuantParam(quant_out); in TEST_F()
104 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
128 out_tensor.set_data(nullptr); in TEST_F()
[all …]
|
| D | l2_norm_int8_tests.cc | 33 lite::Tensor out_tensor(kNumberTypeInt8, {2, 1, 1, 5}); in TEST_F() local
38 out_tensor.set_data(output_data); in TEST_F()
43 out_tensor.AddQuantParam(quant_out); in TEST_F()
46 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
73 out_tensor.set_data(nullptr); in TEST_F()
79 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 51}); in TEST_F() local
85 out_tensor.set_data(output_data); in TEST_F()
90 out_tensor.AddQuantParam(quant_out); in TEST_F()
93 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
119 out_tensor.set_data(nullptr); in TEST_F()
|
| D | relux_int8_tests.cc | 33 lite::Tensor out_tensor(kNumberTypeInt8, {2, 2}); in TEST_F() local
38 out_tensor.set_data(output_data); in TEST_F()
43 out_tensor.AddQuantParam(quant_out); in TEST_F()
46 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
73 out_tensor.set_data(nullptr); in TEST_F()
79 lite::Tensor out_tensor(kNumberTypeInt8, {2, 4}); in TEST_F() local
85 out_tensor.set_data(output_data); in TEST_F()
90 out_tensor.AddQuantParam(quant_out); in TEST_F()
93 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
121 out_tensor.set_data(nullptr); in TEST_F()
|
| D | sub_int_tests.cc | 34 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 5}); in TEST_F() local
41 out_tensor.set_data(output_data); in TEST_F()
48 out_tensor.AddQuantParam(quant_out); in TEST_F()
51 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
77 out_tensor.set_data(nullptr); in TEST_F()
84 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 5}); in TEST_F() local
91 out_tensor.set_data(output_data); in TEST_F()
98 out_tensor.AddQuantParam(quant_out); in TEST_F()
101 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
128 out_tensor.set_data(nullptr); in TEST_F()
|
| D | sigmoid_int8_tests.cc | 32 lite::Tensor out_tensor(kNumberTypeInt8, {4, 4}); in TEST_F() local
37 out_tensor.set_data(output_data); in TEST_F()
42 out_tensor.AddQuantParam(quant_out); in TEST_F()
45 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
72 out_tensor.set_data(nullptr); in TEST_F()
|
| D | hswish_int8_tests.cc | 35 lite::Tensor out_tensor(kNumberTypeInt8, {4, 4}); in TEST_F() local
40 out_tensor.set_data(output_data); in TEST_F()
45 out_tensor.AddQuantParam(quant_out); in TEST_F()
48 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
75 out_tensor.set_data(nullptr); in TEST_F()
|
| D | add_int8_tests.cc | 33 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 5}); in TEST_F() local
40 out_tensor.set_data(output_data); in TEST_F()
47 out_tensor.AddQuantParam(quant_out); in TEST_F()
50 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
75 out_tensor.set_data(nullptr); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | reverse_sequence_fp32_tests.cc | 32 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 3, 4, 2}); in TEST_F() local
40 out_tensor.set_data(output_data); in TEST_F()
42 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
64 EXPECT_EQ(out_tensor.ElementsNum(), 2 * 3 * 4 * 2); in TEST_F()
72 out_tensor.set_data(nullptr); in TEST_F()
79 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 3, 4, 2}); in TEST_F() local
87 out_tensor.set_data(output_data); in TEST_F()
89 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
111 EXPECT_EQ(out_tensor.ElementsNum(), 2 * 3 * 4 * 2); in TEST_F()
119 out_tensor.set_data(nullptr); in TEST_F()
[all …]
|
| D | tile_fp32_tests.cc | 30 lite::Tensor out_tensor(kNumberTypeFloat32, {4, 6}); in TEST_F() local
34 out_tensor.set_data(output_data); in TEST_F()
36 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
70 out_tensor.set_data(nullptr); in TEST_F()
76 lite::Tensor out_tensor(kNumberTypeFloat32, {4, 2}); in TEST_F() local
80 out_tensor.set_data(output_data); in TEST_F()
82 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
118 out_tensor.set_data(nullptr); in TEST_F()
124 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 4}); in TEST_F() local
128 out_tensor.set_data(output_data); in TEST_F()
[all …]
|
| D | lsh_projection_fp32_tests.cc | 42 lite::Tensor out_tensor(kNumberTypeInt32, {6}); in TEST_F() local
51 out_tensor.set_data(output_data); in TEST_F()
54 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
80 out_tensor.set_data(nullptr); in TEST_F()
87 lite::Tensor out_tensor(kNumberTypeInt32, {3}); in TEST_F() local
94 out_tensor.set_data(output_data); in TEST_F()
97 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
123 out_tensor.set_data(nullptr); in TEST_F()
131 lite::Tensor out_tensor(kNumberTypeInt32, {3}); in TEST_F() local
141 out_tensor.set_data(output_data); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | pooling_int8_coder.cc | 34 Tensor *out_tensor = output_tensors_.at(kOutputIndex); in DoCode() local
36 MS_CHECK_PTR(out_tensor); in DoCode()
41 pooling_parameter->output_batch_ = out_tensor->Batch(); in DoCode()
42 pooling_parameter->output_channel_ = out_tensor->Channel(); in DoCode()
43 pooling_parameter->output_h_ = out_tensor->Height(); in DoCode()
44 pooling_parameter->output_w_ = out_tensor->Width(); in DoCode()
48 std::vector<LiteQuantParam> out_quant_args = out_tensor->quant_params(); in DoCode()
67 …code.CodeFunction("MaxPoolingInt8", in_tensor, out_tensor, "(PoolingParameter *)&pooling_parameter… in DoCode()
70 …code.CodeFunction("AvgPoolingInt8", in_tensor, out_tensor, "(PoolingParameter *)&pooling_parameter… in DoCode()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | reshape_base.cc | 43 auto out_tensor = out_tensors().front(); in Run() local
44 CHECK_NULL_RETURN(out_tensor); in Run()
45 if (in_tensor->data_type() != out_tensor->data_type() || in_tensor->data() == nullptr || in Run()
46 in_tensor->Size() != out_tensor->Size()) { in Run()
51 if (in_tensor->allocator() == nullptr || in_tensor->allocator() != out_tensor->allocator() || in Run()
54 CHECK_NULL_RETURN(out_tensor->data()); in Run()
65 auto out_data = static_cast<uint8_t *>(out_tensor->data()); in Run()
86 out_tensor->FreeData(); in Run()
87 out_tensor->ResetRefCount(); in Run()
88 out_tensor->set_data(in_tensor->data()); in Run()
[all …]
|
| D | pooling_base.cc | 62 auto *out_tensor = out_tensors_.at(kOutputIndex); in SetQuantParam() local
63 auto out_quant_arg = out_tensor->quant_params(); in SetQuantParam()
107 auto out_tensor = this->out_tensors_.front(); in ReSize() local
112 pooling_param_->output_batch_ = out_tensor->Batch(); in ReSize()
113 pooling_param_->output_channel_ = out_tensor->Channel(); in ReSize()
114 pooling_param_->output_h_ = out_tensor->Height(); in ReSize()
115 pooling_param_->output_w_ = out_tensor->Width(); in ReSize()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | strided_slice_tests.cc | 55 lite::Tensor out_tensor(kNumberTypeFloat32, {1, 1, 2}); in TEST_F() local
59 out_tensor.set_data(output_data); in TEST_F()
70 std::vector<lite::Tensor *> outputs = {&out_tensor}; in TEST_F()
92 out_tensor.set_data(nullptr); in TEST_F()
99 lite::Tensor out_tensor(kNumberTypeFloat32, {1, 1, 2, 1, 1, 1, 1}); variable
103 out_tensor.set_data(output_data);
114 std::vector<lite::Tensor *> outputs = {&out_tensor};
136 out_tensor.set_data(nullptr);
143 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 2, 1, 1, 1, 1}); variable
147 out_tensor.set_data(output_data);
[all …]
|
| /third_party/mindspore/mindspore/lite/src/delegate/ |
| D | delegate_utils.h | 64 for (auto out_tensor : op->outputs()) { in GetGraphOutTensors() local
65 if (!is_op_input(out_tensor)) { in GetGraphOutTensors()
66 outputs.push_back(out_tensor); in GetGraphOutTensors()
107 for (auto out_tensor : (*iter)->outputs()) { in GraphInTensors()
108 … if (std::find(all_in_tensors.begin(), all_in_tensors.end(), out_tensor) != all_in_tensors.end()) { in GraphInTensors()
109 in_tensors.push_back(out_tensor); in GraphInTensors()
122 for (auto out_tensor : op->outputs()) { in GraphOutTensors() local
123 if (find(out_tensors.begin(), out_tensors.end(), out_tensor) == out_tensors.end()) { in GraphOutTensors()
124 all_out_tensors.push_back(out_tensor); in GraphOutTensors()
159 for (auto out_tensor : cur_op->outputs()) { in FindNextOps() local
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/ |
| D | carry_data_quant_param_propogator.cc | 33 auto &out_tensor = graph->allTensors.at(node.outputIndex.at(0)); in PropogateQuantParams() local
34 auto out_quant_param = GetTensorQuantParam(out_tensor); in PropogateQuantParams()
61 auto &out_tensor = graph->allTensors.at(i); in PropogateQuantParams() local
62 MS_ASSERT(out_tensor != nullptr); in PropogateQuantParams()
63 auto out_quant_param = GetTensorQuantParam(out_tensor); in PropogateQuantParams()
65 out_tensor->quantParams.emplace_back(std::move(in_quant_param)); in PropogateQuantParams()
71 out_tensor->quantParams.front() = std::move(in_quant_param); in PropogateQuantParams()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | size_infer.c | 28 TensorC *out_tensor = outputs[0]; in SizeInferShape() local
29 out_tensor->data_type_ = kNumberTypeInt32; in SizeInferShape()
30 out_tensor->format_ = in_tensor->format_; in SizeInferShape()
35 out_tensor->shape_size_ = 1; in SizeInferShape()
36 out_tensor->shape_[0] = 1; in SizeInferShape()
|
| D | shape_infer.c | 28 TensorC *out_tensor = outputs[0]; in ShapeInferShape() local
30 out_tensor->data_type_ = kNumberTypeInt32; in ShapeInferShape()
31 out_tensor->format_ = in_tensor->format_; in ShapeInferShape()
35 out_tensor->shape_size_ = 1; in ShapeInferShape()
36 out_tensor->shape_[0] = (int)(in_tensor->shape_size_); in ShapeInferShape()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | transpose_int8.cc | 50 auto out_tensor = out_tensors_.front(); in ReSize() local
52 auto out_shape = out_tensor->shape(); in ReSize()
84 …eInt8CPUKernel::GetNHNCTransposeFunc(const lite::Tensor *in_tensor, const lite::Tensor *out_tensor, in GetNHNCTransposeFunc() argument
86 auto out_shape = out_tensor->shape(); in GetNHNCTransposeFunc()
105 auto out_tensor = out_tensors_.front(); in Run() local
108 auto out_dims = out_tensor->shape(); in Run()
112 out_ptr_ = reinterpret_cast<int8_t *>(out_tensor->data()); in Run()
114 GetNHNCTransposeFunc(in_tensor, out_tensor, transpose_param_); in Run()
130 if (out_tensor->shape().size() > DIMENSION_6D) { in Run()
|
| D | resize_int8.cc | 152 auto out_tensor = out_tensors_.front(); in CalRatio() local
153 auto out_width = out_tensor->Width(); in CalRatio()
154 auto out_height = out_tensor->Height(); in CalRatio()
224 auto out_tensor = out_tensors_.front(); in CalFloatRatio() local
225 auto out_width = out_tensor->Width(); in CalFloatRatio()
226 auto out_height = out_tensor->Height(); in CalFloatRatio()
335 auto out_tensor = out_tensors_.front(); in RunImpl() local
336 auto out_c = out_tensor->Channel(); in RunImpl()
337 int plane = out_tensor->Height() * out_tensor->Width(); in RunImpl()
345 … ResizeBilinearInt8(input_data, out_ptr, out_tensor->Batch(), input->Height(), input->Width(), in RunImpl()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | scatter_nd_update_fp32.cc | 140 auto out_tensor = out_tensors().front(); in Run() local
141 if (in_tensor->allocator() == nullptr || in_tensor->allocator() != out_tensor->allocator() || in Run()
143 memcpy(out_tensor->data(), in_tensor->data(), in_tensor->Size()); in Run()
145 out_tensor->FreeData(); in Run()
146 out_tensor->ResetRefCount(); in Run()
147 in_tensor->allocator()->IncRefCount(in_tensor->data(), out_tensor->ref_count()); in Run()
148 out_tensor->set_data(in_tensor->data()); in Run()
149 out_tensor->set_own_data(in_tensor->own_data()); in Run()
150 output_ptr_ = reinterpret_cast<float *>(out_tensor->data()); in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | quant_dtype_cast_fp16.cc | 36 auto out_tensor = out_tensors_.front(); in Init() local
38 CHECK_NULL_RETURN(out_tensor); in Init()
42 … if (in_tensor->data_type() != kNumberTypeFloat16 || out_tensor->data_type() != kNumberTypeInt8) { in Init()
49 … if (in_tensor->data_type() != kNumberTypeInt8 || out_tensor->data_type() != kNumberTypeFloat16) { in Init()
56 … if (in_tensor->data_type() != kNumberTypeFloat16 || out_tensor->data_type() != kNumberTypeUInt8) { in Init()
63 … if (in_tensor->data_type() != kNumberTypeUInt8 || out_tensor->data_type() != kNumberTypeFloat16) { in Init()
186 auto out_tensor = outputs.front(); in CpuQuantDTypeCastFp16KernelCreator() local
189 … if (in_tensor->data_type() != kNumberTypeFloat16 || out_tensor->data_type() != kNumberTypeInt8) { in CpuQuantDTypeCastFp16KernelCreator()
195 … if (in_tensor->data_type() != kNumberTypeInt8 || out_tensor->data_type() != kNumberTypeFloat16) { in CpuQuantDTypeCastFp16KernelCreator()
201 … if (in_tensor->data_type() != kNumberTypeFloat16 || out_tensor->data_type() != kNumberTypeUInt8) { in CpuQuantDTypeCastFp16KernelCreator()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_address.cc | 145 mindspore::tensor::TensorPtr out_tensor = std::make_shared<tensor::Tensor>(host_type, host_shape); in LoadMemToHost() local
146 size_t host_size = out_tensor->data().nbytes(); in LoadMemToHost()
147 auto ret_rt_memcpy = SyncDeviceToHost(host_shape, host_size, host_type, out_tensor->data_c()); in LoadMemToHost()
157 tensor_data->SetTensor(out_tensor); in LoadMemToHost()
158 tensor_data->SetDataPtr(static_cast<char *>(out_tensor->data_c())); in LoadMemToHost()
159 tensor_data->SetByteSize(out_tensor->data().nbytes()); in LoadMemToHost()
161 tensor_data->SetShape(out_tensor->shape()); in LoadMemToHost()
|
| /third_party/mindspore/mindspore/lite/test/st/ |
| D | graph_test.cc | 65 tensor::MSTensor *out_tensor = out_tensor_pair.second; in TEST_F() local
67 void *out_data = malloc(out_tensor->Size()); in TEST_F()
70 out_tensor->set_data(out_data); in TEST_F()
71 out_tensor->set_allocator(nullptr); in TEST_F()
147 for (MSTensor &out_tensor : outputs) { in TEST_F()
148 void *out_data = malloc(out_tensor.DataSize()); in TEST_F()
151 out_tensor.SetData(out_data); in TEST_F()
152 out_tensor.SetAllocator(nullptr); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/examples/runtime_cpp/ |
| D | main.cc | 172 auto out_tensor = output_vec.front(); in GetOutputsByNodeName() local
173 if (out_tensor == nullptr) { in GetOutputsByNodeName()
177 …std::cout << "tensor size is:" << out_tensor.DataSize() << " tensor elements num is:" << out_tenso… in GetOutputsByNodeName()
180 if (out_tensor.DataType() != mindspore::DataType::kNumberTypeFloat32) { in GetOutputsByNodeName()
184 auto out_data = reinterpret_cast<float *>(out_tensor.MutableData()); in GetOutputsByNodeName()
190 for (int i = 0; i < out_tensor.ElementNum() && i < 10; i++) { in GetOutputsByNodeName()
200 auto out_tensor = model->GetOutputByTensorName(tensor_name); in GetOutputByTensorName() local
201 if (out_tensor == nullptr) { in GetOutputByTensorName()
205 …std::cout << "tensor size is:" << out_tensor.DataSize() << " tensor elements num is:" << out_tenso… in GetOutputByTensorName()
208 if (out_tensor.DataType() != mindspore::DataType::kNumberTypeFloat32) { in GetOutputByTensorName()
[all …]
|