Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/data/
Dcompression_utils.cc85 std::vector<tstring> tensor_proto_strs; in UncompressElement() local
89 tensor_proto_strs.reserve(num_components); in UncompressElement()
103 tensor_proto_strs.emplace_back(); in UncompressElement()
104 tstring& tensor_proto_str = tensor_proto_strs.back(); in UncompressElement()
139 if (!tp.ParseFromString(tensor_proto_strs[tensor_proto_strs_index++])) { in UncompressElement()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc726 std::vector<std::pair<std::unique_ptr<char[]>, size_t>> tensor_proto_strs; in ReadTensors() local
727 tensor_proto_strs.reserve(num_complex_); in ReadTensors()
729 SnappyUncompress(&metadata, &simple_tensors, &tensor_proto_strs)); in ReadTensors()
738 auto tensor_proto_str = std::move(tensor_proto_strs[complex_index].first); in ReadTensors()
739 size_t tensor_proto_size = tensor_proto_strs[complex_index].second; in ReadTensors()
780 tensor_proto_strs) { in SnappyUncompress() argument
807 tensor_proto_strs->push_back(std::make_pair( in SnappyUncompress()
Dsnapshot_util.h264 tensor_proto_strs);