| 3188 | } |
| 3189 | |
| 3190 | XlaOp Infeed(XlaBuilder* builder, const Shape& shape, const string& config) { |
| 3191 | return builder->Infeed(shape, config); |
| 3192 | } |
| 3193 | |
| 3194 | void Outfeed(const XlaOp operand, const Shape& shape_with_layout, |
| 3195 | const string& outfeed_config) { |