Searched refs:next_functions (Results 1 – 14 of 14) sorted by relevance
| /external/pytorch/torch/autograd/ |
| D | graph.py | 63 def next_functions(self) -> Tuple[Tuple[Optional["Node"], int], ...]: member in Node
162 return t.view_as(t).grad_fn.next_functions[0][0]
731 for fn, _idx in node.next_functions:
|
| /external/pytorch/torch/distributed/optim/ |
| D | apply_optimizer_in_backward.py | 65 param_to_acc_grad_map[param] = param.view_as(param).grad_fn.next_functions[
|
| /external/pytorch/torch/distributed/pipelining/ |
| D | _backward.py | 30 return grad_fn.next_functions[0][0]
94 for fn, idx in node.next_functions:
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | logging_utils.py | 86 for fn, _idx in node.next_functions:
|
| /external/pytorch/test/jit/ |
| D | test_autodiff.py | 36 backward_fn = s.grad_fn.next_functions[0][0]
|
| /external/pytorch/torch/testing/_internal/distributed/rpc/ |
| D | dist_autograd_test.py | 558 …next_funcs = next(iter(dist_autograd._current_context()._send_functions().values())).next_functions
913 next_funcs = send_function.next_functions
956 next_funcs = send_function.next_functions
962 next_funcs = add_backward_fn.next_functions
992 next_funcs = next(iter(send_functions.values())).next_functions
1005 next_funcs = list(send_functions.values())[1].next_functions
2481 accumulate_grad_0 = a.grad_fn.next_functions[0][0]
2485 accumulate_grad_1 = a.grad_fn.next_functions[1][0]
|
| /external/pytorch/docs/source/ |
| D | autograd.rst | 328 graph.Node.next_functions
343 …esult', '_register_hook_dict', '_saved_result', 'metadata', 'name', 'next_functions', 'register_ho…
|
| /external/pytorch/test/distributed/_composable/ |
| D | test_checkpoint.py | 100 for next_fn, _ in fn.next_functions:
|
| /external/pytorch/test/distributed/fsdp/ |
| D | test_checkpoint_wrapper.py | 379 for next_grad_fn, _ in grad_fn.next_functions:
|
| /external/pytorch/test/ |
| D | test_autograd.py | 100 next_functions = fn.next_functions
101 for next_fn, _ in next_functions:
104 if next_functions:
619 return t.clone().grad_fn.next_functions[0][0]
1398 acc_grad = tensor.view_as(tensor).grad_fn.next_functions[0][0]
1444 acc = b.grad_fn.next_functions[0][0]
1503 acc = b.grad_fn.next_functions[0][0]
1521 acc = c.grad_fn.next_functions[0][0]
1555 acc = b.grad_fn.next_functions[0][0]
3273 next_functions = a.grad_fn.next_functions
[all …]
|
| /external/pytorch/torch/nn/modules/ |
| D | module.py | 1565 next_functions = {n[0] for n in grad_fn.next_functions}
1567 if inputs_grad_fn != next_functions:
|
| /external/pytorch/torch/nn/parallel/ |
| D | distributed.py | 1120 grad_acc = tmp.grad_fn.next_functions[0][0]
|
| /external/pytorch/torch/distributed/fsdp/ |
| D | _runtime_utils.py | 1453 acc_grad = temp_flat_param.grad_fn.next_functions[0][0] # type: ignore[union-attr]
|
| /external/pytorch/torch/_C/ |
| D | __init__.pyi.in | 242 next_functions: Tuple[Tuple[Any, _int], ...]
|