Searched refs:release_checker (Results 1 – 2 of 2) sorted by relevance
190 const std::function<bool(const xla::ShapeIndex&)>& release_checker);
542 const std::function<bool(const xla::ShapeIndex&)>& release_checker) { in ToDeviceMemoryTree() argument545 if (!release_checker(buffer.first)) { in ToDeviceMemoryTree()