Searched defs:rz_allocator (Results 1 – 2 of 2) sorted by relevance
41 size_t scratch_size_limit, const se::RedzoneAllocator& rz_allocator) { in AutotuneConvImpl()129 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in AutotuneFusedConv() local292 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in AutotuneUnfusedConv() local
41 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator, in WrapRedzoneBestEffort()60 void CheckRedzones(const se::RedzoneAllocator& rz_allocator, in CheckRedzones()