Searched refs:simple_tensors (Results 1 – 2 of 2) sorted by relevance
724 std::vector<Tensor> simple_tensors; in ReadTensors() local725 simple_tensors.reserve(num_simple_); in ReadTensors()729 SnappyUncompress(&metadata, &simple_tensors, &tensor_proto_strs)); in ReadTensors()735 read_tensors->push_back(std::move(simple_tensors[simple_index])); in ReadTensors()778 std::vector<Tensor>* simple_tensors, in SnappyUncompress() argument801 simple_tensors->push_back(std::move(simple_tensor)); in SnappyUncompress()
262 std::vector<Tensor>* simple_tensors,