Home
last modified time | relevance | path

Searched defs:out_tensor (Results 1 – 25 of 116) sorted by relevance

12345

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dleaky_relu_int8.cc53 auto *out_tensor = out_tensors_.at(kOutputIndex); in Init() local
80 auto *out_tensor = out_tensors_.at(kOutputIndex); in ReSize() local
123 auto out_tensor = out_tensors_.at(kOutputIndex); in DoExecute() local
Dtranspose_int8.cc50 auto out_tensor = out_tensors_.front(); in ReSize() local
84 …eInt8CPUKernel::GetNHNCTransposeFunc(const lite::Tensor *in_tensor, const lite::Tensor *out_tensor, in GetNHNCTransposeFunc()
105 auto out_tensor = out_tensors_.front(); in Run() local
Dcrop_int8.cc38 auto *out_tensor = out_tensors_.at(kOutputIndex); in Init() local
66 auto out_tensor = out_tensors_.at(kOutputIndex); in DoExecute() local
Darithmetic_self_int8.cc39 auto *out_tensor = out_tensors_.at(kOutputIndex); in Init() local
107 auto out_tensor = out_tensors_.at(0); in Run() local
Dresize_int8.cc152 auto out_tensor = out_tensors_.front(); in CalRatio() local
224 auto out_tensor = out_tensors_.front(); in CalFloatRatio() local
335 auto out_tensor = out_tensors_.front(); in RunImpl() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dpooling_base.cc62 auto *out_tensor = out_tensors_.at(kOutputIndex); in SetQuantParam() local
107 auto out_tensor = this->out_tensors_.front(); in ReSize() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/
Dcarry_data_quant_param_propogator.cc33 auto &out_tensor = graph->allTensors.at(node.outputIndex.at(0)); in PropogateQuantParams() local
61 auto &out_tensor = graph->allTensors.at(i); in PropogateQuantParams() local
/third_party/mindspore/mindspore/lite/src/delegate/
Ddelegate_utils.h64 for (auto out_tensor : op->outputs()) { in GetGraphOutTensors() local
122 for (auto out_tensor : op->outputs()) { in GraphOutTensors() local
159 for (auto out_tensor : cur_op->outputs()) { in FindNextOps() local
195 for (auto out_tensor : outputs) { in GetGraphInOutOps() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconcat_fp16.cc56 auto &out_tensor = out_tensors_.at(0); in MallocTmpBuffer() local
81 auto &out_tensor = out_tensors_.at(0); in FreeTmpBuffer() local
Dquant_dtype_cast_fp16.cc36 auto out_tensor = out_tensors_.front(); in Init() local
186 auto out_tensor = outputs.front(); in CpuQuantDTypeCastFp16KernelCreator() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dslice_int8_tests.cc33 lite::Tensor out_tensor(kNumberTypeInt8, {1, 2, 2, 3}); in TEST_F() local
84 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 2, 3}); in TEST_F() local
136 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 2, 2, 3}); in TEST_F() local
188 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 1, 2, 2, 3}); in TEST_F() local
240 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 1, 1, 1, 2, 3}); in TEST_F() local
293 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 1, 1, 1, 2, 3}); in TEST_F() local
346 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 1, 1, 1, 2, 3}); in TEST_F() local
401 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 1, 1, 1, 2, 3}); in TEST_F() local
Dl2_norm_int8_tests.cc33 lite::Tensor out_tensor(kNumberTypeInt8, {2, 1, 1, 5}); in TEST_F() local
79 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 51}); in TEST_F() local
Drelux_int8_tests.cc33 lite::Tensor out_tensor(kNumberTypeInt8, {2, 2}); in TEST_F() local
79 lite::Tensor out_tensor(kNumberTypeInt8, {2, 4}); in TEST_F() local
Dsub_int_tests.cc34 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 5}); in TEST_F() local
84 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 2, 5}); in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dstrided_slice_tests.cc55 lite::Tensor out_tensor(kNumberTypeFloat32, {1, 1, 2}); in TEST_F() local
99 lite::Tensor out_tensor(kNumberTypeFloat32, {1, 1, 2, 1, 1, 1, 1}); variable
143 lite::Tensor out_tensor(kNumberTypeInt8, {1, 1, 1, 2, 1, 1, 1, 1}); variable
189 lite::Tensor out_tensor(kNumberTypeFloat32, {1, 2, 2, 1, 1, 1, 1}); in TEST_F() local
235 lite::Tensor out_tensor(kNumberTypeFloat32, {1, 2, 2, 1, 1, 1, 1}); in TEST_F() local
281 lite::Tensor out_tensor(kNumberTypeInt8, {2, 3, 4}); in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dtile_fp32_tests.cc30 lite::Tensor out_tensor(kNumberTypeFloat32, {4, 6}); in TEST_F() local
76 lite::Tensor out_tensor(kNumberTypeFloat32, {4, 2}); in TEST_F() local
124 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 4}); in TEST_F() local
Dreverse_sequence_fp32_tests.cc32 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 3, 4, 2}); in TEST_F() local
79 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 3, 4, 2}); in TEST_F() local
126 lite::Tensor out_tensor(kNumberTypeFloat32, {2, 3, 4, 2}); in TEST_F() local
Dlsh_projection_fp32_tests.cc42 lite::Tensor out_tensor(kNumberTypeInt32, {6}); in TEST_F() local
87 lite::Tensor out_tensor(kNumberTypeInt32, {3}); in TEST_F() local
131 lite::Tensor out_tensor(kNumberTypeInt32, {3}); in TEST_F() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dtranspose_fp32.cc121 …oseCPUKernel::GetNHNCTransposeFunc(const lite::Tensor *in_tensor, const lite::Tensor *out_tensor) { in GetNHNCTransposeFunc()
166 auto &out_tensor = out_tensors_.front(); in Run() local
Dsize_fp32.cc38 auto out_tensor = out_tensors_.front(); in Run() local
Dshape_fp32.cc38 auto out_tensor = out_tensors_.front(); in Run() local
Dinvert_permutation_fp32.cc52 auto out_tensor = out_tensors_.front(); in Run() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dsize_infer.c28 TensorC *out_tensor = outputs[0]; in SizeInferShape() local
Dshape_infer.c28 TensorC *out_tensor = outputs[0]; in ShapeInferShape() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_cast_parser.cc35 const auto &out_tensor = tflite_subgraph->tensors[tflite_op->outputs.front()]; in Parse() local

12345