Home
last modified time | relevance | path

Searched defs:kH (Results 1 – 25 of 63) 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.cpp40 int64_t kH, in unfolded2d_acc()
125 int64_t kH, in unfolded2d_acc_channels_last()
182 int64_t kH, in unfolded2d_acc_kernel()
233 int64_t kH, in unfolded2d_copy()
340 int64_t kH, in unfolded2d_copy_channels_last()
397 int64_t kH, in unfolded2d_copy_kernel()
H A DMaxPoolKernel.cpp248 int kH = kWHD[1]; in cpu_max_pool() local
377 int kH = kWHD[1]; 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/cuda/
H A 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()
363 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
469 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
H A DDepthwiseConv2d.cu127 for (int kH = kHmin; kH < kHmax; ++kH) { in C10_LAUNCH_BOUNDS_1() local
195 for (int kH = 0; kH < KH_LIMIT; ++kH) { in C10_LAUNCH_BOUNDS_1() local
304 int kH = (bidx / kernelWidth) % kernelHeight; 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()
304 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in max_pool3d_with_indices_out_cuda_template() local
434 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in max_pool3d_with_indices_backward_out_cuda_template() local
H A DAdaptiveAveragePooling3d.cu89 int kH = iendH - istartH; in adaptiveaveragepool() local
202 int kH = end_index(oh, osizeH, isizeH) - start_index(oh, osizeH, isizeH); in adaptiveaveragegradinput() local
287 int kH = iendH - istartH; in atomicadaptiveaveragegradinput() local
H A DDepthwiseConv3d.cu38 const int kH = kKnownKernelH > 0 ? kKnownKernelH : kernel.size(3); in conv_depthwise3d_cuda_kernel() local
109 const int kH = kKnownKernelH > 0 ? kKnownKernelH : kernel.size(3); in conv_depthwise3d_cuda_backward_input_kernel() local
191 const int kH = grad_kernel.size(3); in conv_depthwise3d_cuda_backward_weight_kernel() local
H A DAdaptiveAveragePooling.cu82 int kH = iendH - istartH; in adaptive_average_pool() local
152 int kH = START_IND(oh, osizeH, isizeH) - END_IND(oh, osizeH, isizeH); in adaptive_average_gradinput() local
197 int kH = iendH - istartH; in atomic_adaptive_average_gradinput() local
H A DAveragePool2d.cu260 const int kH = safe_downcast<int, int64_t>(kH_); in TORCH_IMPL_FUNC() local
375 const int kH = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAveragePool3d.cpp33 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_META_FUNC() local
102 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_META_FUNC() local
168 int kH, in avg_pool3d_out_frame()
265 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in TORCH_IMPL_FUNC() local
354 int kH, in avg_pool3d_backward_out_frame()
438 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); 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.cpp28 const int64_t kH = kernel_size[0]; in TORCH_PRECOMPUTE_META_FUNC() local
109 const int kH = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_META_FUNC() local
189 const int kH = safe_downcast<int, int64_t>(kernel_size[0]); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool3d.cpp35 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in max_pool3d_with_indices_out_cpu_template() local
127 const int kH = kernel_size.size() == 1 ? kT : safe_downcast<int, int64_t>(kernel_size[1]); in max_pool3d_with_indices_backward_out_cpu_template() local
H A DAdaptiveAveragePooling3d.cpp52 auto kH = iendH - istartH; in adaptive_avg_pool3d_out_frame() local
203 auto kH = iendH - istartH; in adaptive_avg_pool3d_backward_out_frame() local
/aosp_15_r20/external/skia/gm/
H A Dshadowutils.cpp41 static constexpr int kH = 960; 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/quantized/cpu/
H A DAveragePool2d.cpp44 int kH, in avg_pool2d_out_frame()
129 const int kH = safe_downcast<int, int64_t>(kernel_size[0]); in get_kernel() local
136 inline std::pair<int, int> get_stride(IntArrayRef stride, int kW, int kH) { in get_stride()
160 int kH, in get_output_shape()
H A DAveragePool3d.cpp31 const int kH = 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()
69 int kH, in get_output_shape()
H A DPooling.cpp52 int64_t kH, in spatial_dilated_max_pooling()
113 int64_t kH, in spatial_dilated_max_pooling3d()
189 int64_t kH, in q_maxpool_2d()
354 int64_t kH, in q_maxpool_3d()
539 int64_t kH = kernel_size[0]; in qnnpack_maxpool2d() local
/aosp_15_r20/external/skia/docs/examples/
H A Dskbug6031.cpp7 constexpr float kH = 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.cpp456 static constexpr int kH = 16; in gpu_read_pixels_test_driver() local
1140 static constexpr int kH = 16; in gpu_write_pixels_test_driver() local
1306 kH = 37; in DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS() local
/aosp_15_r20/external/boringssl/src/crypto/fipsmodule/modes/
H A Dgcm_test.cc118 static const uint64_t kH[2] = { in TEST() local
/aosp_15_r20/external/rust/crates/quiche/deps/boringssl/src/crypto/fipsmodule/modes/
Dgcm_test.cc124 static const uint64_t kH[2] = { in TEST() local

123