Home
last modified time | relevance | path

Searched refs:GetBundleEntryProto (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc775 Status BundleReader::GetBundleEntryProto(StringPiece key, in GetBundleEntryProto() function in tensorflow::BundleReader
882 TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry)); in Lookup()
915 TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry)); in LookupTensorSlices()
927 TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry)); in LookupSlice()
984 GetBundleEntryProto(encoded_stored_slice_name, &stored_slice_entry); in GetSliceValue()
1053 TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry)); in LookupDtypeAndShape()
Dtensor_bundle.h269 Status GetBundleEntryProto(StringPiece key,
Dtensor_bundle_test.cc838 TF_ASSERT_OK(reader->GetBundleEntryProto(key, &full_tensor_entry)); in ExpectAlignment()