Home
last modified time | relevance | path

Searched defs:input_height (Results 1 – 25 of 44) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dadaptive_avg_pool2d_impl.cu29 __global__ void AdaptiveAvgPool2DKernel(const uint size, const uint input_height, const uint input_… in AdaptiveAvgPool2DKernel()
60 __global__ void AdaptiveAvgPool2DKernel(const uint size, const uint input_height, const uint input_… in AdaptiveAvgPool2DKernel()
91 __global__ void AdaptiveAvgPool2DKernel(const uint size, const uint input_height, const uint input_… in AdaptiveAvgPool2DKernel()
122 __global__ void AdaptiveAvgPool2DKernel(const uint size, const uint input_height, const uint input_… in AdaptiveAvgPool2DKernel()
153 cudaError_t ApplyAdaptiveAvgPool2D(const uint size, const uint input_height, const uint input_width, in ApplyAdaptiveAvgPool2D()
Dadaptive_avg_pool2d_grad_impl.cu29 __global__ void AdaptiveAvgPool2DGradKernel(const uint size, const uint input_height, const uint in… in AdaptiveAvgPool2DGradKernel()
63 __global__ void AdaptiveAvgPool2DGradKernel(const uint size, const uint input_height, const uint in… in AdaptiveAvgPool2DGradKernel()
97 __global__ void AdaptiveAvgPool2DGradKernel(const uint size, const uint input_height, const uint in… in AdaptiveAvgPool2DGradKernel()
137 __global__ void AdaptiveAvgPool2DGradKernel(const uint size, const uint input_height, const uint in… in AdaptiveAvgPool2DGradKernel()
171 cudaError_t ApplyAdaptiveAvgPool2DGrad(const uint size, const uint input_height, const uint input_w… in ApplyAdaptiveAvgPool2DGrad()
Dadaptive_avg_pool3d_impl.cu21 …oid AdaptiveAvgPool3DKernel(const uint out_size, const uint input_channel, const uint input_height, in AdaptiveAvgPool3DKernel()
62 …or_t ApplyAdaptiveAvgPool3D(const uint out_size, const uint input_channel, const uint input_height, in ApplyAdaptiveAvgPool3D()
Dadaptive_max_pool2d_impl.cu40 __global__ void AdaptiveMaxPool2DKernel(const size_t size, const size_t input_height, const size_t … in AdaptiveMaxPool2DKernel()
78 cudaError_t ApplyAdaptiveMaxPool2D(const size_t size, const size_t input_height, const size_t input… in ApplyAdaptiveMaxPool2D()
Dadaptive_avg_pool3d_grad_impl.cu32const uint input_height, const uint input_width, const uint input_depth, in AdaptiveAvgPool3DGradKernel()
72const uint input_height, const uint input_width, const uint input_depth, in ApplyAdaptiveAvgPool3DGrad()
Dcrop_and_resize_impl.cu26 … int batch, int input_height, int input_width, int final_height, int final_width, in CropAndResize()
107 … int batch, int input_height, int input_width, int final_height, int final_width, in CalCropAndResize()
Dadaptive_max_pool3d_impl.cu38 const int64_t input_height, const int64_t input_width, in AdaptiveMaxPool3DKernel()
83const int64_t input_height, const int64_t input_width, const T *input_data, in ApplyAdaptiveMaxPool3D()
Dadaptive_max_pool2d_grad_impl.cu34const uint input_height, const uint input_width, const uint output_height, in CalAdaptiveMaxPool2DGrad()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dlite_image_utils.cc371 int input_height = static_cast<int>(input->shape()[0]); in Crop() local
439 int input_height = static_cast<int>(input->shape()[0]); in Normalize() local
621 int input_height = static_cast<int>(input->shape()[0]); in ResizePreserve() local
667 int input_height = static_cast<int>(input->shape()[0]); in RgbToBgr() local
704 int input_height = static_cast<int>(input->shape()[0]); in RgbToGray() local
749 int input_height = static_cast<int>(input->shape()[0]); in Pad() local
796 int input_height = static_cast<int>(input->shape()[0]); in RotateAngleWithOutMirror() local
863 int input_height = static_cast<int>(input->shape()[0]); in RotateAngleWithMirror() local
1021 int input_height = static_cast<int>(input->shape()[0]); in Affine() local
1057 int input_height = static_cast<int>(input->shape()[0]); in GaussianBlur() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dupsample_trilinear3d.h40 int64_t input_height; variable
Dupsample_nearest3d_grad.h38 int64_t input_height; variable
Dim2col.h49 int64_t input_height; variable
Dupsample_trilinear3d_grad.h69 int64_t input_height; variable
Dscale_and_translate.cc372 const int64_t input_height = input0_shape->GetDimSize(1); in ScaleAndTranslateCompute() local
498 const T *image, const int64_t input_height, const int64_t input_width, in GatherColumns()
538 …const T *image, const int64_t input_height, const int64_t input_width, const int64_t output_height, in GatherRows()
572 const int64_t input_height = images.dimension(1); in operator ()() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dmax_unpool2d_cpu_kernel.cc86 size_t input_height = LongToSize(input_shape_[kInputIndex1]); in LaunchKernel() local
119 size_t input_height = LongToSize(input_shape_[kInputIndex2]); in LaunchKernel() local
Dmax_unpool3d_cpu_kernel.cc90 size_t input_height = LongToSize(input_shape_[kInputIndex2]); in LaunchKernel() local
128 size_t input_height = LongToSize(input_shape_[kInputIndex3]); in LaunchKernel() local
Dscale_and_translate_cpu_kernel.cc85 … const T *image, const int64_t input_height, const int64_t input_width, in GatherRows()
111 … const T *image, const int64_t input_height, const int64_t input_width, in GatherColumns()
150 const int64_t input_height = images.dimension(1); in GatherSpans() local
376 const int64_t input_height = input0_shape_[1]; in LaunchKernel() local
Ddilation2d_cpu_kernel.cc81 int64_t input_height = input_shape_[kFormatNCHWIndexH]; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dspace_to_batch_fp32.c24 const int input_height = param->input_shape_[1]; in DoSpaceToBatch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dmaxpool_grad_grad.c22 const int input_height = args->input_h_; in MaxPoolGradGrad() local
81 const int input_height = args->input_h_; in MaxPool3DGradGrad() local
/third_party/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_ml_tp.c297 unsigned input_height = operation->input_height; in create_detranspose_config() local
420 unsigned input_height = operation->input_height; in create_reshuffle_config() local
573 unsigned input_height = operation->input_height; in create_pad_config() local
/third_party/ffmpeg/libavfilter/dnn/
Ddnn_backend_common.c162 …askItem *task, DNNExecBaseParams *exec_params, void *backend_model, int input_height, int input_wi… in ff_dnn_fill_gettingoutput_task()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dim2col_and_col2im_ext.cc132 … const std::vector<int64_t> &stride, int64_t input_height, int64_t output_height, in Im2ColOutputLengthError()
205 auto input_height = input_shape[input_rank - kIndex2]; in InferShape() local
250 auto input_height = input_shape[input_rank - kIndex2]; in InferShape() local
/third_party/ffmpeg/libavfilter/
Ddnn_filter_common.c114 int ff_dnn_get_output(DnnContext *ctx, int input_width, int input_height, int *output_width, int *o… in ff_dnn_get_output()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dconvolution.cc144 int input_height = ci / out_w * conv_param_->stride_h_; in DoExecute() local

12