| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | concat_fp16.cc | 44 for (const auto &in_tensor : in_tensors_) { in MallocTmpBuffer() local
70 auto &in_tensor = in_tensors_.at(i); in FreeTmpBuffer() local
102 const auto in_tensor = in_tensors_.at(i); in Run() local
|
| D | quant_dtype_cast_fp16.cc | 35 auto in_tensor = in_tensors_.front(); in Init() local
82 auto in_tensor = in_tensors_.front(); in ReSize() local
185 auto in_tensor = inputs.front(); in CpuQuantDTypeCastFp16KernelCreator() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | concat.cc | 91 for (auto &in_tensor : in_tensors_) { in CheckSpecs() local
134 for (auto &in_tensor : in_tensors_) { in SetConstArgs() local
150 for (auto &in_tensor : in_tensors_) { in SetConstArgs() local
192 for (auto in_tensor : in_tensors_) { in ConvertWeightToTensor() local
243 for (auto const &in_tensor : in_tensors_) { in Prepare() local
|
| D | scale.cc | 43 auto in_tensor = in_tensors_.at(0); in CheckSpecs() local
85 auto *in_tensor = in_tensors_[0]; in InitWeights() local
159 auto in_tensor = in_tensors_.at(0); in Prepare() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | transpose_int8.cc | 49 auto in_tensor = in_tensors_.front(); in ReSize() local
84 void TransposeInt8CPUKernel::GetNHNCTransposeFunc(const lite::Tensor *in_tensor, const lite::Tensor… in GetNHNCTransposeFunc()
104 auto in_tensor = in_tensors_.front(); in Run() local
|
| D | split_int8.cc | 42 auto in_tensor = in_tensors_.at(kInputIndex); in Init() local
99 auto in_tensor = in_tensors_.at(kInputIndex); in Run() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/ |
| D | carry_data_quant_param_propogator.cc | 40 auto &in_tensor = graph->allTensors.at(node.inputIndex.at(0)); in PropogateQuantParams() local
52 auto &in_tensor = graph->allTensors.at(node.inputIndex.at(0)); in PropogateQuantParams() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/ |
| D | delegate_utils.h | 41 for (auto in_tensor : op->inputs()) { in GetGraphInTensors() local
95 for (auto in_tensor : op->inputs()) { in GraphInTensors() local
146 for (auto in_tensor : cur_op->inputs()) { in FindPreOps() local
182 for (auto in_tensor : inputs) { in GetGraphInOutOps() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | strided_slice_tests.cc | 29 void InitStridedSliceParam(StridedSliceParameter *param, const lite::Tensor *in_tensor, in InitStridedSliceParam()
54 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4}); in TEST_F() local
98 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4, 1, 1, 1, 1}); variable
142 lite::Tensor in_tensor(kNumberTypeInt8, {2, 2, 2, 3, 1, 1, 1, 1}); variable
188 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4, 1, 1, 1, 1}); in TEST_F() local
234 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4, 1, 1, 1, 1}); in TEST_F() local
280 lite::Tensor in_tensor(kNumberTypeInt8, {2, 3, 4}); in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_kernel.cc | 30 return std::all_of(in_tensors.begin(), in_tensors.end(), [&](lite::Tensor *in_tensor) { in IsReady()
86 for (auto &in_tensor : in_tensors()) { in DoExecute() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | quant_dtype_cast.cc | 41 auto in_tensor = in_tensors_.front(); in Init() local
61 auto in_tensor = in_tensors_.front(); in ReSize() local
|
| D | split_base.cc | 41 int SplitBaseCPUKernel::CheckAndInitSplitParam(const lite::Tensor &in_tensor, SplitParameter *param… in CheckAndInitSplitParam()
94 auto in_tensor = in_tensors_.front(); in ReSize() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | slice_int8_tests.cc | 32 lite::Tensor in_tensor(kNumberTypeInt8, {1, 3, 2, 3}); in TEST_F() local
83 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 3, 2, 3}); in TEST_F() local
135 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 3, 2, 3}); in TEST_F() local
187 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 3, 2, 3}); in TEST_F() local
239 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
292 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
345 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
400 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
|
| D | l2_norm_int8_tests.cc | 32 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 5}); in TEST_F() local
78 lite::Tensor in_tensor(kNumberTypeInt8, {1, 1, 1, 51}); in TEST_F() local
|
| D | relux_int8_tests.cc | 32 lite::Tensor in_tensor(kNumberTypeInt8, {2, 2}); in TEST_F() local
78 lite::Tensor in_tensor(kNumberTypeInt8, {2, 4}); in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | tile_fp32_tests.cc | 29 lite::Tensor in_tensor(kNumberTypeFloat32, {2, 2}); in TEST_F() local
75 lite::Tensor in_tensor(kNumberTypeFloat32, {2, 2}); in TEST_F() local
123 lite::Tensor in_tensor(kNumberTypeFloat32, {2, 2}); in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/ |
| D | npu_subgraph.cc | 49 void NPUSubGraph::set_input(mindspore::MSTensor in_tensor, int index) { in set_input()
105 for (auto in_tensor : cur_op->inputs()) { in FindPreOps() local
159 auto in_tensor = op->inputs()[i]; in BuildNPUInputOp() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | scale_fp32.cc | 87 auto in_tensor = in_tensors_.at(0); in CalculateParameter() local
178 auto in_tensor = in_tensors_.front(); in Run() local
|
| D | transpose_fp32.cc | 121 int TransposeCPUKernel::GetNHNCTransposeFunc(const lite::Tensor *in_tensor, const lite::Tensor *out… in GetNHNCTransposeFunc()
165 auto &in_tensor = in_tensors_.front(); in Run() local
|
| D | size_fp32.cc | 37 auto in_tensor = in_tensors_.front(); in Run() local
|
| D | shape_fp32.cc | 39 auto in_tensor = in_tensors_.front(); in Run() local
|
| D | invert_permutation_fp32.cc | 51 auto in_tensor = in_tensors_.front(); in Run() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_subgraph.cc | 141 nvinfer1::ITensor *TensorRTSubGraph::SetTensorRTNetworkInput(const mindspore::MSTensor &in_tensor) { in SetTensorRTNetworkInput()
208 for (auto in_tensor : cur_op->inputs()) { in BuildTensorRTGraph() local
484 …er TensorRTSubGraph::FindTensorRTInputs(TensorRTOp *cur_op, const mindspore::MSTensor &in_tensor) { in FindTensorRTInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | size_infer.c | 27 const TensorC *in_tensor = inputs[0]; in SizeInferShape() local
|
| D | shape_infer.c | 27 const TensorC *in_tensor = inputs[0]; in ShapeInferShape() local
|