Searched refs:broadcast_helper (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | binary_ops.cc | 44 const BCast& broadcast_helper, \
71 xla::XlaOp y, const BCast& broadcast_helper) { in DivNoNanImpl() argument
72 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in DivNoNanImpl()
80 DivNoNanImpl(b, input_type(0), lhs, rhs, broadcast_helper));
89 xla::XlaOp y, const BCast& broadcast_helper) { in MulNoNanImpl() argument
90 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in MulNoNanImpl()
98 MulNoNanImpl(b, input_type(0), lhs, rhs, broadcast_helper));
112 xla::XlaOp y, const BCast& broadcast_helper) { in FloorDivImpl() argument
113 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in FloorDivImpl()
129 FloorDivImpl(b, input_type(0), lhs, rhs, broadcast_helper));
[all …]
|
| D | cwise_ops.cc | 91 xla::XlaOp lhs, xla::XlaOp rhs, const BCast& broadcast_helper) { in Broadcast() argument
92 auto lhs_output = BroadcastTo(lhs, broadcast_helper.output_shape()); in Broadcast()
97 auto rhs_output = BroadcastTo(rhs, broadcast_helper.output_shape()); in Broadcast()
|
| D | cwise_ops.h | 61 const absl::Span<const int64>& rhs_shape, const BCast& broadcast_helper,
70 xla::XlaOp lhs, xla::XlaOp rhs, const BCast& broadcast_helper);
|