Searched refs:DEFINE_GPU_KERNELS (Results 1 – 15 of 15) sorted by relevance
56 #define DEFINE_GPU_KERNELS(T) \ macro59 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);60 TF_CALL_int32(DEFINE_GPU_KERNELS);61 TF_CALL_int64(DEFINE_GPU_KERNELS);62 TF_CALL_int8(DEFINE_GPU_KERNELS);63 #undef DEFINE_GPU_KERNELS65 #define DEFINE_GPU_KERNELS(T) \ macro67 TF_CALL_GPU_ALL_TYPES(DEFINE_GPU_KERNELS);68 TF_CALL_int32(DEFINE_GPU_KERNELS);69 TF_CALL_int64(DEFINE_GPU_KERNELS);[all …]
30 #define DEFINE_GPU_KERNELS(T) \ macro39 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);40 TF_CALL_complex64(DEFINE_GPU_KERNELS);41 TF_CALL_complex128(DEFINE_GPU_KERNELS);42 TF_CALL_bfloat16(DEFINE_GPU_KERNELS);43 TF_CALL_bool(DEFINE_GPU_KERNELS);44 TF_CALL_int8(DEFINE_GPU_KERNELS);45 DEFINE_GPU_KERNELS(int32);46 DEFINE_GPU_KERNELS(int64);48 #undef DEFINE_GPU_KERNELS
51 #define DEFINE_GPU_KERNELS(T) \ macro55 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);56 TF_CALL_complex64(DEFINE_GPU_KERNELS);57 TF_CALL_complex128(DEFINE_GPU_KERNELS);58 TF_CALL_int64(DEFINE_GPU_KERNELS);59 TF_CALL_bfloat16(DEFINE_GPU_KERNELS);60 TF_CALL_uint8(DEFINE_GPU_KERNELS);61 TF_CALL_bool(DEFINE_GPU_KERNELS);63 #undef DEFINE_GPU_KERNELS64 #define DEFINE_GPU_KERNELS(T) template struct SplitCustom<Eigen::GpuDevice, T>; macro[all …]
24 TF_CALL_int8(DEFINE_GPU_KERNELS);25 TF_CALL_int32(DEFINE_GPU_KERNELS);26 TF_CALL_int64(DEFINE_GPU_KERNELS);
35 #define DEFINE_GPU_KERNELS(T) \ macro39 DEFINE_GPU_KERNELS(float);40 DEFINE_GPU_KERNELS(double);
33 #define DEFINE_GPU_KERNELS(T) \ macro36 DEFINE_GPU_KERNELS(Eigen::half)37 DEFINE_GPU_KERNELS(float) in DEFINE_GPU_KERNELS() function38 DEFINE_GPU_KERNELS(double) in DEFINE_GPU_KERNELS()40 #undef DEFINE_GPU_KERNELS in DEFINE_GPU_KERNELS()
24 TF_CALL_complex64(DEFINE_GPU_KERNELS);25 TF_CALL_complex128(DEFINE_GPU_KERNELS);
28 #define DEFINE_GPU_KERNELS(T) template struct functor::Snapshot<GPUDevice, T>; macro30 TF_CALL_POD_TYPES(DEFINE_GPU_KERNELS);
32 #define DEFINE_GPU_KERNELS(T) \ macro36 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);
148 #define DEFINE_GPU_KERNELS(T) \ macro160 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);
24 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);
24 TF_CALL_bool(DEFINE_GPU_KERNELS);
33 #define DEFINE_GPU_KERNELS(T) \ macro
517 #define DEFINE_GPU_KERNELS(T) \ macro525 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS);527 #undef DEFINE_GPU_KERNELS