Home
last modified time | relevance | path

Searched refs:full_tensor_key (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.h143 Status AddSlice(StringPiece full_tensor_key,
270 Status LookupSlice(StringPiece full_tensor_key, const TensorSlice& slice_spec,
307 Status GetSliceValue(StringPiece full_tensor_key,
Dtensor_bundle.cc478 Status BundleWriter::AddSlice(StringPiece full_tensor_key, in AddSlice() argument
483 CHECK_NE(full_tensor_key, kHeaderEntryKey); in AddSlice()
487 return Add(full_tensor_key, slice_tensor); in AddSlice()
495 const string full_tensor_key_string(full_tensor_key); in AddSlice()
996 Status BundleReader::LookupSlice(StringPiece full_tensor_key, in LookupSlice() argument
1000 TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry)); in LookupSlice()
1001 return GetSliceValue(full_tensor_key, entry, slice_spec, val); in LookupSlice()
1004 Status BundleReader::GetSliceValue(StringPiece full_tensor_key, in GetSliceValue() argument
1013 const string full_tensor_key_string(full_tensor_key); in GetSliceValue()
1039 full_tensor_key, in GetSliceValue()