Searched refs:get_async_copy_elapsed_override_ (Results 1 – 1 of 1) sorted by relevance
358 if (get_async_copy_elapsed_override_) { in GetAsyncCopyElapsed()359 return get_async_copy_elapsed_override_(shape); in GetAsyncCopyElapsed()377 get_async_copy_elapsed_override_ = function; in SetOverrideForGetAsyncCopyElapsed()398 std::function<float(const Shape&)> get_async_copy_elapsed_override_ = nullptr; member in xla::__anona324b8620111::FakeMemorySpaceAssignmentCostAnalysis