Home
last modified time | relevance | path

Searched refs:buffered_file (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc69 Status ReadStringTensor(io::InputBuffer* buffered_file, size_t num_elements, in ReadStringTensor() argument
76 TF_RETURN_IF_ERROR(buffered_file->Seek(offset)); in ReadStringTensor()
79 TF_RETURN_IF_ERROR(buffered_file->ReadVarint64(&string_lengths[i])); in ReadStringTensor()
93 if (offset + size < buffered_file->Tell()) { in ReadStringTensor()
101 TF_RETURN_IF_ERROR(buffered_file->ReadNBytes( in ReadStringTensor()
122 buffered_file->ReadNBytes(string_length, &(*buffer)[0], &bytes_read)); in ReadStringTensor()
128 Status ReadVariantTensor(io::InputBuffer* buffered_file, Tensor* ret, in ReadVariantTensor() argument
140 TF_RETURN_IF_ERROR(buffered_file->Seek(offset)); in ReadVariantTensor()
144 TF_RETURN_IF_ERROR(buffered_file->ReadVarint64(&string_length)); in ReadVariantTensor()
153 buffered_file->ReadNBytes(string_length, &buffer[0], &bytes_read)); in ReadVariantTensor()
[all …]