Home
last modified time | relevance | path

Searched defs:input_h (Results 1 – 25 of 68) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/ascend910b/
Ddvpp_normalize_v2_op.cc59 int64_t input_h = input->GetShape()[kHeightIndexNCHW]; in Compute() local
73 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_crop_op.cc48 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
71 int32_t input_h = inputs[0][kHeightIndex]; in OutputShape() local
Ddvpp_resize_op.cc57 int32_t input_h = size[kHeightIndex]; in Compute() local
120 int32_t input_h = inputs[0][kHeightIndex]; in OutputShape() local
Ddvpp_pad_op.cc70 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
96 int32_t input_h = inputs[0][kHeightIndex]; in OutputShape() local
Ddvpp_convert_color_op.cc47 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_equalize_op.cc47 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_perspective_op.cc61 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_erase_op.cc50 int32_t input_h = size[kHeightIndex]; in Compute() local
Ddvpp_auto_contrast_op.cc50 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_adjust_hue_op.cc46 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_invert_op.cc46 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_solarize_op.cc46 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_adjust_brightness_op.cc46 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; in Compute() local
Ddvpp_vertical_flip_op.cc46 int64_t input_h = input->GetShape()[kHeightIndexNHWC]; 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()
62 int input_h = input->shape_[1]; in PoolingInferShape() local
Dformat_transpose_infer.c44 int input_h = GetHeight(input); in FormatTransposeInferShape() local
Dpooling_grad_infer.c32 int input_h = input->shape_[1]; in PoolingGradInferShape() local
Ddepthwise_conv2d_infer.c38 int input_h = input->shape_[1]; in DepthwiseConv2dInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dresize_bilinear_impl.cu34 __global__ void ResizeBilinear(const T *input, const int n, const int c, const int input_h, const i… in ResizeBilinear()
64 __global__ void ResizeBilinear_HPC(const T *input, const int n, const int c, const int input_h, con… in ResizeBilinear_HPC()
94 …lobal__ void ResizeBilinearGradHalf(const half *input, const int n, const int c, const int input_h, in ResizeBilinearGradHalf()
128 __global__ void ResizeBilinearGrad(const T *input, const int n, const int c, const int input_h, con… in ResizeBilinearGrad()
161 …l__ void ResizeBilinearGradHalf_HPC(const half *input, const int n, const int c, const int input_h, in ResizeBilinearGradHalf_HPC()
195 __global__ void ResizeBilinearGrad_HPC(const T *input, const int n, const int c, const int input_h,… in ResizeBilinearGrad_HPC()
235 cudaError_t CalResizeBilinear(const T *input, const int n, const int c, const int input_h, const in… in CalResizeBilinear()
252 …ResizeBilinearGradHalf(const half *input, const int n, const int c, const int input_h, const int i… in CalResizeBilinearGradHalf()
273 cudaError_t CalResizeBilinearGrad(const T *input, const int n, const int c, const int input_h, cons… in CalResizeBilinearGrad()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dcrop_op.cc30 auto input_h = static_cast<int>(input->shape()[0]); in Compute() local
Dresize_with_bbox_op.cc34 auto input_h = static_cast<int32_t>(input[0]->shape()[0]); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dfractional_max_pool_grad_with_fixed_ksize.cc80 int64_t input_h = out_backprop_dim_sizes[2]; in Compute() local
122 …op, Tensor *argmax, const int64_t data_nums, const int n_size, const int c_size, const int input_h, 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.c59 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad32Arm() local
132 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad16Arm() local
189 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad8Arm() local
238 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad4Arm() local
287 int input_h = idx / out_w * conv_param->stride_h_; in FilterGradLeftoverArm() local
346 int input_h = idx / out_w * conv_param->stride_h_; in ConvDwFilterFp16Grad() local
Dpack_fp16_ext.c46 int input_h = block_start / output_w * stride_h; in RollingIm2ColPackDwUnitFp16() local
97 int input_h = block_start / output_w * stride_h; in RollingIm2ColPackUnitFp16() local
117 int input_h = block_start / output_w * stride_h; in RollingIm2ColPackUnitFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dconvolution_grad_filter.c52 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad16Arm() local
129 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad12Arm() local
199 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad8Arm() local
257 int input_h = idx / out_w * conv_param->stride_h_; in FilterGrad4Arm() local
307 int input_h = idx / out_w * conv_param->stride_h_; in Filtergrad2Arm() local
365 int input_h = idx / out_w * conv_param->stride_h_; in ConvDwFilterGrad() local

123