Searched defs:policy_fn (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/utils/ |
| D | checkpoint.py | 1272 def __init__(self, policy_fn, storage): argument
1302 def __init__(self, policy_fn, storage, allow_cache_entry_mutation): argument
1403 def policy_fn(ctx, op, *args, **kwargs): function
|
| /external/pytorch/test/ |
| D | test_autograd.py | 13476 def policy_fn(ctx, op, *args, **kwargs): function
13592 def policy_fn(ctx, op, *args, **kwargs): function
13683 def policy_fn(ctx, op, *args, **kwargs): function
13738 def policy_fn(ctx, op, *args, **kwargs): function
13771 def policy_fn(ctx, op, *args, **kwargs): function
13836 def policy_fn(ctx, op, *args, **kwargs): function
|