| 37 | // Constructs a flat tensor with 'vals'. |
| 38 | template <typename T> |
| 39 | Tensor AsTensor(gtl::ArraySlice<T> vals) { |
| 40 | Tensor ret(DataTypeToEnum<T>::value, {static_cast<int64>(vals.size())}); |
| 41 | std::copy_n(vals.data(), vals.size(), ret.flat<T>().data()); |
| 42 | return ret; |
| 43 | } |
| 44 | |
| 45 | // Constructs a tensor of "shape" with values "vals". |
| 46 | template <typename T> |
no test coverage detected