Searched refs:tensor_vector (Results 1 – 2 of 2) sorted by relevance
264 std::vector<Offset<Tensor>> tensor_vector; in ExportTensors() local265 tensor_vector.reserve(ordered_tensors.size()); in ExportTensors()267 tensor_vector.push_back(tensor.second); in ExportTensors()270 return builder->CreateVector(tensor_vector); in ExportTensors()
539 std::vector<Tensor>& tensor_vector = (*task.output)[task.split_index]; in SplitOutputTensors() local540 tensor_vector[i] = std::move(split_tensor[j]); in SplitOutputTensors()