| 55 | // test::FillValues<float>(&x, {11, 21, 21, 22}); |
| 56 | template <typename T> |
| 57 | void FillValues(Tensor* tensor, gtl::ArraySlice<T> vals) { |
| 58 | auto flat = tensor->flat<T>(); |
| 59 | CHECK_EQ(flat.size(), vals.size()); |
| 60 | if (flat.size() > 0) { |
| 61 | std::copy_n(vals.data(), vals.size(), flat.data()); |
| 62 | } |
| 63 | } |
| 64 | |
| 65 | // Fills in '*tensor' with 'vals', converting the types as needed. |
| 66 | template <typename T, typename SrcType> |