Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dresize_bilinear_impl.cu22 __global__ void ResizeBilinear(const T *input, const int n, const int c, const int input_h, const i… in ResizeBilinear()
52 __global__ void ResizeBilinearGrad(const half *input, const int n, const int c, const int input_h, … in ResizeBilinearGrad()
85 __global__ void ResizeBilinearGrad(const float *input, const int n, const int c, const int input_h,… in ResizeBilinearGrad()
125 void CalResizeBilinear(const T *input, const int n, const int c, const int input_h, const int input… in CalResizeBilinear()
136 void CalResizeBilinearGrad(const half *input, const int n, const int c, const int input_h, const in… in CalResizeBilinearGrad()
149 void CalResizeBilinearGrad(const float *input, const int n, const int c, const int input_h, const i… 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()
97 int input_h = in_shape[1]; in Conv2dInferShape() local
Dpooling_grad_infer.c32 int input_h = input->shape_[1]; in PoolingGradInferShape() local
Ddepthwise_conv2d_infer.c37 int input_h = input->shape_[1]; in DepthwiseConv2dInferShape() local
Dpooling_infer.c41 int input_h = input->shape_[1]; in PoolingInferShape() local
Ddeconv2d_infer.c43 int32_t input_h = GetHeight(input); in Deconv2dInferShape() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dcrop_op.cc31 int32_t input_h = static_cast<int>(input->shape()[0]); in Compute() local
Dresize_with_bbox_op.cc34 int32_t input_h = input[0]->shape()[0]; in Compute() local
Dresize_op.cc35 int32_t input_h = static_cast<int>(input->shape()[0]); in Compute() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/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
Dpack_ext.c47 int input_h = block_start / output_w * stride_h; in RollingIm2ColPackDwUnitFp32() local
98 int input_h = block_start / output_w * stride_h; in rolling_im2col_hwc() local
118 int input_h = block_start / output_w * stride_h; in rolling_im2col_hwc() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/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/ccsrc/minddata/dataset/kernels/image/soft_dvpp/
Dsoft_dvpp_decode_resize_jpeg_op.cc41 int input_h = 0; in Compute() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.cc31 int64_t input_h = input_shape.at(kShapeH); in ApproximateFLOPs() local
Dmulti_conv_info.cc281 int64_t input_h = input_shape.at(kAxisH); in AdJustConvPrim() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dpooling_fp16_grad.cc54 int input_h = in_shape.at(1); in ReSize() local
Dconvolution_fp16_grad_filter.cc173 int input_h = ci / out_w * conv_param->stride_h_; in DoExecute() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dpooling_grad.cc54 int input_h = in_shape.at(kNumShapeDim2); in ReSize() local
Dconvolution_grad_filter.cc180 int input_h = ci / out_w * conv_param->stride_h_; in Execute() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc84 int input_h = conv_param_->input_h_; 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.cc32 …<int64_t> GetSplitPadList(const std::shared_ptr<ops::Conv2DFusion> &ori_conv_prim, int64_t input_h, in GetSplitPadList()
130 int64_t input_h = in_shape.at(kAxisH); in CalSplitInShape() local
/third_party/libevdev/
Dmeson.build41 input_h = join_paths(meson.source_root(), 'include', 'linux', host_machine.system(), 'input.h') variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_gru_v2_grad_fission.cc219 auto input_h = dynamic_gru_v2_grad_cnode->input(input_index["h"]); in AddHSplitNode() local
471 auto input_h = ori_inputs[input_index["h"]]; in Process() local

12