| 1205 | } |
| 1206 | |
| 1207 | Status BundleReader::LookupSlice(StringPiece full_tensor_key, |
| 1208 | const TensorSlice& slice_spec, Tensor* val) { |
| 1209 | CHECK(val != nullptr); |
| 1210 | BundleEntryProto entry; |
| 1211 | TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry)); |
| 1212 | return GetSliceValue(full_tensor_key, entry, slice_spec, val); |
| 1213 | } |
| 1214 | |
| 1215 | Status BundleReader::LookupTensorSliceProtos( |
| 1216 | StringPiece key, std::vector<TensorSliceProto>* slices) { |
no outgoing calls