Searching.
- torch.distributed.algorithms.ddp_comm_hooks (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.default_hooks (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook (Python module, in Distributed communication package - torch.distributed)
- torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks (Python module, in Distributed communication package - torch.distributed)
- torch.utils.hooks (Python module, in torch.utils)
- Autograd mechanics
- DDP Communication Hooks
- Modules
- PyTorch 2.0 NNModule Support
- torch.autograd.graph.disable_saved_tensors_hooks (Python class, in Automatic differentiation package - torch.autograd)
- torch.autograd.graph.Node.register_hook (Python method, in torch.autograd.graph.Node.register_hook)
- torch.autograd.graph.Node.register_prehook (Python method, in torch.autograd.graph.Node.register_prehook)
- torch.autograd.graph.register_multi_grad_hook (Python class, in Automatic differentiation package - torch.autograd)
- torch.autograd.graph.saved_tensors_hooks (Python class, in Automatic differentiation package - torch.autograd)
- torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks.noop_hook (Python function, in DDP Communication Hooks)
- torch.distributed.algorithms.ddp_comm_hooks.default_hooks.allreduce_hook (Python function, in DDP Communication Hooks)
- torch.distributed.algorithms.ddp_comm_hooks.default_hooks.bf16_compress_hook (Python function, in DDP Communication Hooks)
- torch.distributed.algorithms.ddp_comm_hooks.default_hooks.fp16_compress_hook (Python function, in DDP Communication Hooks)
- torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.batched_powerSGD_hook (Python function, in DDP Communication Hooks)
- torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.powerSGD_hook (Python function, in DDP Communication Hooks)
- torch.distributed.algorithms.Joinable.join_hook (Python method, in Generic Join Context Manager)
- torch.distributed.algorithms.JoinHook (Python class, in Generic Join Context Manager)
- torch.distributed.algorithms.JoinHook.main_hook (Python method, in Generic Join Context Manager)
- torch.distributed.algorithms.JoinHook.post_hook (Python method, in Generic Join Context Manager)
- torch.distributed.fsdp.FullyShardedDataParallel.register_comm_hook (Python method, in FullyShardedDataParallel)
- torch.distributed.optim.ZeroRedundancyOptimizer.join_hook (Python method, in Distributed Optimizers)
- torch.export.graph_signature.ExportGraphSignature.get_replace_hook (Python method, in torch.export)
- torch.jit.ScriptModule.register_backward_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_forward_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_forward_pre_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_full_backward_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_full_backward_pre_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_load_state_dict_post_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_load_state_dict_pre_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_state_dict_post_hook (Python method, in ScriptModule)
- torch.jit.ScriptModule.register_state_dict_pre_hook (Python method, in ScriptModule)
- torch.nn.Module.register_backward_hook (Python method, in Module)
- torch.nn.Module.register_forward_hook (Python method, in Module)
- torch.nn.Module.register_forward_pre_hook (Python method, in Module)
- torch.nn.Module.register_full_backward_hook (Python method, in Module)
- torch.nn.Module.register_full_backward_pre_hook (Python method, in Module)
- torch.nn.Module.register_load_state_dict_post_hook (Python method, in Module)
- torch.nn.Module.register_load_state_dict_pre_hook (Python method, in Module)
- torch.nn.Module.register_state_dict_post_hook (Python method, in Module)
- torch.nn.Module.register_state_dict_pre_hook (Python method, in Module)
- torch.nn.modules.module.register_module_backward_hook (Python function, in torch.nn.modules.module.register_module_backward_hook)
- torch.nn.modules.module.register_module_buffer_registration_hook (Python function, in torch.nn.modules.module.register_module_buffer_registration_hook)
- torch.nn.modules.module.register_module_forward_hook (Python function, in torch.nn.modules.module.register_module_forward_hook)
- torch.nn.modules.module.register_module_forward_pre_hook (Python function, in torch.nn.modules.module.register_module_forward_pre_hook)
- torch.nn.modules.module.register_module_full_backward_hook (Python function, in torch.nn.modules.module.register_module_full_backward_hook)
- torch.nn.modules.module.register_module_full_backward_pre_hook (Python function, in torch.nn.modules.module.register_module_full_backward_pre_hook)
- torch.nn.modules.module.register_module_module_registration_hook (Python function, in torch.nn.modules.module.register_module_module_registration_hook)
- torch.nn.modules.module.register_module_parameter_registration_hook (Python function, in torch.nn.modules.module.register_module_parameter_registration_hook)
- torch.nn.parallel.DistributedDataParallel.join_hook (Python method, in DistributedDataParallel)
- torch.nn.parallel.DistributedDataParallel.register_comm_hook (Python method, in DistributedDataParallel)
- torch.optim.Adadelta.register_load_state_dict_post_hook (Python method, in Adadelta)
- torch.optim.Adadelta.register_load_state_dict_pre_hook (Python method, in Adadelta)
- torch.optim.Adadelta.register_state_dict_post_hook (Python method, in Adadelta)
- torch.optim.Adadelta.register_state_dict_pre_hook (Python method, in Adadelta)
- torch.optim.Adadelta.register_step_post_hook (Python method, in Adadelta)
- torch.optim.Adadelta.register_step_pre_hook (Python method, in Adadelta)
- torch.optim.Adafactor.register_load_state_dict_post_hook (Python method, in Adafactor)
- torch.optim.Adafactor.register_load_state_dict_pre_hook (Python method, in Adafactor)
- torch.optim.Adafactor.register_state_dict_post_hook (Python method, in Adafactor)
- torch.optim.Adafactor.register_state_dict_pre_hook (Python method, in Adafactor)
- torch.optim.Adafactor.register_step_post_hook (Python method, in Adafactor)
- torch.optim.Adafactor.register_step_pre_hook (Python method, in Adafactor)
- torch.optim.Adagrad.register_load_state_dict_post_hook (Python method, in Adagrad)
- torch.optim.Adagrad.register_load_state_dict_pre_hook (Python method, in Adagrad)
- torch.optim.Adagrad.register_state_dict_post_hook (Python method, in Adagrad)
- torch.optim.Adagrad.register_state_dict_pre_hook (Python method, in Adagrad)
- torch.optim.Adagrad.register_step_post_hook (Python method, in Adagrad)
- torch.optim.Adagrad.register_step_pre_hook (Python method, in Adagrad)
- torch.optim.Adam.register_load_state_dict_post_hook (Python method, in Adam)
- torch.optim.Adam.register_load_state_dict_pre_hook (Python method, in Adam)
- torch.optim.Adam.register_state_dict_post_hook (Python method, in Adam)
- torch.optim.Adam.register_state_dict_pre_hook (Python method, in Adam)
- torch.optim.Adam.register_step_post_hook (Python method, in Adam)
- torch.optim.Adam.register_step_pre_hook (Python method, in Adam)
- torch.optim.Adamax.register_load_state_dict_post_hook (Python method, in Adamax)
- torch.optim.Adamax.register_load_state_dict_pre_hook (Python method, in Adamax)
- torch.optim.Adamax.register_state_dict_post_hook (Python method, in Adamax)
- torch.optim.Adamax.register_state_dict_pre_hook (Python method, in Adamax)
- torch.optim.Adamax.register_step_post_hook (Python method, in Adamax)
- torch.optim.Adamax.register_step_pre_hook (Python method, in Adamax)
- torch.optim.AdamW.register_load_state_dict_post_hook (Python method, in AdamW)
- torch.optim.AdamW.register_load_state_dict_pre_hook (Python method, in AdamW)
- torch.optim.AdamW.register_state_dict_post_hook (Python method, in AdamW)
- torch.optim.AdamW.register_state_dict_pre_hook (Python method, in AdamW)
- torch.optim.AdamW.register_step_post_hook (Python method, in AdamW)
- torch.optim.AdamW.register_step_pre_hook (Python method, in AdamW)
- torch.optim.ASGD.register_load_state_dict_post_hook (Python method, in ASGD)
- torch.optim.ASGD.register_load_state_dict_pre_hook (Python method, in ASGD)
- torch.optim.ASGD.register_state_dict_post_hook (Python method, in ASGD)
- torch.optim.ASGD.register_state_dict_pre_hook (Python method, in ASGD)
- torch.optim.ASGD.register_step_post_hook (Python method, in ASGD)
- torch.optim.ASGD.register_step_pre_hook (Python method, in ASGD)
- torch.optim.LBFGS.register_load_state_dict_post_hook (Python method, in LBFGS)
- torch.optim.LBFGS.register_load_state_dict_pre_hook (Python method, in LBFGS)
- torch.optim.LBFGS.register_state_dict_post_hook (Python method, in LBFGS)
- torch.optim.LBFGS.register_state_dict_pre_hook (Python method, in LBFGS)
- torch.optim.LBFGS.register_step_post_hook (Python method, in LBFGS)
- torch.optim.LBFGS.register_step_pre_hook (Python method, in LBFGS)
- torch.optim.NAdam.register_load_state_dict_post_hook (Python method, in NAdam)
- torch.optim.NAdam.register_load_state_dict_pre_hook (Python method, in NAdam)
- torch.optim.NAdam.register_state_dict_post_hook (Python method, in NAdam)
- torch.optim.NAdam.register_state_dict_pre_hook (Python method, in NAdam)
- torch.optim.NAdam.register_step_post_hook (Python method, in NAdam)
- torch.optim.NAdam.register_step_pre_hook (Python method, in NAdam)
- torch.optim.Optimizer.register_load_state_dict_post_hook (Python method, in torch.optim.Optimizer.register_load_state_dict_post_hook)
- torch.optim.Optimizer.register_load_state_dict_pre_hook (Python method, in torch.optim.Optimizer.register_load_state_dict_pre_hook)
- torch.optim.Optimizer.register_state_dict_post_hook (Python method, in torch.optim.Optimizer.register_state_dict_post_hook)
- torch.optim.Optimizer.register_state_dict_pre_hook (Python method, in torch.optim.Optimizer.register_state_dict_pre_hook)
- torch.optim.Optimizer.register_step_post_hook (Python method, in torch.optim.Optimizer.register_step_post_hook)
- torch.optim.Optimizer.register_step_pre_hook (Python method, in torch.optim.Optimizer.register_step_pre_hook)
- torch.optim.RAdam.register_load_state_dict_post_hook (Python method, in RAdam)
- torch.optim.RAdam.register_load_state_dict_pre_hook (Python method, in RAdam)
- torch.optim.RAdam.register_state_dict_post_hook (Python method, in RAdam)
- torch.optim.RAdam.register_state_dict_pre_hook (Python method, in RAdam)
- torch.optim.RAdam.register_step_post_hook (Python method, in RAdam)
- torch.optim.RAdam.register_step_pre_hook (Python method, in RAdam)
- torch.optim.RMSprop.register_load_state_dict_post_hook (Python method, in RMSprop)
- torch.optim.RMSprop.register_load_state_dict_pre_hook (Python method, in RMSprop)
- torch.optim.RMSprop.register_state_dict_post_hook (Python method, in RMSprop)
- torch.optim.RMSprop.register_state_dict_pre_hook (Python method, in RMSprop)
- torch.optim.RMSprop.register_step_post_hook (Python method, in RMSprop)
- torch.optim.RMSprop.register_step_pre_hook (Python method, in RMSprop)
- torch.optim.Rprop.register_load_state_dict_post_hook (Python method, in Rprop)
- torch.optim.Rprop.register_load_state_dict_pre_hook (Python method, in Rprop)
- torch.optim.Rprop.register_state_dict_post_hook (Python method, in Rprop)
- torch.optim.Rprop.register_state_dict_pre_hook (Python method, in Rprop)
- torch.optim.Rprop.register_step_post_hook (Python method, in Rprop)
- torch.optim.Rprop.register_step_pre_hook (Python method, in Rprop)
- torch.optim.SGD.register_load_state_dict_post_hook (Python method, in SGD)
- torch.optim.SGD.register_load_state_dict_pre_hook (Python method, in SGD)
- torch.optim.SGD.register_state_dict_post_hook (Python method, in SGD)
- torch.optim.SGD.register_state_dict_pre_hook (Python method, in SGD)
- torch.optim.SGD.register_step_post_hook (Python method, in SGD)
- torch.optim.SGD.register_step_pre_hook (Python method, in SGD)
- torch.optim.SparseAdam.register_load_state_dict_post_hook (Python method, in SparseAdam)
- torch.optim.SparseAdam.register_load_state_dict_pre_hook (Python method, in SparseAdam)
- torch.optim.SparseAdam.register_state_dict_post_hook (Python method, in SparseAdam)
- torch.optim.SparseAdam.register_state_dict_pre_hook (Python method, in SparseAdam)
- torch.optim.SparseAdam.register_step_post_hook (Python method, in SparseAdam)
- torch.optim.SparseAdam.register_step_pre_hook (Python method, in SparseAdam)
- torch.optim.swa_utils.AveragedModel.register_backward_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_forward_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_forward_pre_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_full_backward_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_full_backward_pre_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_load_state_dict_post_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_load_state_dict_pre_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_state_dict_post_hook (Python method, in AveragedModel)
- torch.optim.swa_utils.AveragedModel.register_state_dict_pre_hook (Python method, in AveragedModel)
- torch.package.PackageExporter.register_extern_hook (Python method, in torch.package)
- torch.package.PackageExporter.register_intern_hook (Python method, in torch.package)
- torch.package.PackageExporter.register_mock_hook (Python method, in torch.package)