| 122 | } |
| 123 | |
| 124 | xla::StatusOr<std::vector<int>> GetShapeLayoutVector(const xla::Shape& shape) { |
| 125 | std::vector<int> layouts; |
| 126 | TF_RETURN_IF_ERROR(PopulateInfeedLayoutVector(shape, &layouts)); |
| 127 | return layouts; |
| 128 | } |
| 129 | |
| 130 | Status GetShapeWithLayout( |
| 131 | const xla::Shape& input_shape, absl::Span<const int64> minor_to_major, |