Home
last modified time | relevance | path

Searched defs:top_data (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc72 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 ()()
/external/tensorflow/tensorflow/core/kernels/image/
Dresize_nearest_neighbor_op_gpu.cu.cc39 T* top_data) { in ResizeNearestNeighborNHWC()
70 T* __restrict__ top_data) { in LegacyResizeNearestNeighborNHWC()
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/
Dall.cpp108 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