| 67 | } |
| 68 | |
| 69 | std::vector<int> GetAttrLayout(absl::Span<const int64> minor_to_mayor) { |
| 70 | std::vector<int> layout; |
| 71 | for (auto dim : minor_to_mayor) { |
| 72 | layout.push_back(static_cast<int>(dim)); |
| 73 | } |
| 74 | return layout; |
| 75 | } |
| 76 | |
| 77 | xla::LiteralProto TwoElementTuple() { |
| 78 | auto array = xla::LiteralUtil::CreateR1<float>({1.0f, 3.0f}); |