| 3202 | } |
| 3203 | |
| 3204 | XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name, |
| 3205 | absl::Span<const XlaOp> operands, const Shape& shape, |
| 3206 | const string& opaque) { |
| 3207 | return builder->CustomCall(call_target_name, operands, shape, opaque, |
| 3208 | /*operand_shapes_with_layout=*/absl::nullopt); |
| 3209 | } |
| 3210 | |
| 3211 | XlaOp CustomCallWithLayout(XlaBuilder* builder, const string& call_target_name, |
| 3212 | absl::Span<const XlaOp> operands, const Shape& shape, |