Searched refs:XLA_MAKE_BINARY (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | binary_ops.cc | 36 #define XLA_MAKE_BINARY(NAME, HLO) \ macro
56 XLA_MAKE_BINARY(Add, xla::Add(lhs, rhs, extend_dimensions));
57 XLA_MAKE_BINARY(Sub, xla::Sub(lhs, rhs, extend_dimensions));
58 XLA_MAKE_BINARY(Mul, xla::Mul(lhs, rhs, extend_dimensions));
59 XLA_MAKE_BINARY(Div, xla::Div(lhs, rhs, extend_dimensions));
61 XLA_MAKE_BINARY(Atan2, xla::Atan2(lhs, rhs, extend_dimensions));
62 XLA_MAKE_BINARY(Complex, xla::Complex(lhs, rhs, extend_dimensions));
79 XLA_MAKE_BINARY(DivNoNan,
97 XLA_MAKE_BINARY(MulNoNan,
128 XLA_MAKE_BINARY(FloorDiv,
[all …]
|