Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc310 float MemorySpaceAssignmentCostAnalysis::GetAsyncCopyElapsed( in GetAsyncCopyElapsed() function in xla::MemorySpaceAssignmentCostAnalysis
449 float async_copy_elapsed = cost_analysis_.GetAsyncCopyElapsed(shape); in CanAllocateInAlternateMemoryNoCopy()
459 float async_copy_elapsed = cost_analysis_.GetAsyncCopyElapsed(shape); in PreferredEvictionEndTime()
476 float async_copy_elapsed = cost_analysis_.GetAsyncCopyElapsed(shape); in LatestPrefetchStartTime()
511 float async_copy_elapsed = cost_analysis_.GetAsyncCopyElapsed(shape); in PreferredPrefetchStartTime()
553 async_copy_elapsed_ = cost_analysis_.GetAsyncCopyElapsed(shape); in Begin()
691 float async_copy_elapsed = cost_analysis_.GetAsyncCopyElapsed(shape); in ToNoCopyDebugString()
Dmemory_space_assignment.h142 virtual float GetAsyncCopyElapsed(const Shape& shape) const;
Dmemory_space_assignment_test.cc357 float GetAsyncCopyElapsed(const Shape& shape) const override { in GetAsyncCopyElapsed() function in xla::__anonab57fde90111::FakeMemorySpaceAssignmentCostAnalysis