Searched refs:SyncTensorCollection (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/lazy/core/ |
| D | lazy_graph_executor.h | 148 struct SyncTensorCollection { struct
149 SyncTensorCollection() : hash(0) {} in SyncTensorCollection() function
316 SyncTensorCollection* coll,
336 virtual void TensorCollectionBarrier(SyncTensorCollection* coll);
341 SyncTensorCollection* coll);
353 SyncTensorCollection CollectSyncTensors(
376 SyncTensorCollection* coll,
383 const SyncTensorCollection& coll,
398 SyncTensorCollection* coll,
405 SyncTensorCollection* coll,
|
| D | lazy_graph_executor.cpp | 537 SyncTensorCollection* coll, in Async()
574 LazyGraphExecutor::SyncTensorCollection LazyGraphExecutor::CollectSyncTensors( in CollectSyncTensors()
581 SyncTensorCollection coll; in CollectSyncTensors()
707 SyncTensorCollection* coll) { in RunPostOrder()
740 SyncTensorCollection* coll, in TryRunCachedSync()
766 const SyncTensorCollection& coll, in Compile()
833 SyncTensorCollection coll = CollectSyncTensors(*tensors, config); in SyncTensorsGraphInternal()
881 SyncTensorCollection* coll, in ScheduleSyncTensorsGraph()
945 SyncTensorCollection* coll, in ScheduleSyncTensorsGraph()
1045 void LazyGraphExecutor::TensorCollectionBarrier(SyncTensorCollection* coll) { in TensorCollectionBarrier()
|