Searched refs:DEFINE_BINARY4 (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcwise_op_gpu_maximum.cu.cc22 DEFINE_BINARY4(maximum, Eigen::half, float, double, int64);
H A Dcwise_op_gpu_minimum.cu.cc22 DEFINE_BINARY4(minimum, Eigen::half, float, double, int64);
H A Dcwise_op_gpu_pow.cu.cc22 DEFINE_BINARY4(pow, Eigen::half, float, double, int64);
H A Dcwise_op_gpu_squared_difference.cu.cc22 DEFINE_BINARY4(squared_difference, float, Eigen::half, double, int64);
H A Dcwise_ops_sycl_common.h138 #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 Dcwise_ops_gpu_common.cu.h157 #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