Home
last modified time | relevance | path

Searched defs:kW (Results 1 – 25 of 55) sorted by relevance

123

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAvgPoolKernel.cpp20 int64_t kW, int64_t kH, in cpu_avg_pool2d()
107 int64_t kW, int64_t kH, in cpu_avg_pool2d_channels_last()
221 int64_t kW, int64_t kH, in cpu_avg_pool2d_channels_last()
352 int kW, int kH, in cpu_avg_pool2d_backward()
420 int kW, int kH, in cpu_avg_pool2d_backward_channels_last()
499 int64_t kW, int64_t kH, in avg_pool2d_kernel_impl()
525 int kW, int kH, in avg_pool2d_backward_kernel_impl()
553 int64_t kW, int64_t kH, int64_t kD, in cpu_avg_pool3d()
649 int64_t kW, int64_t kH, int64_t kD, in cpu_avg_pool3d_channels_last()
772 int64_t kW, int64_t kH, int64_t kD, in cpu_avg_pool3d_channels_last()
[all …]
H A DUnfold2d.cpp41 int64_t kW, in unfolded2d_acc()
126 int64_t kW, in unfolded2d_acc_channels_last()
183 int64_t kW, in unfolded2d_acc_kernel()
234 int64_t kW, in unfolded2d_copy()
341 int64_t kW, in unfolded2d_copy_channels_last()
398 int64_t kW, in unfolded2d_copy_kernel()
H A DMaxPoolKernel.cpp247 int kW = kWHD[0]; in cpu_max_pool() local
376 int kW = kWHD[0]; in cpu_max_pool_channels_last() local
603 int kW, int kH, in max_pool2d_kernel_impl()
629 int kW, int kH, int kD, in max_pool3d_kernel_impl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAveragePool3d.cpp34 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in TORCH_META_FUNC() local
103 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in TORCH_META_FUNC() local
167 int kW, in avg_pool3d_out_frame()
266 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in TORCH_IMPL_FUNC() local
353 int kW, in avg_pool3d_backward_out_frame()
439 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in TORCH_IMPL_FUNC() local
H A DPool.h123 …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()
202 int kH, int kW, int dH, int dW, int padH, int padW, in avg_pool2d_backward_shape_check()
291 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()
H A DAveragePool2d.cpp29 const int64_t kW = kernel_size.size() == 1 ? kH : kernel_size[1]; in TORCH_PRECOMPUTE_META_FUNC() local
110 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_META_FUNC() local
190 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool2d.cpp29 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_META_FUNC() local
102 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_META_FUNC() local
174 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool3d.cpp36 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in max_pool3d_with_indices_out_cpu_template() local
128 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in max_pool3d_with_indices_backward_out_cpu_template() local
H A DAdaptiveAveragePooling3d.cpp57 auto kW = iendW - istartW; in adaptive_avg_pool3d_out_frame() local
208 auto kW = iendW - istartW; in adaptive_avg_pool3d_backward_out_frame() local
/aosp_15_r20/external/skia/gm/
H A Dshadowutils.cpp40 static constexpr int kW = 800; variable
226 DEF_SIMPLE_GM(shadow_utils, canvas, kW, kH) { in DEF_SIMPLE_GM() argument
230 DEF_SIMPLE_GM(shadow_utils_occl, canvas, kW, kH) { in DEF_SIMPLE_GM() argument
234 DEF_SIMPLE_GM(shadow_utils_gray, canvas, kW, kH) { in DEF_SIMPLE_GM() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAveragePool3d.cu39 int kT, int kH, int kW, 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()
364 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in TORCH_IMPL_FUNC() local
470 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in TORCH_IMPL_FUNC() local
H A DDepthwiseConv2d.cu129 for (int kW = kWmin; kW < kWmax; ++kW) { in C10_LAUNCH_BOUNDS_1() local
199 for (int kW = 0; kW < KW_LIMIT; ++kW) { in C10_LAUNCH_BOUNDS_1() local
303 int kW = bidx % kernelWidth; in conv_depthwise2d_grad_weight_kernel() local
365 const int kW, const int kH, in conv_depthwise2d_forward_out()
458 const int kW, const int kH, in conv_depthwise2d_backward_out()
589 const int kW, const int kH, in conv_depthwise2d_grad_weight_out()
H A DConvolutionMM2d.cu27 int64_t kH, int64_t kW, in slow_conv2d_shape_check()
130 int64_t kH, int64_t kW, in slow_conv2d_forward()
222 int kH, int kW, in slow_conv2d_backward()
296 int64_t kH, int64_t kW, in slow_conv2d_grad_weight()
H A 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()
305 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in max_pool3d_with_indices_out_cuda_template() local
435 const int kW = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[2]); in max_pool3d_with_indices_backward_out_cuda_template() local
H A DAdaptiveAveragePooling3d.cu94 int kW = iendW - istartW; in adaptiveaveragepool() local
204 int kW = end_index(ow, osizeW, isizeW) - start_index(ow, osizeW, isizeW); in adaptiveaveragegradinput() local
292 int kW = iendW - istartW; in atomicadaptiveaveragegradinput() local
H A DDepthwiseConv3d.cu39 const int kW = kKnownKernelW > 0 ? kKnownKernelW : kernel.size(4); in conv_depthwise3d_cuda_kernel() local
110 const int kW = kKnownKernelW > 0 ? kKnownKernelW : kernel.size(4); in conv_depthwise3d_cuda_backward_input_kernel() local
192 const int kW = grad_kernel.size(4); in conv_depthwise3d_cuda_backward_weight_kernel() local
H A DAdaptiveAveragePooling.cu88 int kW = iendW - istartW; in adaptive_average_pool() local
154 int kW = START_IND(ow, osizeW, isizeW) - END_IND(ow, osizeW, isizeW); in adaptive_average_gradinput() local
203 int kW = iendW - istartW; in atomic_adaptive_average_gradinput() local
H A DAveragePool2d.cu261 const int kW = safe_downcast<int, int64_t>(kW_); in TORCH_IMPL_FUNC() local
376 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool2d.cu315 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
441 const int kW = kernel_size.size() == 1 ? kH : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DAveragePool2d.cpp43 int kW, in avg_pool2d_out_frame()
130 const int kW = kernel_size.size() == 1 in get_kernel() local
136 inline std::pair<int, int> get_stride(IntArrayRef stride, int kW, int kH) { in get_stride()
159 int kW, in get_output_shape()
H A DAveragePool3d.cpp34 const int kW = kernel_size.size() == 1 in get_kernel() local
40 inline std::tuple<int, int, int> get_stride(IntArrayRef stride, int kW, int kH, int kD) { in get_stride()
68 int kW, in get_output_shape()
H A DPooling.cpp53 int64_t kW, // kernel size in spatial_dilated_max_pooling()
114 int64_t kW, // kernel size in spatial_dilated_max_pooling3d()
190 int64_t kW, // kernel size in q_maxpool_2d()
355 int64_t kW, // kernel size in q_maxpool_3d()
540 int64_t kW = kernel_size[1]; in qnnpack_maxpool2d() local
/aosp_15_r20/external/skia/docs/examples/
H A Dskbug6031.cpp6 constexpr float kW = 512; in draw() local
/aosp_15_r20/external/pdfium/fxjs/xfa/
H A Dcjx_layoutpseudomodel.h50 enum class HWXY { kH, kW, kX, kY }; enumerator
/aosp_15_r20/external/skia/tests/
H A DReadWritePixelsGpuTest.cpp455 static constexpr int kW = 16; in gpu_read_pixels_test_driver() local
1139 static constexpr int kW = 16; in gpu_write_pixels_test_driver() local
1305 static constexpr int kW = 25, in DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS() local

123