Home
last modified time | relevance | path

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

1234

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/
Dconcat_checker.cc27 bool CheckConcatInputW(const ShapeVector &input_shape, int64_t axis, int64_t input_w) { in CheckConcatInputW()
74 int64_t input_w; in Check() local
Dpooling_checker.cc90 bool CheckAttr(const api::PrimitivePtr &primitive, int64_t input_w) { in CheckAttr()
117 int64_t input_w; in Check() local
Dconv2d_checker.cc119 bool CheckAttr(const api::CNodePtr &op, const api::PrimitivePtr &primitive, int64_t input_w) { in CheckAttr()
179 int64_t input_w; in Check() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/ascend910b/
Ddvpp_normalize_v2_op.cc60 int64_t input_w = input->GetShape()[kWidthIndexNCHW]; in Compute() local
74 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
Ddvpp_crop_op.cc49 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
72 int32_t input_w = inputs[0][kWidthIndex]; in OutputShape() local
Ddvpp_resize_op.cc58 int32_t input_w = size[kWidthIndex]; in Compute() local
121 int32_t input_w = inputs[0][kWidthIndex]; in OutputShape() local
Ddvpp_pad_op.cc71 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
97 int32_t input_w = inputs[0][kWidthIndex]; in OutputShape() local
Ddvpp_convert_color_op.cc48 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
Ddvpp_equalize_op.cc48 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
Ddvpp_perspective_op.cc62 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
Ddvpp_erase_op.cc51 int32_t input_w = size[kWidthIndex]; in Compute() local
Ddvpp_auto_contrast_op.cc51 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
Ddvpp_adjust_hue_op.cc47 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
Ddvpp_invert_op.cc47 int64_t input_w = input->GetShape()[kWidthIndexNHWC]; in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dpooling_infer.c21 int ComputePadList(PoolingParameter *param, int input_h, int input_w, int output_h, int output_w) { in ComputePadList()
63 int input_w = input->shape_[2]; in PoolingInferShape() local
Dformat_transpose_infer.c45 int input_w = GetWidth(input); in FormatTransposeInferShape() local
Dpooling_grad_infer.c33 int input_w = input->shape_[2]; in PoolingGradInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dresize_bilinear_impl.cu34 …void ResizeBilinear(const T *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinear()
64 … ResizeBilinear_HPC(const T *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinear_HPC()
95 … const int input_w, const int output_h, const int output_w, const int nchw, in ResizeBilinearGradHalf()
128 … ResizeBilinearGrad(const T *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinearGrad()
162 … const int input_w, const int output_h, const int output_w, const int nchw, in ResizeBilinearGradHalf_HPC()
195 …izeBilinearGrad_HPC(const T *input, const int n, const int c, const int input_h, const int input_w, in ResizeBilinearGrad_HPC()
235 …t CalResizeBilinear(const T *input, const int n, const int c, const int input_h, const int input_w, in CalResizeBilinear()
252 …BilinearGradHalf(const half *input, const int n, const int c, const int input_h, const int input_w, in CalResizeBilinearGradHalf()
273 …lResizeBilinearGrad(const T *input, const int n, const int c, const int input_h, const int input_w, in CalResizeBilinearGrad()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dcrop_op.cc31 auto input_w = static_cast<int>(input->shape()[1]); in Compute() local
Dresize_with_bbox_op.cc35 auto input_w = static_cast<int32_t>(input[0]->shape()[1]); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dpad_v3.cc116 int64_t input_w = input_shape[kNum2]; in EdgeCompute1D() local
137 int64_t input_w = input_shape[kNum3]; in EdgeCompute2D() local
164 int64_t input_w = input_shape[kNum4]; in EdgeCompute3D() local
222 int64_t input_w = input_shape[kNum2]; in ReflectCompute1D() local
242 int64_t input_w = input_shape[kNum3]; in ReflectCompute2D() local
269 int64_t input_w = input_shape[kNum4]; in ReflectCompute3D() local
313 int64_t input_w = input_shape[kNum2]; in CircularCompute1D() local
335 int64_t input_w = input_shape[kNum3]; in CircularCompute2D() local
364 int64_t input_w = input_shape[kNum4]; in CircularCompute3D() local
Dfractional_max_pool_grad_with_fixed_ksize.cc81 int64_t input_w = out_backprop_dim_sizes[3]; in Compute() local
123 const int input_w, const int output_h, const int output_w, CpuKernelContext &ctx) { in FractionalMaxPoolGradWithFixedKsizeCompute()
167 const int c_size, const int input_h, const int input_w, const int output_h, const int output_w) { in ComputeSingleBatch()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/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-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/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

1234