Searched refs:DEFINE_GPU_SPECS (Results 1 – 16 of 16) sorted by relevance
69 #define DEFINE_GPU_SPECS(T) \ macro82 TF_CALL_uint8(DEFINE_GPU_SPECS);83 TF_CALL_int8(DEFINE_GPU_SPECS);84 TF_CALL_uint16(DEFINE_GPU_SPECS);85 TF_CALL_int16(DEFINE_GPU_SPECS);86 TF_CALL_int32(DEFINE_GPU_SPECS);87 TF_CALL_int64(DEFINE_GPU_SPECS);89 #undef DEFINE_GPU_SPECS
30 #define DEFINE_GPU_SPECS(T) \ macro34 TF_CALL_int32(DEFINE_GPU_SPECS);35 TF_CALL_int64(DEFINE_GPU_SPECS);36 TF_CALL_GPU_ALL_TYPES(DEFINE_GPU_SPECS);38 #undef DEFINE_GPU_SPECS
92 #define DEFINE_GPU_SPECS(T) template struct Roll<GPUDevice, T>; macro94 TF_CALL_int32(DEFINE_GPU_SPECS);95 TF_CALL_int64(DEFINE_GPU_SPECS);96 TF_CALL_uint32(DEFINE_GPU_SPECS);97 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);98 TF_CALL_COMPLEX_TYPES(DEFINE_GPU_SPECS);100 #undef DEFINE_GPU_SPECS
40 #define DEFINE_GPU_SPECS(T) \ macro44 DEFINE_GPU_SPECS(Eigen::half);45 DEFINE_GPU_SPECS(float);46 DEFINE_GPU_SPECS(double);50 #undef DEFINE_GPU_SPECS
39 #define DEFINE_GPU_SPECS(T) \ macro43 DEFINE_GPU_SPECS(Eigen::half);44 DEFINE_GPU_SPECS(float);45 DEFINE_GPU_SPECS(double);49 #undef DEFINE_GPU_SPECS
41 #define DEFINE_GPU_SPECS(T) \ macro45 TF_CALL_GPU_ALL_TYPES(DEFINE_GPU_SPECS);46 TF_CALL_int8(DEFINE_GPU_SPECS);47 TF_CALL_uint8(DEFINE_GPU_SPECS);
117 #define DEFINE_GPU_SPECS(T) \ macro121 TF_CALL_int32(DEFINE_GPU_SPECS);122 TF_CALL_int64(DEFINE_GPU_SPECS);123 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);124 TF_CALL_COMPLEX_TYPES(DEFINE_GPU_SPECS);126 #undef DEFINE_GPU_SPECS
35 #define DEFINE_GPU_SPECS(T) \ macro41 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);42 TF_CALL_bool(DEFINE_GPU_SPECS);
115 #define DEFINE_GPU_SPECS(T) \ macro131 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS)132 TF_CALL_bool(DEFINE_GPU_SPECS)
168 #define DEFINE_GPU_SPECS(T) template struct functor::MaxPool3dGradBackward<T>; macro169 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);170 #undef DEFINE_GPU_SPECS
294 #define DEFINE_GPU_SPECS(T) \ macro299 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);301 #undef DEFINE_GPU_SPECS
286 #define DEFINE_GPU_SPECS(T) \ macro290 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
26 #define DEFINE_GPU_SPECS(T) \ macro30 DEFINE_GPU_SPECS(float);31 #undef DEFINE_GPU_SPECS
27 #define DEFINE_GPU_SPECS(T) \ macro49 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);50 #undef DEFINE_GPU_SPECS
482 #define DEFINE_GPU_SPECS(T) \ macro487 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);489 #undef DEFINE_GPU_SPECS