Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_flatten.cpp72 std::vector<Tensor> ordered_tensors; in reorder_tensors_like() local
73 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dexport.cc221 std::map<int, Offset<Tensor>> ordered_tensors; in ExportTensors() local
261 ordered_tensors[index] = in ExportTensors()
267 tensor_vector.reserve(ordered_tensors.size()); in ExportTensors()
268 for (const auto& tensor : ordered_tensors) { in ExportTensors()
/aosp_15_r20/external/pytorch/torch/
H A D_utils.py586 def _reorder_tensors_as(tensors, ordered_tensors): argument
605 return tuple(next(type_dict_[tensor.type()]) for tensor in ordered_tensors)