Searched defs:top_data (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op_gpu.cu.cc | 72 const int pad_l, dtype* __restrict__ top_data, int64* __restrict__ mask, in MaxPoolForwardNCHW()
116 int32* __restrict__ top_data) { in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
149 dtype* __restrict__ top_data, int64* __restrict__ mask, in MaxPoolForwardNHWC()
359 int32* top_data, const Eigen::GpuDevice& d) { in operator ()()
378 const int stride_w, const int pad_t, const int pad_l, T* top_data, in operator ()()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/ |
| H A D | all.cpp | 108 float16x4_t top_data = read_4_boundary_aware_fp16(src_w, src_h, pool_pad_left, pool_pad_top, in pooling3_fp16_neon_nchw() local
214 auto top_data = read_2_boundary_aware(src_w, src_h, pool_pad_left, pool_pad_top, in pooling2_nchw_maxpool_indices() local
276 … float16x4_t top_data = read_4_boundary_aware_fp16(src_w, src_h, pool_pad_left, pool_pad_top, in pooling2_fp16_neon_nchw() local
526 …auto top_data = READ_2_BOUNDARY_AWARE(src_h, src_w, pool_pad_left, pool_pad_top, x_val, … in pooling2_fp32_neon_nchw() local
603 …auto top_data = READ_4_BOUNDARY_AWARE(src_h, src_w, pool_pad_left, pool_pad_top, x_val, y… in pooling3_fp32_neon_nchw() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | resize_nearest_neighbor_op_gpu.cu.cc | 39 T* top_data) { in ResizeNearestNeighborNHWC()
70 T* __restrict__ top_data) { in LegacyResizeNearestNeighborNHWC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 38 scalar_t* const top_data, const int divisor_override, in avg_pool2d_out_cuda_frame()
87 scalar_t* const top_data, const int divisor_override, in avg_pool2d_out_cuda_frame_nhwc()
|
| H A D | DilatedMaxPool2d.cu | 50 const int dilation_h, const int dilation_w, scalar_t* top_data, in max_pool_forward_nchw()
|