| 3209 | } |
| 3210 | |
| 3211 | XlaOp CustomCallWithLayout(XlaBuilder* builder, const string& call_target_name, |
| 3212 | absl::Span<const XlaOp> operands, const Shape& shape, |
| 3213 | absl::Span<const Shape> operand_shapes_with_layout, |
| 3214 | const string& opaque) { |
| 3215 | return builder->CustomCall(call_target_name, operands, shape, opaque, |
| 3216 | operand_shapes_with_layout); |
| 3217 | } |
| 3218 | |
| 3219 | XlaOp Complex(const XlaOp lhs, const XlaOp rhs, |
| 3220 | absl::Span<const int64> broadcast_dimensions) { |
no test coverage detected