Searched defs:full_tensor_entry (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/util/tensor_bundle/ |
H A D | tensor_bundle_test.cc | 778 BundleEntryProto full_tensor_entry; local 779 TF_ASSERT_OK(reader->GetBundleEntryProto(key, &full_tensor_entry)); 780 EXPECT_EQ(0, full_tensor_entry.offset() % alignment);
|
H A D | tensor_bundle.cc | 914 const BundleEntryProto& full_tensor_entry, 918 DCHECK_GE(full_tensor_entry.slices_size(), 0); 920 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); 928 if (full_tensor_entry.slices().empty()) { 933 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), 937 for (const TensorSliceProto& slice : full_tensor_entry.slices()) { 939 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), 954 BundleEntryProto stored_slice_entry = full_tensor_entry; 996 const DataType common_dtype = full_tensor_entry.dtype(); 913 GetSliceValue(StringPiece full_tensor_key, const BundleEntryProto& full_tensor_entry, const TensorSlice& slice_spec, Tensor* val) argument
|
Completed in 107 milliseconds