Searched defs:full_tensor_entry (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle_test.cc778 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 Dtensor_bundle.cc914 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