| 172 | } |
| 173 | |
| 174 | void FillStringTensor(Tensor* dst, const Tensor& src) { |
| 175 | auto dst_flat = dst->flat<tstring>(); |
| 176 | auto src_flat = src.flat<tstring>(); |
| 177 | for (int i = 0; i < src.NumElements(); i++) { |
| 178 | dst_flat(i) = src_flat(i); |
| 179 | } |
| 180 | } |
| 181 | |
| 182 | // Add the added_variables as an inputs to the Save op. |
| 183 | // We change the inputs of the SaveV2 op to include the names of the added |
no test coverage detected