Searched defs:scratch_size_limit (Results 1 – 1 of 1) sorted by relevance
41 size_t scratch_size_limit, const se::RedzoneAllocator& rz_allocator) { in AutotuneConvImpl()119 se::DeviceMemory<T> side_input_ptr, int64_t scratch_size_limit) { in AutotuneFusedConv()281 int64_t scratch_size_limit) { in AutotuneUnfusedConv()