Searched refs:tensor_vector (Results 1 – 4 of 4) sorted by relevance
29 std::vector<torch::Tensor> tensor_vector = {t0, t1, t2, t3, t4}; in TEST() local30 c10::List<torch::Tensor> tensor_list(tensor_vector); in TEST()
218 std::vector<float> tensor_vector; in gen_latent_from_file() local222 return tensor_vector; in gen_latent_from_file()228 tensor_vector.push_back(value); in gen_latent_from_file()231 return tensor_vector; in gen_latent_from_file()
266 std::vector<Offset<Tensor>> tensor_vector; in ExportTensors() local267 tensor_vector.reserve(ordered_tensors.size()); in ExportTensors()269 tensor_vector.push_back(tensor.second); in ExportTensors()272 return builder->CreateVector(tensor_vector); in ExportTensors()
646 std::vector<Tensor>& tensor_vector = (*task.output)[task.split_index]; in SplitOutputTensors() local647 tensor_vector[i] = std::move(split_tensor[j]); in SplitOutputTensors()