Searched defs:iH (Results 1 – 6 of 6) sorted by relevance
48 int64_t iH, in spatial_dilated_max_pooling()107 int64_t iH, in spatial_dilated_max_pooling3d()226 int64_t iH = qx.size(dimh); in q_maxpool_2d() local392 int64_t iH = qx.size(dimh); in q_maxpool_3d() local
46 const int iH = input.size(3); in conv_depthwise3d_cuda_kernel() local117 const int iH = grad_input.size(3); in conv_depthwise3d_cuda_backward_input_kernel() local207 const int iH = input.size(3); in conv_depthwise3d_cuda_backward_weight_kernel() local
241 int iH = (height + gridDim.z-1) / gridDim.z; in C10_LAUNCH_BOUNDS_1() local
395 index_t iH = (isizeH + gridDim.z-1) / gridDim.z; in C10_LAUNCH_BOUNDS_1() local
1455 int64_t iH, in _qmaxpool_2d_nhwc_kernel()1555 int64_t iH, in qmaxpool_2d_nhwc_kernel()1583 int64_t iH, in qmaxpool_3d_nthwc_kernel()
2105 String iH = i.shallowClone().toString(); in testShallowCloneToString() local