Home
last modified time | relevance | path

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

/external/pytorch/torch/testing/_internal/
Dcommon_utils.py1435 def allocator_option_enabled_fn(allocator_config, _, option): function
1449 enabled_fn=functools.partial(allocator_option_enabled_fn, option='expandable_segments'),