Home
last modified time | relevance | path

Searched defs:output_h (Results 1 – 25 of 45) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/ascend910b/
Ddvpp_resize_op.cc59 int32_t output_h; in Compute() local
122 int32_t output_h = outputs[0][kHeightIndex]; in OutputShape() local
130 TensorShape DvppResizeOp::ComputeOutputShape(const TensorShape &input, int32_t output_h, int32_t ou… in ComputeOutputShape()
Ddvpp_pad_op.cc72 int64_t output_h = input_h + pad_top_ + pad_bottom_; in Compute() local
100 int32_t output_h = input_h + pad_top_ + pad_bottom_; in OutputShape() local
Ddvpp_resized_crop_op.cc66 int32_t output_h; in Compute() local
145 TensorShape DvppResizedCropOp::ComputeOutputShape(const TensorShape &input, int32_t output_h, int32… in ComputeOutputShape()
Ddvpp_crop_op.cc73 int32_t output_h = height_; in OutputShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dresized_crop_op.cc43 int32_t output_h; in Compute() local
71 int32_t output_h = 0; in OutputShape() local
Dresize_op.cc40 int32_t output_h; in Compute() local
113 TensorShape ResizeOp::ComputeOutputShape(const TensorShape &input, int32_t output_h, int32_t output… in ComputeOutputShape()
Dresize_with_bbox_op.cc44 …auto output_h = static_cast<int32_t>((*output)[0]->shape()[0]); // output height if ResizeWithBBox 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()
71 int output_h = 0; in PoolingInferShape() local
Dconv2d_infer.c20 int ConvInferShape(int input_h, int input_w, int *output_h, int *output_w, ConvParameter *param) { in ConvInferShape()
152 int output_w = 0, output_h = 0; in Conv2dInferShape() local
Dpooling_grad_infer.c52 int output_h = ceil((float)(input_h) / (float)(param->stride_h_)); in PoolingGradInferShape() local
Ddepthwise_conv2d_infer.c41 int output_w = 0, output_h = 0; in DepthwiseConv2dInferShape() local
Ddeconv2d_infer.c48 int32_t output_h = 0; in Deconv2dInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dresize_bilinear_impl.cu35 … const int output_h, const int output_w, const int nchw, const int chw, const int hw, in ResizeBilinear()
65 … const int output_h, const int output_w, const int nchw, const int chw, const int hw, in ResizeBilinear_HPC()
95 … const int input_w, const int output_h, const int output_w, const int nchw, in ResizeBilinearGradHalf()
129 … const int output_h, const int output_w, const int nchw, const int chw, const int hw, in ResizeBilinearGrad()
162 … const int input_w, const int output_h, const int output_w, const int nchw, in ResizeBilinearGradHalf_HPC()
196 … const int output_h, const int output_w, const int nchw, const int chw, in ResizeBilinearGrad_HPC()
236 … const int output_h, const int output_w, const float h_scale, const float w_scale, in CalResizeBilinear()
253 … const int output_h, const int output_w, const float h_scale, const float w_scale, in CalResizeBilinearGradHalf()
274 … const int output_h, const int output_w, const float h_scale, const float w_scale, in CalResizeBilinearGrad()
Dresize_bicubic_impl.cu34 … const int input_w, const int output_h, const int output_w, const int nchw, const int chw, in ResizeBicubic()
97 … const int input_h, const int input_w, const int output_h, in ResizeBicubicHalfPixelCenters()
202 … const int output_h, const int output_w, const float h_scale, const float w_scale, in CalResizeBicubic()
/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.cc94 int64_t output_h = origin_input_dim_sizes[2]; 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()
Dfractional_max_pool_with_fixed_ksize.cc82 int64_t output_h = output_shape_values[0]; in Compute() local
156 …const int output_h, const int output_w, const int pool_h, const int pool_w, CpuKernelContext &ctx)… in FractionalMaxPoolWithFixedKsizeCompute()
207 … const int input_w, const int output_h, const int output_w, in ComputeSingleBatch()
Dpad_v3_grad.h38 int64_t output_h; variable
Dpad_v3.cc138 int64_t output_h = input_h + pad_t + paddings[kNum3]; in EdgeCompute2D() local
166 int64_t output_h = output_shape[kNum3]; in EdgeCompute3D() local
243 int64_t output_h = input_h + pad_t + paddings[kNum3]; in ReflectCompute2D() local
271 int64_t output_h = output_shape[kNum3]; in ReflectCompute3D() local
336 int64_t output_h = input_h + pad_t + paddings[kNum3]; in CircularCompute2D() local
366 int64_t output_h = output_shape[kNum3]; in CircularCompute3D() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/
Drandom_resize_op_test.cc35 int output_h = 0.5 * s[0]; in TEST_F() local
Dresize_op_test.cc37 int output_h = (s[0] * output_w) / s[1]; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dpooling_fp32.c27 int output_w = pooling_args->output_w_, output_h = pooling_args->output_h_; in AvgPoolingBatch() local
86 int output_h = pooling_args->output_h_; in AvgPooling() local
105 int output_w = pooling_args->output_w_, output_h = pooling_args->output_h_; in AvgPoolingFromNC4HW4ToNHWCLessC() local
189 int output_w = pooling_args->output_w_, output_h = pooling_args->output_h_; in AvgPoolingFromNC4HW4ToNHWCBatch() local
318 int output_h = pooling_args->output_h_; in AvgPoolingFromNC4HW4ToNHWC() local
342 int output_w = pooling_args->output_w_, output_h = pooling_args->output_h_; in MaxPoolingBatch() local
395 int output_h = pooling_args->output_h_; in MaxPooling() local
414 int output_w = pooling_args->output_w_, output_h = pooling_args->output_h_; in MaxPoolingFromNC4HW4ToNHWCLessC() local
495 int output_w = pooling_args->output_w_, output_h = pooling_args->output_h_; in MaxPoolingFromNC4HW4ToNHWCBatch() local
617 int output_h = pooling_args->output_h_; in MaxPoolingFromNC4HW4ToNHWC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dspace_to_depth_base.c26 int output_h = out_shape[kNHWC_H]; in SpaceToDepthForNHWC() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc87 int output_h = conv_param_->output_h_; in InitTmpBuffer() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dp_s_r_o_i_pooling_grad.cc69 int64_t output_h = input_size[0]; in PSROIPoolingGradInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dpooling_grad.c34 int output_h = pooling_args->output_h_; in AvgPoolingGrad() local
117 int output_h = pooling_args->output_h_; in MaxPoolingGrad() local

12