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);
|