Searched defs:strideH (Results 1 – 7 of 7) sorted by relevance
33 int strideT, int strideH, int strideW, in conv_depthwise3d_cuda_kernel()124 const int strideH = kKnownStrideH > 0 ? kKnownStrideH : strideH_; in conv_depthwise3d_cuda_backward_input_kernel() local194 const int strideH = kKnownStrideH > 0 ? kKnownStrideH : strideH_; in conv_depthwise3d_cuda_backward_weight_kernel() local
237 const int strideH = (stride != 0) ? stride : strideHeight; in C10_LAUNCH_BOUNDS_1() local
79 int32_t strideH = context->getConstantScalar<int32_t>(strideHOperand); in convert() local
194 int32_t strideH = context->getConstantScalar<int32_t>(strideHOperand); in convert() local
144 int64_t strideH = stride[0]; in quantized_max_pool2d_cudnn() local
541 int64_t strideH = stride[0]; in qnnpack_maxpool2d() local
378 int64_t strideH = extra_args[0]; in nnc_aten_conv2d() local