| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | sparse_to_dense_fp32.cc | 59 MS_CHECK_TRUE_MSG(output_dim <= DIMENSION_4D, RET_ERROR, "output_dim should <= 4"); in ReSize()
60 for (int i = 0; i < DIMENSION_4D - output_dim; i++) { in ReSize()
64 output_shape[i + DIMENSION_4D - output_dim] = output_shape_tmp[i]; in ReSize()
117 sparse_indices_vect[i] = new int[DIMENSION_4D]; in GenerateIndices()
122 for (int j = 0; j < DIMENSION_4D - 1; j++) { in GenerateIndices()
125 sparse_indices_vect[i][DIMENSION_4D - 1] = sparse_indices[i]; in GenerateIndices()
131 MS_ASSERT(true_dims <= DIMENSION_4D); in GenerateIndices()
133 sparse_indices_vect[i] = new int[DIMENSION_4D]; in GenerateIndices()
138 for (int j = 0; j < DIMENSION_4D - true_dims; j++) { in GenerateIndices()
142 sparse_indices_vect[i][j + DIMENSION_4D - true_dims] = sparse_indices[i * true_dims + j]; in GenerateIndices()
|
| D | depth_to_space_fp32.cc | 50 if (shape_size != DIMENSION_4D) { in ReSize()
51 MS_LOG(ERROR) << "Input shape size should be " << DIMENSION_4D; in ReSize()
54 int32_t in_strides[DIMENSION_4D]; in ReSize()
59 int32_t out_strides[DIMENSION_4D]; in ReSize()
73 MS_CHECK_TRUE_MSG(in_shape.size() == DIMENSION_4D, RET_ERROR, "input shape should be 4!"); in Run()
|
| D | space_to_batch_fp32.cc | 31 for (size_t i = 0; i < DIMENSION_4D; i++) { in ProcessInput()
35 ComputeStrides(param_->input_shape_, param_->in_stride_, DIMENSION_4D); in ProcessInput()
36 ComputeStrides(param_->output_shape_, param_->out_stride_, DIMENSION_4D); in ProcessInput()
75 for (size_t i = 0; i < DIMENSION_4D; i++) { in ReSize()
80 ComputeStrides(param_->input_shape_, param_->in_stride_, DIMENSION_4D); in ReSize()
81 ComputeStrides(param_->output_shape_, param_->out_stride_, DIMENSION_4D); in ReSize()
|
| D | crop_and_resize_fp32.cc | 160 CHECK_LESS_RETURN(input_shape.size(), DIMENSION_4D); in Run()
162 CHECK_LESS_RETURN(output_shape.size(), DIMENSION_4D); in Run()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | reduce_tensorrt.cc | 62 if (tensorrt_in_tensors_[0].trt_tensor_->getDimensions().nbDims == DIMENSION_4D && in AddInnerOp()
81 if (reduce_input->getDimensions().nbDims < DIMENSION_4D) { in AddInnerOp()
89 for (int i = unsqueeze_dims.nbDims; i < DIMENSION_4D; i++) { in AddInnerOp()
92 unsqueeze_dims.nbDims = DIMENSION_4D; in AddInnerOp()
107 if (in_tensors_[0].Shape().size() != DIMENSION_4D) { in AddInnerOp()
|
| D | scale_tensorrt.cc | 146 } else if (tensorrt_in_tensors_[0].trt_tensor_->getDimensions().nbDims == DIMENSION_4D && in PreProcessInputTensor()
158 } else if (tensorrt_in_tensors_[0].trt_tensor_->getDimensions().nbDims == DIMENSION_4D && in PreProcessInputTensor()
183 (input_data_shape.size() == DIMENSION_4D && axis == DIMENSION_3D)) { in GetScaleMode()
203 size_t unsqueeze_size = DIMENSION_4D - unsqueeze_shape.size(); in AddUnsqueezeOp()
|
| D | resize_tensorrt.cc | 46 …if (resize_in_tensor->getDimensions().nbDims == DIMENSION_4D && tensorrt_in_tensors_[0].format_ ==… in AddInnerOp()
132 if (out_shape.size() > DIMENSION_4D) { in SetOutputDims()
136 float scales[DIMENSION_4D]; in SetOutputDims()
|
| D | concate_tensorrt.cc | 62 if (tensorrt_in_tensors_[0].trt_tensor_->getDimensions().nbDims == DIMENSION_4D) { in AddInnerOp()
88 if (trt_input_tensors[0]->getDimensions().nbDims == DIMENSION_4D && out_format == Format::NCHW) { in AddInnerOp()
|
| D | deconvolution_tensorrt.cc | 54 if (tensorrt_in_tensors_[0].trt_tensor_->getDimensions().nbDims == DIMENSION_4D && in AddInnerOp()
159 if (out_pad_val.size() != DIMENSION_2D || padding_val.size() != DIMENSION_4D) { in SetAttributes()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_scale_parser.cc | 27 if (axis < -4 || axis >= DIMENSION_4D) { in GetAxisIndex()
36 *axis_index = (axis + DIMENSION_4D) % DIMENSION_4D; in GetAxisIndex()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | crop_fp32.c | 35 int64_t offset_pad[DIMENSION_4D] = {0}; in Crop4D()
36 Pad4DOffset(crop_param, offset_pad, DIMENSION_4D); in Crop4D()
73 int64_t offset_pad[DIMENSION_4D] = {0}; in Crop4DNoParallel()
74 Pad4DOffset(crop_param, offset_pad, DIMENSION_4D); in Crop4DNoParallel()
|
| D | roi_pooling_fp32.h | 29 int in_strides_[DIMENSION_4D];
30 int out_strides_[DIMENSION_4D];
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | resize.cc | 40 …if (in_shape.size() != DIMENSION_4D || out_shape.size() != DIMENSION_4D || in_shape[0] != out_shap… in CheckSpecs()
96 MS_CHECK_GE(in_shape.size(), DIMENSION_4D, RET_ERROR); in SetConstArgs()
97 MS_CHECK_GE(out_shape.size(), DIMENSION_4D, RET_ERROR); in SetConstArgs()
|
| D | sparse_to_dense.cc | 114 if (in_tensors_.size() < DIMENSION_3D || out_tensors_.at(0)->shape().size() > DIMENSION_4D) { in CheckSpecs()
118 …if (in_tensors_.at(0)->shape().size() > DIMENSION_4D || out_tensors_.at(0)->shape().size() > DIMEN… in CheckSpecs()
124 if ((in_tensors_[0]->shape()[1] > DIMENSION_4D)) { in CheckSpecs()
230 if (in_tensors_[0]->shape().size() <= DIMENSION_4D) { in InferShapeTo4D()
238 if (out_tensors_[0]->shape().size() <= DIMENSION_4D) { in InferShapeTo4D()
|
| D | transpose.cc | 42 if (in_ndim > DIMENSION_4D) { in CheckSpecs()
72 } else if (tensor_size_.NDim == DIMENSION_4D) { in Prepare()
96 if (in_tensors_[0]->shape().size() == DIMENSION_4D && in Prepare()
|
| D | softmax.cc | 49 if (in_shape.size() > DIMENSION_4D) { in CheckSpecs()
56 axis_ += DIMENSION_4D - in_shape.size(); in CheckSpecs()
|
| D | pad.cc | 46 if (in_ndim < DIMENSION_1D || in_ndim > DIMENSION_4D) { in CheckSpecs()
51 if (out_ndim < DIMENSION_1D || out_ndim > DIMENSION_4D) { in CheckSpecs()
|
| D | batch_to_space_nd.cc | 40 …if (in_tensors_[0]->shape().size() != DIMENSION_4D && out_tensors_[0]->shape().size() != DIMENSION… in CheckSpecs()
|
| D | space_to_batch_nd.cc | 39 …if (in_tensors_[0]->shape().size() != DIMENSION_4D && out_tensors_[0]->shape().size() != DIMENSION… in CheckSpecs()
|
| D | arithmetic_self.cc | 39 …if (in_tensors_[0]->shape().size() != DIMENSION_4D && in_tensors_[0]->shape().size() != DIMENSION_… in CheckSpecs()
71 if (output_shape.size() == DIMENSION_4D) { in SetGlobalLocal()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_conv_parser.cc | 66 if (weight_shape.empty() || weight_shape.size() < DIMENSION_4D) { in Parse()
123 if (weight_shape.empty() || weight_shape.size() < DIMENSION_4D) { in Parse()
143 MS_CHECK_GE(static_cast<int>(data_shape.size()), DIMENSION_4D, nullptr); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | group_depthwise_op_convert_pass.cc | 84 MS_CHECK_TRUE_RET(data_shape.size() == DIMENSION_4D, false); in Run()
85 MS_CHECK_TRUE_RET(weight_shape.size() == DIMENSION_4D, false); in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | transpose_int8.cc | 87 if (in_tensor->shape().size() == DIMENSION_4D && param->perm_[0] == 0 && param->perm_[1] == 2 && in GetNHNCTransposeFunc()
94 if (in_tensor->shape().size() == DIMENSION_4D && param->perm_[0] == 0 && param->perm_[1] == 3 && in GetNHNCTransposeFunc()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_subgraph.cc | 64 if (inputs_[i].Shape().size() != DIMENSION_4D) { in Init()
270 … if (out_op->GetInnerOutTensor()[index].trt_tensor_->getDimensions().nbDims == DIMENSION_4D && in MarkOutputs()
378 if (contruct_dim.d[DIMENSION_4D - 1] != inputs_[i].Shape()[DIMENSION_4D - 1]) { in ReSize()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_depthwise_fp16.cc | 33 MS_CHECK_TRUE_RET(in_tensors_.front()->shape().size() == DIMENSION_4D, RET_ERROR); in InitSlideParam()
34 MS_CHECK_TRUE_RET(out_tensors_.front()->shape().size() == DIMENSION_4D, RET_ERROR); in InitSlideParam()
|