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,
244 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
281 Status GetSliceValue(StringPiece full_tensor_key,
|
| D | tensor_bundle.cc | 476 Status BundleWriter::AddSlice(StringPiece full_tensor_key, in AddSlice() argument
481 CHECK_NE(full_tensor_key, kHeaderEntryKey); in AddSlice()
485 return Add(full_tensor_key, slice_tensor); in AddSlice()
493 const string full_tensor_key_string(full_tensor_key); in AddSlice()
983 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice() argument
987 TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry)); in LookupSlice()
988 return GetSliceValue(full_tensor_key, entry, slice_spec, val); in LookupSlice()
991 Status BundleReader::GetSliceValue(StringPiece full_tensor_key, in GetSliceValue() argument
1000 const string full_tensor_key_string(full_tensor_key); in GetSliceValue()
1026 full_tensor_key, in GetSliceValue()
|