Home
last modified time | relevance | path

Searched defs:kT (Results 1 – 16 of 16) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DAveragePool3d.cpp32 const int kT = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_META_FUNC() local
101 const int kT = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_META_FUNC() local
166 int kT, in avg_pool3d_out_frame()
264 const int kT = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_IMPL_FUNC() local
352 int kT, in avg_pool3d_backward_out_frame()
437 const int kT = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_IMPL_FUNC() local
Dvol2col.h17 const int64_t kT, in vol2col()
68 const int64_t kT, in col2vol()
DPool.h291 int kT, int kH, int kW, in max_pool3d_backward_shape_check()
327 int kT, int kH, int kW, in avg_pool3d_backward_shape_check()
DAdaptiveAveragePooling3d.cpp47 auto kT = iendT - istartT; in adaptive_avg_pool3d_out_frame() local
198 auto kT = iendT - istartT; in adaptive_avg_pool3d_backward_out_frame() local
DAdaptiveMaxPooling3d.cpp108 int64_t kT = iendT - istartT; in adaptive_max_pool3d_single_out_frame() local
/external/pytorch/aten/src/ATen/native/cuda/
DAveragePool3d.cu39 int kT, int kH, int kW, in avg_pool3d_cuda_update_output()
108 int kT, int kH, in avg_pool3d_cuda_update_output()
174 int kT, int kH, int kW, in avg_pool3d_single_backward_out_frame_stride1()
219 int kT, int kH, int kW, in avg_pool3d_cuda_update_grad_input_atomic()
278 int kT, int kH, int kW, in avg_pool3d_cuda_update_grad_input()
362 const int kT = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_IMPL_FUNC() local
468 const int kT = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_IMPL_FUNC() local
DAdaptiveAveragePooling3d.cu78 int kT = iendT - istartT; in adaptiveaveragepool() local
200 int kT = end_index(ot, osizeT, isizeT) - start_index(ot, osizeT, isizeT); in adaptiveaveragegradinput() local
276 int kT = iendT - istartT; in atomicadaptiveaveragegradinput() local
DDepthwiseConv3d.cu37 const int kT = kKnownKernelT > 0 ? kKnownKernelT : kernel.size(2); in conv_depthwise3d_cuda_kernel() local
108 const int kT = kKnownKernelT > 0 ? kKnownKernelT : kernel.size(2); in conv_depthwise3d_cuda_backward_input_kernel() local
190 const int kT = grad_kernel.size(2); in conv_depthwise3d_cuda_backward_weight_kernel() local
DDilatedMaxPool3d.cu41 int kT, int kH, int kW, in max_pool3d_with_indices_single_out_frame()
140 int kT, int kH, int kW, in max_pool3d_with_indices_out_frame()
DAdaptiveMaxPooling3d.cu73 int kT = iendT - istartT; in adaptivemaxpool() local
/external/pdfium/constants/
Dform_fields.cpp14 const char kT[] = "T"; variable
/external/skia/gm/
Dtexelsubset.cpp173 static constexpr SkVector kT = {-100, 300}; in onDraw() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/
DPooling.cpp112 int64_t kT, in spatial_dilated_max_pooling3d()
353 int64_t kT, in q_maxpool_3d()
/external/skia/tools/viewer/
DPatchSlide.cpp205 static constexpr SkScalar kT = 40; variable
/external/pytorch/torch/
D_meta_registrations.py3936 kT, argument
/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp1588 int64_t kT, in qmaxpool_3d_nthwc_kernel()