Searched refs:tensor_data_vec (Results 1 – 2 of 2) sorted by relevance
655 std::vector<BackendDataPtr>& tensor_data_vec) { in ExtractIRAndPrepareTensorData() argument657 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() argument693 handle = tensor_data_vec[i]; in SetTensorData()742 const std::vector<BackendDataPtr>& tensor_data_vec) { in TryRunCachedSync() argument760 tensor_data_vec); in TryRunCachedSync()843 std::vector<BackendDataPtr> tensor_data_vec; in SyncTensorsGraphInternal() local845 tensors, coll.config, coll.indices, ir_values, tensor_data_vec); in SyncTensorsGraphInternal()850 TryRunCachedSync(tensors, &coll, &po_data, tensor_data_vec); in SyncTensorsGraphInternal()[all …]
365 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);