| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DilatedMaxPool2d.cpp | 47 const int dilationW = dilation.size() == 1 ? dilationH : safe_downcast<int, int64_t>(dilation[1]); in TORCH_META_FUNC() local
120 const int dilationW = dilation.size() == 1 ? dilationH : safe_downcast<int, int64_t>(dilation[1]); in TORCH_META_FUNC() local
184 const int dilationW = dilation.size() == 1 ? dilationH : safe_downcast<int, int64_t>(dilation[1]); in TORCH_IMPL_FUNC() local
|
| H A D | Pool.h | 123 …t64_t kW, int64_t dH, int64_t dW, int64_t padH, int64_t padW, int64_t dilationH, int64_t dilationW, in pool2d_shape_check()
174 int kH, int kW, int dH, int dW, int padH, int padW, int dilationH, int dilationW, in max_pool2d_backward_shape_check()
294 int dilationT, int dilationH, int dilationW, in max_pool3d_backward_shape_check()
|
| H A D | vol2col.h | 28 const int64_t dilationW, in vol2col()
79 const int64_t dilationW, in col2vol()
|
| H A D | DilatedMaxPool3d.cpp | 56 const int dilationW = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[2]); in max_pool3d_with_indices_out_cpu_template() local
148 const int dilationW = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[2]); in max_pool3d_with_indices_backward_out_cpu_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | MaxPoolKernel.cpp | 83 int64_t dilationW) { in compute_internal()
158 int64_t dilationW) { in compute_internal()
253 int dilationW = dilWHD[0]; in cpu_max_pool() local
382 int dilationW = dilWHD[0]; in cpu_max_pool_channels_last() local
606 int dilationW, int dilationH) { in max_pool2d_kernel_impl()
632 int dilationW, int dilationH, int dilationD) { in max_pool3d_kernel_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | DilatedMaxPool3d.cu | 44 int dilationT, int dilationH, int dilationW, in max_pool3d_with_indices_single_out_frame()
143 int dilationT, int dilationH, int dilationW, in max_pool3d_with_indices_out_frame()
325 const int dilationW = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[2]); in max_pool3d_with_indices_out_cuda_template() local
455 const int dilationW = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[2]); in max_pool3d_with_indices_backward_out_cuda_template() local
|
| H A D | DepthwiseConv2d.cu | 368 const int dilationW, const int dilationH) { in conv_depthwise2d_forward_out()
461 const int dilationW, const int dilationH) { in conv_depthwise2d_backward_out()
592 const int dilationW, const int dilationH) { in conv_depthwise2d_grad_weight_out()
|
| H A D | DepthwiseConv3d.cu | 51 const int dilationW = kKnownDilationW > 0 ? kKnownDilationW : dilationW_; in conv_depthwise3d_cuda_kernel() local
122 const int dilationW = kKnownDilationW > 0 ? kKnownDilationW : dilationW_; in conv_depthwise3d_cuda_backward_input_kernel() local
188 int dilationT, int dilationH, int dilationW) { in conv_depthwise3d_cuda_backward_weight_kernel()
|
| H A D | DilatedMaxPool2d.cu | 325 const int dilationW = dilation.size() == 1 ? dilationH : safe_downcast<int, int64_t>(dilation[1]); in TORCH_IMPL_FUNC() local
451 const int dilationW = dilation.size() == 1 ? dilationH : safe_downcast<int, int64_t>(dilation[1]); in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cudnn_support_utils.cc | 99 const int64_t dilationW = conv.window().dimensions()[0].base_dilation(); in CudnnSupportsOptimizedIntegerConvolution() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/operation_converters/ |
| D | DepthwiseConv2DOperationConverter.cpp | 98 int32_t dilationW = 1; in convert() local
|
| D | Conv2DOperationConverter.cpp | 212 int32_t dilationW = 1; in convert() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Pooling.cpp | 153 int64_t dilationW = dilation[1]; in quantized_max_pool2d_cudnn() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | Pooling.cpp | 544 int64_t dilationW = dilation[1]; in qnnpack_maxpool2d() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3785 dilationW, argument
3947 dilationW, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions.cpp | 383 int64_t dilationW = extra_args[5]; in nnc_aten_conv2d() local
|