Home
last modified time | relevance | path

Searched defs:input_w (Results 1 – 25 of 37) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dresize_bilinear_impl.cu22 …void ResizeBilinear(const T *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinear()
52 …sizeBilinearGrad(const half *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinearGrad()
85 …izeBilinearGrad(const float *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinearGrad()
125 …d CalResizeBilinear(const T *input, const int n, const int c, const int input_h, const int input_w, in CalResizeBilinear()
136 …sizeBilinearGrad(const half *input, const int n, const int c, const int input_h, const int input_w, in CalResizeBilinearGrad()
149 …izeBilinearGrad(const float *input, const int n, const int c, const int input_h, const int input_w, in CalResizeBilinearGrad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dconv2d_infer.c19 int ConvInferShape(int input_h, int input_w, int *output_h, int *output_w, ConvParameter *param) { in ConvInferShape()
98 int input_w = in_shape[2]; in Conv2dInferShape() local
Dpooling_grad_infer.c33 int input_w = input->shape_[2]; in PoolingGradInferShape() local
Ddepthwise_conv2d_infer.c38 int input_w = input->shape_[2]; in DepthwiseConv2dInferShape() local
Dpooling_infer.c42 int input_w = input->shape_[2]; in PoolingInferShape() local
Ddeconv2d_infer.c44 int32_t input_w = GetWidth(input); in Deconv2dInferShape() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dcrop_op.cc32 int32_t input_w = static_cast<int>(input->shape()[1]); in Compute() local
Dresize_with_bbox_op.cc35 int32_t input_w = input[0]->shape()[1]; in Compute() local
Dresize_op.cc36 int32_t input_w = static_cast<int>(input->shape()[1]); in Compute() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dconvolution_grad_filter.c53 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad16Arm() local
130 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad12Arm() local
200 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad8Arm() local
258 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad4Arm() local
308 int input_w = idx % out_w * conv_param->stride_w_; in Filtergrad2Arm() local
366 int input_w = idx % out_w * conv_param->stride_w_; in ConvDwFilterGrad() local
Dpack_ext.c48 int input_w = block_start % output_w * stride_w; in RollingIm2ColPackDwUnitFp32() local
99 int input_w = block_start % output_w * stride_w; in rolling_im2col_hwc() local
119 int input_w = block_start % output_w * stride_w; in rolling_im2col_hwc() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dconvolution_grad_filter.c60 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad32Arm() local
133 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad16Arm() local
190 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad8Arm() local
239 int input_w = idx % out_w * conv_param->stride_w_; in FilterGrad4Arm() local
288 int input_w = idx % out_w * conv_param->stride_w_; in FilterGradLeftoverArm() local
347 int input_w = idx % out_w * conv_param->stride_w_; in ConvDwFilterFp16Grad() local
Dpack_fp16_ext.c47 int input_w = block_start % output_w * stride_w; in RollingIm2ColPackDwUnitFp16() local
98 int input_w = block_start % output_w * stride_w; in RollingIm2ColPackUnitFp16() local
118 int input_w = block_start % output_w * stride_w; in RollingIm2ColPackUnitFp16() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/soft_dvpp/
Dsoft_dvpp_decode_resize_jpeg_op.cc40 int input_w = 0; in Compute() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.cc32 int64_t input_w = input_shape.at(kShapeW); in ApproximateFLOPs() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dpooling_fp16_grad.cc55 int input_w = in_shape.at(kNumShapeDim_2); in ReSize() local
Dconvolution_fp16_grad_input.cc81 auto *input_w = in_tensors_.at(1); in DoExecute() local
Dconvolution_fp16_grad_filter.cc174 int input_w = ci % out_w * conv_param->stride_w_; in DoExecute() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dpooling_grad.cc55 int input_w = in_shape.at(kNumShapeDim3); in ReSize() local
Dconvolution_grad_input.cc82 auto *input_w = in_tensors_.at(1); in Execute() local
Dconvolution.cc82 auto *input_w = in_tensors_.at(kWeightIndex); in Execute() local
Dconvolution_grad_filter.cc181 int input_w = ci % out_w * conv_param->stride_w_; in Execute() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc83 int input_w = conv_param_->input_w_; in InitTmpBuffer() local
/third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/
Dtransforms.py278 def _choose_candidate_by_constraints(max_trial, input_w, input_h, image_w, image_h, jitter, box, us… argument
330 def _correct_bbox_by_candidates(candidates, input_w, input_h, image_w, argument
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc33 int64_t input_w) { in GetSplitPadList()
131 int64_t input_w = in_shape.at(kAxisW); in CalSplitInShape() local

12