Home
last modified time | relevance | path

Searched defs:bottom_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc68 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 ()()
Dpooling_ops_3d_gpu.cu.cc31 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 ()()
/external/tensorflow/tensorflow/core/kernels/image/
Dresize_nearest_neighbor_op_gpu.cu.cc36 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()
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/
Dall.cpp112 … 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