| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | crop_fp32_test.cc | 33 int out_shape[4] = {2, 1, 1, 1}; in TEST_F() local
55 int out_shape[4] = {1, 1, 2, 2}; in TEST_F() local
78 int out_shape[4] = {2, 2, 2, 1}; in TEST_F() local
98 int out_shape[4] = {2, 2, 2, 1}; in TEST_F() local
119 int out_shape[4] = {2, 1, 1, 1}; in TEST_F() local
140 int out_shape[4] = {1, 1, 2, 2}; in TEST_F() local
162 int out_shape[4] = {2, 2, 2, 1}; in TEST_F() local
181 int out_shape[4] = {1, 1, 2, 2}; in TEST_F() local
207 int out_shape[4] = {1, 2, 2, 2}; in TEST_F() local
232 int out_shape[4] = {1, 2, 2, 2}; in TEST_F() local
[all …]
|
| D | reduce_fp32_tests.cc | 70 void TestReduceFp32::Prepare(const std::vector<int> &in_shape, const std::vector<int> &out_shape, f… in Prepare()
111 std::vector<int> out_shape{2, 1, 4, 3}; in TEST_F() local
138 std::vector<int> out_shape{2, 1, 1, 1}; in TEST_F() local
166 std::vector<int> out_shape{2, 1, 4, 3}; in TEST_F() local
193 std::vector<int> out_shape{1, 1, 1, 1}; in TEST_F() local
219 std::vector<int> out_shape{2, 1, 4, 3}; in TEST_F() local
248 std::vector<int> out_shape{2, 4, 4, 1}; in TEST_F() local
276 std::vector<int> out_shape{2, 4, 4, 1}; in TEST_F() local
302 std::vector<int> out_shape{1, 1, 1, 1}; in TEST_F() local
328 std::vector<int> out_shape{2, 1, 4, 3}; in TEST_F() local
[all …]
|
| D | space_to_batch_fp32_tests.cc | 37 std::vector<int> out_shape = {2, 2, 4, 1}; in TEST_F() local
58 std::vector<int> out_shape = {2, 4, 2, 1}; in TEST_F() local
79 std::vector<int> out_shape = {4, 2, 2, 1}; in TEST_F() local
|
| D | depth_to_space_fp32_test.cc | 35 int out_shape[4] = {1, 4, 4, 1}; in TEST_F() local
64 int out_shape[4] = {1, 2, 2, 2}; in TEST_F() local
|
| D | space_to_depth_fp32_tests.cc | 40 int out_shape[4] = {1, 2, 2, 4}; in TEST_F() local
65 std::vector<int> out_shape = {1, 2, 2, 4}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | resize_nearest_neighbor_int8_tests.cc | 50 …zeNearestNeighborInt8::Prepare(const std::vector<int> &in_shape, const std::vector<int> &out_shape, in Prepare()
92 std::vector<int> out_shape = {1, 4, 4, 1}; in TEST_F() local
112 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
138 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
165 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
191 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
|
| D | resize_bilinear_int8_tests.cc | 55 …estResizeBilinearInt8::Prepare(const std::vector<int> &in_shape, const std::vector<int> &out_shape, in Prepare()
92 std::vector<int> out_shape = {1, 4, 4, 1}; in TEST_F() local
108 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
133 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
159 std::vector<int> out_shape = {2, 4, 4, 5}; in TEST_F() local
|
| D | quant_dtype_cast_tests.cc | 56 std::vector<int> out_shape = {1, 4, 4, 1}; in TEST_F() local
106 std::vector<int> out_shape = {1, 4, 4, 1}; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | reshape_infer.c | 21 int CalShape(const int *data, const TensorC *const *inputs, int *out_shape, size_t *out_shape_size,… in CalShape()
47 int CalNewShape(const TensorC *in_tensor, int *out_shape, size_t out_shape_size) { in CalNewShape()
90 int CalShapeByType(const TensorC *const *inputs, size_t shape_size, int *out_shape, size_t *out_sha… in CalShapeByType()
180 int out_shape[MAX_SHAPE_SIZE] = {0}; in ReshapeInferShape() local
|
| D | reduce_infer.c | 20 int ReduceOnAllAxes(const TensorC *input, TensorC *output, int *out_shape, size_t out_shape_size, b… in ReduceOnAllAxes()
31 …xes(const TensorC *input, size_t num_axes, const int *actual_axes, TensorC *output, int *out_shape, in ReduceOnSelectedAxes()
84 int out_shape[MAX_SHAPE_SIZE] = {0}; in ReduceInferShape() local
|
| D | transpose_infer.c | 30 int *out_shape) { in SetOutputShape()
106 int out_shape[MAX_TRANSPOSE_DIM_SIZE] = {0}; in TransposeInferShape() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | resize.cc | 39 auto out_shape = out_tensors_[0]->shape(); in CheckSpecs() local
95 auto out_shape = out_tensors_[0]->shape(); in SetConstArgs() local
126 auto out_shape = GpuTensorInfo(out_tensors_[0]); in SetGlobalLocal() local
|
| D | batch_to_space_nd.cc | 63 std::vector<int> out_shape = out_tensors_[0]->shape(); in SetConstArgs() local
90 std::vector<int> out_shape = out_tensors_[0]->shape(); in SetGlobalLocal() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | strided_slice_base_coder.cc | 46 std::vector<int> out_shape = output_tensor_->shape(); in MatchFastPattern() local
65 std::vector<int> out_shape = output_tensor_->shape(); in InitFastRunParam() local
101 std::vector<int> out_shape = output_tensor_->shape(); in DoFastCode() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | transpose_int8_coder.cc | 30 auto out_shape = out_tensor->shape(); in Prepare() local
68 auto out_shape = output_tensors_[0]->shape(); in DoCode() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | strided_slice.cc | 48 auto out_shape = out_tensors_.front()->shape(); in InitFastRunParam() local
96 auto out_shape = out_tensors_.front()->shape(); in MatchFastPattern() local
115 auto out_shape = out_tensors_.front()->shape(); in FastRunImpl() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | transpose_fp32_coder.cc | 41 auto out_shape = output_tensor_->shape(); in Resize() local
69 auto out_shape = output_tensor_->shape(); in GetNHNCTransposeFunc() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | zeros.cc | 34 …std::vector<int64_t> out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("shape", shape_value… in ZerosInferShape() local
62 auto out_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(abs->BuildShape())[kShape]; in ZerosInferValue() local
|
| D | ones.cc | 33 …std::vector<int64_t> out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("shape", shape_value… in OnesInferShape() local
64 auto out_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(abs->BuildShape())[kShape]; in OnesInferValue() local
|
| D | reduce.cc | 56 std::vector<int64_t> out_shape; in infer_shape_reduce() local
76 auto out_shape = infer_shape_reduce(input_x_shape, axis_value, keep_dims); in InferShape() local
|
| D | fft_real.cc | 36 …auto out_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kSha… in FftRealInfer() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | transpose_int8.cc | 52 auto out_shape = out_tensor->shape(); in ReSize() local
86 auto out_shape = out_tensor->shape(); in GetNHNCTransposeFunc() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | resize_tensorrt.cc | 93 std::vector<float> out_shape; in SetOutputDims() local
148 …orRT::IsScaleOutputDim(const std::vector<int64_t> &in_shape, const std::vector<int64_t> &out_shape, in IsScaleOutputDim()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | transpose_fp32.cc | 68 auto out_shape = outTensor->shape(); in ReSize() local
125 auto out_shape = out_tensor->shape(); in GetNHNCTransposeFunc() local
|
| D | unique_fp32.cc | 46 std::vector<int> out_shape = out_tensors_.at(0)->shape(); in Run() local
|