| 95 | } |
| 96 | |
| 97 | xla::LiteralProto NestedTuple() { |
| 98 | auto array = xla::LiteralUtil::CreateR1<float>({1.0f, 3.0f}); |
| 99 | auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); |
| 100 | auto tuple = xla::LiteralUtil::MakeTuple({&array, &matrix}); |
| 101 | auto scalar = xla::LiteralUtil::CreateR0<float>(12.0f); |
| 102 | auto nested = xla::LiteralUtil::MakeTuple({&tuple, &scalar}); |
| 103 | return nested.ToProto(); |
| 104 | } |
| 105 | |
| 106 | xla::LiteralProto MakeTuple0() { |
| 107 | auto scalar = xla::LiteralUtil::CreateR0<float>(12.0f); |