Searched refs:AddBroadcastSequence (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 431 StatusOr<XlaOp> XlaBuilder::AddBroadcastSequence(const Shape& output_shape, in AddBroadcastSequence() function in xla::XlaBuilder
541 AddBroadcastSequence(shape, updated_lhs)); in BinaryOp()
546 AddBroadcastSequence(shape, updated_rhs)); in BinaryOp()
571 TF_ASSIGN_OR_RETURN(updated_lhs, AddBroadcastSequence(shape, lhs)); in TernaryOp()
576 TF_ASSIGN_OR_RETURN(updated_rhs, AddBroadcastSequence(shape, rhs)); in TernaryOp()
581 TF_ASSIGN_OR_RETURN(updated_ehs, AddBroadcastSequence(shape, ehs)); in TernaryOp()
709 return AddBroadcastSequence(output_shape, in_dim_broadcast); in BroadcastInDim()
1772 AddBroadcastSequence(output_shape, new_operand)); in Map()
|
| D | xla_builder.h | 618 StatusOr<XlaOp> AddBroadcastSequence(const Shape& output_shape,
|