Searched refs:DEFINE_BINARY4 (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | cwise_op_gpu_maximum.cu.cc | 22 DEFINE_BINARY4(maximum, Eigen::half, float, double, int64);
|
H A D | cwise_op_gpu_minimum.cu.cc | 22 DEFINE_BINARY4(minimum, Eigen::half, float, double, int64);
|
H A D | cwise_op_gpu_pow.cu.cc | 22 DEFINE_BINARY4(pow, Eigen::half, float, double, int64);
|
H A D | cwise_op_gpu_squared_difference.cu.cc | 22 DEFINE_BINARY4(squared_difference, float, Eigen::half, double, int64);
|
H A D | cwise_ops_sycl_common.h | 138 #define DEFINE_BINARY4(F, T0, T1, T2, T3) \ macro 149 DEFINE_BINARY4(F, T3, T4, T5, T6) 151 DEFINE_BINARY4(F, T0, T1, T2, T3); \ 152 DEFINE_BINARY4(F, T4, T5, T6, T7) 154 DEFINE_BINARY4(F, T0, T1, T2, T3); \
|
H A D | cwise_ops_gpu_common.cu.h | 157 #define DEFINE_BINARY4(F, T0, T1, T2, T3) \ macro 168 DEFINE_BINARY4(F, T3, T4, T5, T6) 170 DEFINE_BINARY4(F, T0, T1, T2, T3); \ 171 DEFINE_BINARY4(F, T4, T5, T6, T7) 173 DEFINE_BINARY4(F, T0, T1, T2, T3); \
|
Completed in 103 milliseconds