Home
last modified time | relevance | path

Searched refs:SINGLE_ARG (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreduction_ops_half_prod_max_min.cu.cc47 #define SINGLE_ARG(...) __VA_ARGS__ macro
50 DEFINE(T, SINGLE_ARG(R), 1, 1); \
51 DEFINE(T, SINGLE_ARG(R), 2, 1); \
52 DEFINE(T, SINGLE_ARG(R), 3, 1); \
53 DEFINE(T, SINGLE_ARG(R), 3, 2); \
54 DEFINE_IDENTITY(T, SINGLE_ARG(R))
58 T, SINGLE_ARG(Eigen::internal::MinReducer<T, Eigen::PropagateNaN>)); \
60 T, SINGLE_ARG(Eigen::internal::MaxReducer<T, Eigen::PropagateNaN>)); \
64 #undef SINGLE_ARG
Dreduction_ops_gpu_double.cu.cc47 #define SINGLE_ARG(...) __VA_ARGS__ macro
50 DEFINE(T, SINGLE_ARG(R), 1, 1); \
51 DEFINE(T, SINGLE_ARG(R), 2, 1); \
52 DEFINE(T, SINGLE_ARG(R), 3, 1); \
53 DEFINE(T, SINGLE_ARG(R), 3, 2); \
54 DEFINE_IDENTITY(T, SINGLE_ARG(R))
61 T, SINGLE_ARG(Eigen::internal::MinReducer<T, Eigen::PropagateNaN>)); \
63 T, SINGLE_ARG(Eigen::internal::MaxReducer<T, Eigen::PropagateNaN>)); \
67 #undef SINGLE_ARG
Dreduction_ops_gpu_int.cu.cc47 #define SINGLE_ARG(...) __VA_ARGS__ macro
50 DEFINE(T, SINGLE_ARG(R), 1, 1); \
51 DEFINE(T, SINGLE_ARG(R), 2, 1); \
52 DEFINE(T, SINGLE_ARG(R), 3, 1); \
53 DEFINE(T, SINGLE_ARG(R), 3, 2); \
54 DEFINE_IDENTITY(T, SINGLE_ARG(R))
61 T, SINGLE_ARG(Eigen::internal::MinReducer<T, Eigen::PropagateNaN>)); \
63 T, SINGLE_ARG(Eigen::internal::MaxReducer<T, Eigen::PropagateNaN>)); \
68 #undef SINGLE_ARG
Dreduction_ops_gpu_float.cu.cc47 #define SINGLE_ARG(...) __VA_ARGS__ macro
50 DEFINE(T, SINGLE_ARG(R), 1, 1); \
51 DEFINE(T, SINGLE_ARG(R), 2, 1); \
52 DEFINE(T, SINGLE_ARG(R), 3, 1); \
53 DEFINE(T, SINGLE_ARG(R), 3, 2); \
54 DEFINE_IDENTITY(T, SINGLE_ARG(R))
61 T, SINGLE_ARG(Eigen::internal::MinReducer<T, Eigen::PropagateNaN>)); \
63 T, SINGLE_ARG(Eigen::internal::MaxReducer<T, Eigen::PropagateNaN>)); \
67 #undef SINGLE_ARG
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc719 #define SINGLE_ARG(...) __VA_ARGS__ macro
728 CASE(float, SINGLE_ARG(STMTS)) \
729 CASE(double, SINGLE_ARG(STMTS)) \
730 CASE(int32, SINGLE_ARG(STMTS)) \
731 CASE(uint8, SINGLE_ARG(STMTS)) \
732 CASE(uint16, SINGLE_ARG(STMTS)) \
733 CASE(uint32, SINGLE_ARG(STMTS)) \
734 CASE(uint64, SINGLE_ARG(STMTS)) \
735 CASE(int16, SINGLE_ARG(STMTS)) \
736 CASE(int8, SINGLE_ARG(STMTS)) \
[all …]