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

  /external/tensorflow/tensorflow/core/kernels/
gather_functor_gpu.cu.cc 27 #define DEFINE_GPU_SPECS_INDEX(T, Index) \
31 DEFINE_GPU_SPECS_INDEX(T, int32); \
32 DEFINE_GPU_SPECS_INDEX(T, int64);
39 #undef DEFINE_GPU_SPECS_INDEX
scatter_functor_gpu.cu.cc 29 #define DEFINE_GPU_SPECS_INDEX(T, Index) \
37 DEFINE_GPU_SPECS_INDEX(T, int32); \
38 DEFINE_GPU_SPECS_INDEX(T, int64);
47 #undef DEFINE_GPU_SPECS_INDEX
scatter_op_gpu.cu.cc 30 #define DEFINE_GPU_SPECS_INDEX(T, Index) \
38 DEFINE_GPU_SPECS_INDEX(T, int32); \
39 DEFINE_GPU_SPECS_INDEX(T, int64);
47 #undef DEFINE_GPU_SPECS_INDEX
gather_nd_op_gpu.cu.cc 108 #define DEFINE_GPU_SPECS_INDEX(T, Index) \
119 DEFINE_GPU_SPECS_INDEX(T, int32); \
120 DEFINE_GPU_SPECS_INDEX(T, int64);
127 #undef DEFINE_GPU_SPECS_INDEX

Completed in 167 milliseconds