Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc2595 stats.max_outstanding_async_copies = 0; in CalculateAsyncCopyStats()
2621 stats.max_outstanding_async_copies = in CalculateAsyncCopyStats()
2622 std::max(stats.max_outstanding_async_copies, current_copies); in CalculateAsyncCopyStats()
2678 << stats.max_outstanding_async_copies; in RunMemorySpaceAssignment()
Dmemory_space_assignment.h784 int64 max_outstanding_async_copies; member
Dmemory_space_assignment_test.cc73 HloModule* module, int64 max_outstanding_async_copies = -1, in AssignMemorySpace() argument
78 return AssignMemorySpace(module, max_outstanding_async_copies, in AssignMemorySpace()
84 HloModule* module, int64 max_outstanding_async_copies, in AssignMemorySpace() argument
136 options.max_outstanding_prefetches = max_outstanding_async_copies; in AssignMemorySpace()
137 options.max_outstanding_evictions = max_outstanding_async_copies; in AssignMemorySpace()