Home
last modified time | relevance | path

Searched defs:output_w (Results 1 – 25 of 27) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dresize_bilinear_impl.cu23 …const int output_h, const int output_w, const int nchw, const int chw, const int hw, const float h… in ResizeBilinear()
53 …const int output_h, const int output_w, const int nchw, const int chw, const int hw, const float h… in ResizeBilinearGrad()
86 …const int output_h, const int output_w, const int nchw, const int chw, const int hw, const float h… in ResizeBilinearGrad()
126 const int output_h, const int output_w, const float h_scale, const float w_scale, T *output, in CalResizeBilinear()
137 …const int output_h, const int output_w, const float h_scale, const float w_scale, half *output, fl… in CalResizeBilinearGrad()
150 …const int output_h, const int output_w, const float h_scale, const float w_scale, float *output, f… 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()
100 int output_w = 0, output_h = 0; in Conv2dInferShape() local
Dpooling_grad_infer.c51 int output_w = ceil((float)(input_w) / (float)(param->stride_w_)); in PoolingGradInferShape() local
Ddepthwise_conv2d_infer.c40 int output_w = 0, output_h = 0; in DepthwiseConv2dInferShape() local
Dpooling_infer.c51 int output_w = 0; in PoolingInferShape() local
Ddeconv2d_infer.c48 int32_t output_w = 0; in Deconv2dInferShape() local
/third_party/mindspore/tests/ut/cpp/dataset/
Dresize_op_test.cc35 int output_w = 0.5 * s[0]; in TEST_F() local
Drandom_resize_op_test.cc36 int output_w = 0.5 * s[1]; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dresize_with_bbox_op.cc45 int32_t output_w = (*output)[0]->shape()[1]; // output width if ResizeWithBBox in Compute() local
Dresize_op.cc34 int32_t output_w = 0; in Compute() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c38 const int output_w = conv_param->output_w_; in RollingIm2ColPackDwUnitFp32() local
88 const int output_w = conv_param->output_w_; in rolling_im2col_hwc() local
159 const int output_w = conv_param->input_w_; in rolling_im2row_hwc() local
208 const int output_w = conv_param->output_w_; in col2im_hwc() local
255 const int output_w = conv_param->output_w_; in rolling_col2im_hwc() local
Dpooling_grad.c31 int output_w = pooling_param->output_w_; in AvgPoolingGrad() local
114 int output_w = pooling_param->output_w_; in MaxPoolingGrad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c37 const int output_w = conv_param->output_w_; in RollingIm2ColPackDwUnitFp16() local
87 const int output_w = conv_param->output_w_; in RollingIm2ColPackUnitFp16() local
154 const int output_w = conv_param->output_w_; in RollingCol2ImPackUnitFp16() local
Dpooling_grad.c31 int output_w = pooling_param->output_w_; in AvgPoolingFp16Grad() local
116 int output_w = pooling_param->output_w_; in MaxPoolingFp16Grad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c31 int output_w = pooling_param->output_w_; in AvgPoolingInt8() local
88 int output_w = pooling_param->output_w_; in AvgPoolingOptInt8() local
287 int output_w = pooling_param->output_w_; in MaxPoolingInt8() local
332 int output_w = pooling_param->output_w_; in MaxPoolingWithQuantInt8() local
426 int output_w = pooling_param->output_w_; in MaxPoolingOptInt8() local
Dconv_depthwise_int8.c36 void ConvDwInt8Post(int8_t *dst, int32_t *buffer, int output_w, int channel, int32_t output_zp, in ConvDwInt8Post()
157 … int row_size, int channel, int output_h, int output_w, int8_t in_zp, int32_t out_zp, in ConvDw3x3Int8Window()
210 … int end_c, int col_size, int row_size, int channel, int output_h, int output_w, int8_t in_zp, in ConvDw3x3Int8Block()
Dconv3x3_int8.c455 bool w_not_bound, int output_w, int real_num, int oc_start, in Conv3x3Int8OutputUnit()
751 int output_w = conv_param->output_w_; in Conv3x3Int8OutputTransform() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc86 int output_w = conv_param_->output_w_; in InitTmpBuffer() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpooling_fp32.c29 int output_w = pooling_param->output_w_; in AvgPooling() local
143 int output_w = pooling_param->output_w_; in MaxPooling() local
Dwinograd_transform.c102 int output_w = conv_param->output_w_; in WinogradOutputNHWCTransform() local
161 int output_w = conv_param->output_w_; in WinogradOutputNC4HW4Transform() local
Dconv_1x1_x86_fp32.c27 int output_w = conv_param->output_w_; in Conv1x1SWFp32() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_3x3_int8.cc135 int output_w = conv_param_->output_w_; in InitTmpBuffer() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_transform_fp16.c132 int output_w = conv_param->output_w_; in WinogradOutputNHWCTransformFp16() local
171 int output_w = conv_param->output_w_; in WinogradOutputNC8HW8TransformFp16() local
Dpooling_fp16.c28 int output_w = pooling_param->output_w_; in AvgPoolingFp16() local
264 int output_w = pooling_param->output_w_; in MaxPoolingFp16() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc48 int64_t output_w = static_cast<int64_t>( in GetSplitPadList() local

12