Searched refs:full_tensor_key (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.h | 139 Status AddSlice(StringPiece full_tensor_key,
243 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
280 Status GetSliceValue(StringPiece full_tensor_key,
|
| D | tensor_bundle.cc | 449 Status BundleWriter::AddSlice(StringPiece full_tensor_key, in AddSlice() argument
454 CHECK_NE(full_tensor_key, kHeaderEntryKey); in AddSlice()
458 return Add(full_tensor_key, slice_tensor); in AddSlice()
466 const string full_tensor_key_string(full_tensor_key); in AddSlice()
923 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice() argument
927 TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry)); in LookupSlice()
928 return GetSliceValue(full_tensor_key, entry, slice_spec, val); in LookupSlice()
931 Status BundleReader::GetSliceValue(StringPiece full_tensor_key, in GetSliceValue() argument
940 const string full_tensor_key_string(full_tensor_key); in GetSliceValue()
966 full_tensor_key, in GetSliceValue()
|