Searched refs:AddBroadcastSequence (Results 1 – 2 of 2) sorted by relevance
564 StatusOr<XlaOp> XlaBuilder::AddBroadcastSequence(const Shape& output_shape, in AddBroadcastSequence() function in xla::XlaBuilder672 AddBroadcastSequence(shape, updated_lhs)); in BinaryOp()678 AddBroadcastSequence(shape, updated_rhs)); in BinaryOp()760 AddBroadcastSequence(*non_scalar_shape, lhs)); in TernaryOp()764 AddBroadcastSequence(*non_scalar_shape, rhs)); in TernaryOp()768 AddBroadcastSequence(*non_scalar_shape, ehs)); in TernaryOp()936 return AddBroadcastSequence(output_shape, in_dim_broadcast); in BroadcastInDim()2157 AddBroadcastSequence(output_shape, new_operand)); in Map()
962 StatusOr<XlaOp> AddBroadcastSequence(const Shape& output_shape,