Searched refs:updated_rhs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 509 XlaOp updated_rhs = rhs; in BinaryOp() local
535 updated_rhs = !should_broadcast_lhs ? broadcasted_operand : rhs; in BinaryOp()
543 TF_ASSIGN_OR_RETURN(Shape updated_rhs_shape, GetShape(updated_rhs)); in BinaryOp()
545 TF_ASSIGN_OR_RETURN(updated_rhs, in BinaryOp()
546 AddBroadcastSequence(shape, updated_rhs)); in BinaryOp()
549 return AddInstruction(std::move(instr), binop, {updated_lhs, updated_rhs}); in BinaryOp()
565 XlaOp updated_rhs = rhs; in TernaryOp() local
576 TF_ASSIGN_OR_RETURN(updated_rhs, AddBroadcastSequence(shape, rhs)); in TernaryOp()
585 {updated_lhs, updated_rhs, updated_ehs}); in TernaryOp()
|