Home
last modified time | relevance | path

Searched refs:tensor_data_vec (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dlazy_graph_executor.cpp655 std::vector<BackendDataPtr>& tensor_data_vec) { in ExtractIRAndPrepareTensorData() argument
657 tensor_data_vec.reserve(indices.size()); in ExtractIRAndPrepareTensorData()
665 tensor_data_vec.push_back(handle); in ExtractIRAndPrepareTensorData()
676 const std::vector<BackendDataPtr>& tensor_data_vec) { in SetTensorData() argument
693 handle = tensor_data_vec[i]; in SetTensorData()
742 const std::vector<BackendDataPtr>& tensor_data_vec) { in TryRunCachedSync() argument
760 tensor_data_vec); in TryRunCachedSync()
843 std::vector<BackendDataPtr> tensor_data_vec; in SyncTensorsGraphInternal() local
845 tensors, coll.config, coll.indices, ir_values, tensor_data_vec); in SyncTensorsGraphInternal()
850 TryRunCachedSync(tensors, &coll, &po_data, tensor_data_vec); in SyncTensorsGraphInternal()
[all …]
H A Dlazy_graph_executor.h365 const std::vector<torch::lazy::BackendDataPtr>& tensor_data_vec);
372 std::vector<BackendDataPtr>& tensor_data_vec);
378 const std::vector<BackendDataPtr>& tensor_data_vec);
408 const std::vector<BackendDataPtr>& tensor_data_vec);