Home
last modified time | relevance | path

Searched defs:dilationT (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDilatedMaxPool3d.cu44 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()
323 const int dilationT = safe_downcast<int, int64_t>(dilation[0]); in max_pool3d_with_indices_out_cuda_template() local
453 const int dilationT = safe_downcast<int, int64_t>(dilation[0]); in max_pool3d_with_indices_backward_out_cuda_template() local
H A DDepthwiseConv3d.cu49 const int dilationT = kKnownDilationT > 0 ? kKnownDilationT : dilationT_; in conv_depthwise3d_cuda_kernel() local
120 const int dilationT = kKnownDilationT > 0 ? kKnownDilationT : dilationT_; in conv_depthwise3d_cuda_backward_input_kernel() local
188 int dilationT, int dilationH, int dilationW) { in conv_depthwise3d_cuda_backward_weight_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dvol2col.h26 const int64_t dilationT, in vol2col()
77 const int64_t dilationT, in col2vol()
H A DDilatedMaxPool3d.cpp54 const int dilationT = safe_downcast<int, int64_t>(dilation[0]); in max_pool3d_with_indices_out_cpu_template() local
146 const int dilationT = safe_downcast<int, int64_t>(dilation[0]); in max_pool3d_with_indices_backward_out_cpu_template() local
H A DPool.h294 int dilationT, int dilationH, int dilationW, in max_pool3d_backward_shape_check()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py3945 dilationT, argument