Searched defs:bottom_data (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op_gpu.cu.cc | 68 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolForwardNCHW()
112 const int nthreads, const int32* __restrict__ bottom_data, const int height, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
145 const int nthreads, const dtype* __restrict__ bottom_data, const int height, in MaxPoolForwardNHWC()
231 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNCHW()
272 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNHWC()
355 const int32* bottom_data, const int batch, const int height, in operator ()()
375 const T* bottom_data, const int batch, const int height, const int width, in operator ()()
425 TensorFormat data_format, const T* bottom_data, const T* output_data, in operator ()()
|
| H A D | pooling_ops_3d_gpu.cu.cc | 31 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNCDHW()
82 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNDHWC()
138 TensorFormat data_format, const T* bottom_data, const T* output_data, in operator ()()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/ |
| H A D | all.cpp | 112 … float16x4_t bottom_data = read_4_boundary_aware_fp16(src_w, src_h, pool_pad_left, pool_pad_top, in pooling3_fp16_neon_nchw() local
216 auto bottom_data = read_2_boundary_aware(src_w, src_h, pool_pad_left, pool_pad_top, in pooling2_nchw_maxpool_indices() local
278 … float16x4_t bottom_data = read_4_boundary_aware_fp16(src_w, src_h, pool_pad_left, pool_pad_top, in pooling2_fp16_neon_nchw() local
527 …auto bottom_data = READ_2_BOUNDARY_AWARE(src_h, src_w, pool_pad_left, pool_pad_top, x_val, … in pooling2_fp32_neon_nchw() local
605 …auto bottom_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 | 36 const int nthreads, const T* __restrict__ bottom_data, const int in_height, in ResizeNearestNeighborNHWC()
67 const int nthreads, const T* __restrict__ bottom_data, const int in_height, in LegacyResizeNearestNeighborNHWC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 34 const scalar_t* const bottom_data, const int64_t channels, in avg_pool2d_out_cuda_frame()
83 const scalar_t* const bottom_data, const int64_t channels, in avg_pool2d_out_cuda_frame_nhwc()
|
| H A D | DilatedMaxPool2d.cu | 45 __global__ void max_pool_forward_nchw(const int nthreads, const scalar_t* bottom_data, in max_pool_forward_nchw()
|