Searched defs:nInputPlane (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ConvolutionMM2d.cu | 80 int64_t nInputPlane = w_sizes[1]; in slow_conv2d_shape_check() local
143 int64_t nInputPlane = in_sizes[dimf]; in slow_conv2d_forward() local
231 int nInputPlane = weight_sizes[1]/(kW*kH); in slow_conv2d_backward() local
309 int64_t nInputPlane = input_sizes[1]; in slow_conv2d_grad_weight() local
|
| H A D | AveragePool2d.cu | 270 const int64_t nInputPlane = input_.size(-3); in TORCH_IMPL_FUNC() local
389 const int64_t nInputPlane = input.size(-3); in TORCH_IMPL_FUNC() local
|
| H A D | DilatedMaxPool2d.cu | 330 const int64_t nInputPlane = input_.size(-3); in TORCH_IMPL_FUNC() local
458 const int64_t nInputPlane = input.size(-3); in TORCH_IMPL_FUNC() local
|
| H A D | NaiveDilatedConvolution.cu | 202 int64_t nInputPlane = weight.size(1); in slow_conv_dilated_all_cuda_template() local
|
| H A D | MaxUnpooling.cu | 429 int64_t nInputCols, nInputRows, nInputPlane; in max_unpooling2d_backward_out_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AveragePool2d.cpp | 38 int64_t nInputPlane, in avg_pool2d_out_frame()
167 const int64_t nInputPlane = input_.size(-3); in get_output_shape() local
195 const int64_t nInputPlane = input.size(-3); in q_avg_pool2d() local
|
| H A D | AveragePool3d.cpp | 79 const int64_t nInputPlane = input_.size(-4); in get_output_shape() local
109 const int64_t nInputPlane = input.size(-4); in q_avg_pool3d() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Pool.h | 124 int64_t nInputPlane, in pool2d_shape_check()
175 int64_t nInputPlane, in max_pool2d_backward_shape_check()
203 int64_t nInputPlane, in avg_pool2d_backward_shape_check()
|
| H A D | AveragePool2d.cpp | 45 const int64_t nInputPlane = input.size(-3); in TORCH_PRECOMPUTE_META_FUNC() local
127 const int64_t nInputPlane = input.size(-3); // number of channels (or colors) in TORCH_META_FUNC() local
|
| H A D | DilatedMaxPool2d.cpp | 62 const int64_t nInputPlane = input.size(-3); in TORCH_META_FUNC() local
137 const int64_t nInputPlane = input.size(-3); in TORCH_META_FUNC() local
|
| H A D | NaiveDilatedConvolution.cpp | 194 int64_t nInputPlane = weight.size(1); in slow_conv_dilated_all_cpu_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 1999 int64_t nInputPlane,
2116 int64_t nInputPlane,
2159 int64_t nInputPlane,
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2514 nInputPlane, argument
3786 nInputPlane, argument
|