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.cc822 Status BundleReader::GetBundleEntryProto(StringPiece key, in GetBundleEntryProto() function in tensorflow::BundleReader
942 TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry)); in Lookup()
975 TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry)); in LookupTensorSlices()
987 TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry)); in LookupSlice()
1044 GetBundleEntryProto(encoded_stored_slice_name, &stored_slice_entry); in GetSliceValue()
1114 TF_RETURN_IF_ERROR(GetBundleEntryProto(key, &entry)); in LookupDtypeAndShape()
Dtensor_bundle.h270 Status GetBundleEntryProto(StringPiece key,
Dtensor_bundle_test.cc1067 TF_ASSERT_OK(reader->GetBundleEntryProto(key, &full_tensor_entry)); in ExpectAlignment()