Searched defs:oW (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | MaxUnpooling.cpp | 120 int64_t oW = output_size[2]; in max_unpooling3d_shape_check() local
180 int64_t oW = output_size[2]; in max_unpooling3d_forward_out_cpu() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | MaxUnpooling.cu | 59 const int64_t oW, in max_unpooling3d_forward_kernel()
99 int64_t oW, in max_unpooling3d_backward_kernel()
222 int64_t oW = output_size[2]; in max_unpooling3d_shape_check() local
307 int64_t oW = output_size[2]; in max_unpooling3d_forward_out_cuda() local
510 int64_t oW = output_size[2]; in max_unpooling3d_backward_out_cuda() local
|
| D | DepthwiseConv3d.cu | 43 const int oW = output.size(4); in conv_depthwise3d_cuda_kernel() local
114 const int oW = grad_output.size(4); in conv_depthwise3d_cuda_backward_input_kernel() local
205 const int oW = grad_output.size(4); in conv_depthwise3d_cuda_backward_weight_kernel() local
|
| D | DilatedMaxPool2d.cu | 123 int oW = (pooled_width + gridDim.y-1) / gridDim.y; in C10_LAUNCH_BOUNDS_1() local
|
| D | AdaptiveAveragePooling.cu | 271 index_t oW = (osizeW + gridDim.y-1) / gridDim.y; in C10_LAUNCH_BOUNDS_1() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | Pooling.cpp | 51 int64_t oW, // output sizes in spatial_dilated_max_pooling()
111 int64_t oW, // output sizes in spatial_dilated_max_pooling3d()
239 int64_t oW = pooling_output_shape(iW, kW, pW, sW, dW, ceil_mode); in q_maxpool_2d() local
403 int64_t oW = pooling_output_shape(iW, kW, pW, sW, dW, ceil_mode); in q_maxpool_3d() local
|
| D | AveragePool2d.cpp | 289 const int64_t oW = output_shape[output_shape.size() - 1]; in qnnpack_avg_pool2d() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 1458 int64_t oW, // output sizes in _qmaxpool_2d_nhwc_kernel()
1558 int64_t oW, // output sizes in qmaxpool_2d_nhwc_kernel()
1587 int64_t oW, // output sizes in qmaxpool_3d_nthwc_kernel()
|