| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DilatedMaxPool2d.cpp | 46 const int dilationH = safe_downcast<int, int64_t>(dilation[0]); in TORCH_META_FUNC() local
119 const int dilationH = safe_downcast<int, int64_t>(dilation[0]); in TORCH_META_FUNC() local
183 const int dilationH = safe_downcast<int, int64_t>(dilation[0]); in TORCH_IMPL_FUNC() local
|
| H A D | Pool.h | 123 …int64_t kH, int64_t kW, int64_t dH, int64_t dW, int64_t padH, int64_t padW, int64_t dilationH, int… 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 | 27 const int64_t dilationH, in vol2col()
78 const int64_t dilationH, in col2vol()
|
| H A D | DilatedMaxPool3d.cpp | 55 const int dilationH = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[1]); in max_pool3d_with_indices_out_cpu_template() local
147 const int dilationH = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[1]); 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 | 82 int64_t dilationH, in compute_internal()
157 int64_t dilationH, in compute_internal()
254 int dilationH = dilWHD[1]; in cpu_max_pool() local
383 int dilationH = dilWHD[1]; 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()
324 const int dilationH = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[1]); in max_pool3d_with_indices_out_cuda_template() local
454 const int dilationH = dilation.size() == 1 ? dilationT : safe_downcast<int, int64_t>(dilation[1]); 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 | 50 const int dilationH = kKnownDilationH > 0 ? kKnownDilationH : dilationH_; in conv_depthwise3d_cuda_kernel() local
121 const int dilationH = kKnownDilationH > 0 ? kKnownDilationH : dilationH_; 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 | 324 const int dilationH = safe_downcast<int, int64_t>(dilation[0]); in TORCH_IMPL_FUNC() local
450 const int dilationH = safe_downcast<int, int64_t>(dilation[0]); in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cudnn_support_utils.cc | 100 const int64_t dilationH = conv.window().dimensions()[1].base_dilation(); in CudnnSupportsOptimizedIntegerConvolution() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/operation_converters/ |
| D | DepthwiseConv2DOperationConverter.cpp | 99 int32_t dilationH = 1; in convert() local
|
| D | Conv2DOperationConverter.cpp | 213 int32_t dilationH = 1; in convert() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Pooling.cpp | 152 int64_t dilationH = dilation[0]; in quantized_max_pool2d_cudnn() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | Pooling.cpp | 543 int64_t dilationH = dilation[0]; in qnnpack_maxpool2d() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3784 dilationH, argument
3946 dilationH, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions.cpp | 382 int64_t dilationH = extra_args[4]; in nnc_aten_conv2d() local
|