Searched refs:full_tensor_entry (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 932 const BundleEntryProto& full_tensor_entry, in GetSliceValue() argument
936 DCHECK_GE(full_tensor_entry.slices_size(), 0); in GetSliceValue()
938 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue()
946 if (full_tensor_entry.slices().empty()) { in GetSliceValue()
951 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
955 for (const TensorSliceProto& slice : full_tensor_entry.slices()) { in GetSliceValue()
957 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
972 BundleEntryProto stored_slice_entry = full_tensor_entry; in GetSliceValue()
1014 const DataType common_dtype = full_tensor_entry.dtype(); in GetSliceValue()
|
| D | tensor_bundle.h | 281 const BundleEntryProto& full_tensor_entry,
|
| D | tensor_bundle_test.cc | 837 BundleEntryProto full_tensor_entry; in ExpectAlignment() local
838 TF_ASSERT_OK(reader->GetBundleEntryProto(key, &full_tensor_entry)); in ExpectAlignment()
839 EXPECT_EQ(0, full_tensor_entry.offset() % alignment); in ExpectAlignment()
|