| 114 | } |
| 115 | |
| 116 | xla::LiteralProto FloatVector(absl::Span<const float> v) { |
| 117 | auto array = xla::LiteralUtil::CreateR1<float>(v); |
| 118 | return array.ToProto(); |
| 119 | } |
| 120 | |
| 121 | xla::LiteralProto FloatMatrix( |
| 122 | std::initializer_list<std::initializer_list<float>> v, |