HomeSort by relevance Sort by last modified time
    Searched refs:GPU_PROVIDED_DIM (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
where_op_gpu_impl_1.cu.cc 16 #define GPU_PROVIDED_DIM 1
18 #undef GPU_PROVIDED_DIM
where_op_gpu_impl_2.cu.cc 16 #define GPU_PROVIDED_DIM 2
18 #undef GPU_PROVIDED_DIM
where_op_gpu_impl_3.cu.cc 16 #define GPU_PROVIDED_DIM 3
18 #undef GPU_PROVIDED_DIM
where_op_gpu_impl_4.cu.cc 16 #define GPU_PROVIDED_DIM 4
18 #undef GPU_PROVIDED_DIM
where_op_gpu_impl_5.cu.cc 16 #define GPU_PROVIDED_DIM 5
18 #undef GPU_PROVIDED_DIM
where_op_gpu.cu.h 191 #if GPU_PROVIDED_DIM == 1
195 #endif // GPU_PROVIDED_DIM == 1
336 DECLARE_GPU_SPEC_INDEX(GPU_PROVIDED_DIM, T, int32); \
337 DECLARE_GPU_SPEC_INDEX(GPU_PROVIDED_DIM, T, int64)

Completed in 71 milliseconds