Searched refs:ordered_tensors (Results 1 – 3 of 3) sorted by relevance
72 std::vector<Tensor> ordered_tensors; in reorder_tensors_like() local73 ordered_tensors.reserve(tensors.size()); in reorder_tensors_like()78 ordered_tensors.push_back(tensors[indices[used++]]); in reorder_tensors_like()80 std::swap(tensors, ordered_tensors); in reorder_tensors_like()
221 std::map<int, Offset<Tensor>> ordered_tensors; in ExportTensors() local261 ordered_tensors[index] = in ExportTensors()267 tensor_vector.reserve(ordered_tensors.size()); in ExportTensors()268 for (const auto& tensor : ordered_tensors) { in ExportTensors()
586 def _reorder_tensors_as(tensors, ordered_tensors): argument605 return tuple(next(type_dict_[tensor.type()]) for tensor in ordered_tensors)