Searched refs:full_tensor_entry (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 992 const BundleEntryProto& full_tensor_entry, in GetSliceValue() argument
996 DCHECK_GE(full_tensor_entry.slices_size(), 0); in GetSliceValue()
998 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue()
1006 if (full_tensor_entry.slices().empty()) { in GetSliceValue()
1011 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
1015 for (const TensorSliceProto& slice : full_tensor_entry.slices()) { in GetSliceValue()
1017 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
1032 BundleEntryProto stored_slice_entry = full_tensor_entry; in GetSliceValue()
1074 const DataType common_dtype = full_tensor_entry.dtype(); in GetSliceValue()
|
| D | tensor_bundle.h | 282 const BundleEntryProto& full_tensor_entry,
|
| D | tensor_bundle_test.cc | 1066 BundleEntryProto full_tensor_entry; in ExpectAlignment() local
1067 TF_ASSERT_OK(reader->GetBundleEntryProto(key, &full_tensor_entry)); in ExpectAlignment()
1068 EXPECT_EQ(0, full_tensor_entry.offset() % alignment); in ExpectAlignment()
|