Skip to content

torch 2.10.0

Version history | Download JSON

  • PyPI https://pypi.org/project/torch/
    Repository https://github.com/pytorch/pytorch

  • py.typed

  • Coverage

    %%{init: {"pie": {"textPosition": 0.85}, "theme": "neutral", "themeVariables": {"pieStrokeWidth": "1px"}}}%%
pie title
    "Typed" : 13313
    "Any" : 318
    "Untyped" : 10376
    • 56.8% coverage
    • 55.5% coverage (strict)
    • 24007 typable
      • 13313 typed
      • 10376 untyped
      • 318 Any

  • Typables

    %%{init: {"pie": {"textPosition": 0.85}, "theme": "neutral", "themeVariables": {"pieStrokeWidth": "1px"}}}%%
pie title
    "functions" : 6918
    "classes" : 14477
    "other" : 422
    • 1922 functions (+52 overloads)
      • 4996 parameters
    • 1287 classes
      • 5319 methods (+25 overloads)
        • 8670 parameters
      • 467 properties
    • 657 modules
      • 422 attrs

Modules

Module Coverage Coverage (strict) Typables Ignores
torch._C._VariableFunctions 100.0% 100.0% 151 0
torch._C 100.0% 100.0% 83 6
torch._C._autograd 100.0% 100.0% 1 0
torch._C._distributed 100.0% 100.0% 5 0
torch._C._distributed_c10d 93.6% 93.2% 236 4
torch._C._nn 100.0% 100.0% 59 0
torch._C._profiler 0.0% 0.0% 0 0
torch 37.4% 36.0% 211 3
torch._functorch.apis 92.9% 92.9% 14 0
torch._functorch.batch_norm_replacement 100.0% 100.0% 2 0
torch._functorch.eager_transforms 63.6% 57.6% 33 0
torch._functorch.functional_call 88.9% 88.9% 9 0
torch._higher_order_ops.cond 100.0% 80.0% 5 0
torch._jit_internal 38.9% 38.9% 18 1
torch._library.custom_ops 100.0% 100.0% 7 0
torch._library.infer_schema 75.0% 75.0% 4 0
torch._library.triton 100.0% 85.7% 7 0
torch._lobpcg 100.0% 100.0% 15 1
torch._lowrank 100.0% 100.0% 10 0
torch._tensor 32.1% 31.4% 156 5
torch._tensor_str 0.0% 0.0% 7 0
torch.accelerator 83.3% 83.3% 24 0
torch.accelerator.memory 100.0% 100.0% 17 0
torch.amp.autocast_mode 42.9% 32.1% 28 2
torch.amp.grad_scaler 100.0% 92.6% 54 0
torch.ao.nn.intrinsic.modules.fused 0.0% 0.0% 64 2
torch.ao.nn.intrinsic.qat.modules.conv_fused 25.3% 25.3% 154 11
torch.ao.nn.intrinsic.qat.modules.linear_fused 0.0% 0.0% 25 0
torch.ao.nn.intrinsic.qat.modules.linear_relu 100.0% 100.0% 11 0
torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu 100.0% 92.3% 13 1
torch.ao.nn.intrinsic.quantized.modules.bn_relu 0.0% 0.0% 32 2
torch.ao.nn.intrinsic.quantized.modules.conv_add 0.0% 0.0% 46 6
torch.ao.nn.intrinsic.quantized.modules.conv_relu 0.0% 0.0% 66 6
torch.ao.nn.intrinsic.quantized.modules.linear_relu 12.8% 12.8% 47 3
torch.ao.nn.qat.dynamic.modules.linear 100.0% 100.0% 7 0
torch.ao.nn.qat.modules.conv 62.1% 62.1% 58 3
torch.ao.nn.qat.modules.embedding_ops 9.1% 9.1% 44 0
torch.ao.nn.qat.modules.linear 6.7% 6.7% 15 0
torch.ao.nn.quantizable.modules.activation 60.9% 60.9% 46 0
torch.ao.nn.quantizable.modules.rnn 38.4% 38.4% 73 0
torch.ao.nn.quantized.dynamic.modules.conv 34.8% 34.8% 112 0
torch.ao.nn.quantized.dynamic.modules.linear 0.0% 0.0% 23 1
torch.ao.nn.quantized.dynamic.modules.rnn 32.7% 32.7% 220 0
torch.ao.nn.quantized.functional 34.6% 34.6% 136 0
torch.ao.nn.quantized.modules 10.5% 10.5% 19 0
torch.ao.nn.quantized.modules.activation 13.9% 13.9% 108 0
torch.ao.nn.quantized.modules.batchnorm 17.9% 17.9% 28 2
torch.ao.nn.quantized.modules.conv 26.8% 26.8% 168 8
torch.ao.nn.quantized.modules.dropout 0.0% 0.0% 10 0
torch.ao.nn.quantized.modules.embedding_ops 40.0% 40.0% 75 0
torch.ao.nn.quantized.modules.functional_modules 70.7% 70.7% 92 0
torch.ao.nn.quantized.modules.linear 13.1% 13.1% 61 0
torch.ao.nn.quantized.modules.normalization 4.4% 4.4% 113 0
torch.ao.nn.quantized.modules.rnn 100.0% 66.7% 6 1
torch.ao.nn.quantized.modules.utils 0.0% 0.0% 4 0
torch.ao.nn.quantized.reference.modules.conv 33.3% 33.3% 120 6
torch.ao.nn.quantized.reference.modules.linear 100.0% 100.0% 13 0
torch.ao.nn.quantized.reference.modules.rnn 43.9% 43.9% 114 0
torch.ao.nn.quantized.reference.modules.sparse 68.3% 68.3% 41 0
torch.ao.nn.quantized.reference.modules.utils 0.0% 0.0% 17 0
torch.ao.nn.sparse.quantized.dynamic.linear 18.9% 18.9% 37 0
torch.ao.nn.sparse.quantized.linear 17.4% 17.4% 69 0
torch.ao.nn.sparse.quantized.utils 85.7% 85.7% 14 0
torch.ao.ns.fx.graph_matcher 100.0% 100.0% 5 0
torch.ao.ns.fx.graph_passes 100.0% 100.0% 15 0
torch.ao.ns.fx.mappings 100.0% 100.0% 10 0
torch.ao.ns.fx.n_shadows_utils 80.0% 73.3% 45 0
torch.ao.ns.fx.ns_types 100.0% 100.0% 1 0
torch.ao.ns.fx.pattern_utils 100.0% 100.0% 10 0
torch.ao.ns.fx.qconfig_multi_mapping 96.4% 96.4% 28 0
torch.ao.ns.fx.utils 95.2% 95.2% 42 0
torch.ao.ns.fx.weight_utils 100.0% 100.0% 33 0
torch.ao.pruning.scheduler.base_scheduler 0.0% 0.0% 23 0
torch.ao.pruning.scheduler.cubic_scheduler 0.0% 0.0% 24 0
torch.ao.pruning.scheduler.lambda_scheduler 75.0% 75.0% 8 0
torch.ao.pruning.sparsifier.base_sparsifier 59.0% 59.0% 39 0
torch.ao.pruning.sparsifier.nearly_diagonal_sparsifier 14.3% 14.3% 7 0
torch.ao.pruning.sparsifier.utils 69.6% 69.6% 23 0
torch.ao.pruning.sparsifier.weight_norm_sparsifier 12.5% 12.5% 32 0
torch.ao.quantization 14.3% 14.3% 7 2
torch.ao.quantization.backend_config.backend_config 89.7% 89.7% 68 0
torch.ao.quantization.backend_config.executorch 100.0% 100.0% 1 0
torch.ao.quantization.backend_config.fbgemm 100.0% 100.0% 1 0
torch.ao.quantization.backend_config.native 18.2% 18.2% 11 0
torch.ao.quantization.backend_config.onednn 100.0% 100.0% 1 0
torch.ao.quantization.backend_config.qnnpack 100.0% 100.0% 1 0
torch.ao.quantization.backend_config.tensorrt 50.0% 50.0% 2 0
torch.ao.quantization.backend_config.utils 87.5% 79.2% 24 0
torch.ao.quantization.backend_config.x86 100.0% 100.0% 1 0
torch.ao.quantization.fake_quantize 21.7% 19.3% 83 3
torch.ao.quantization.fuse_modules 0.0% 0.0% 16 0
torch.ao.quantization.fuser_method_mappings 8.7% 8.7% 23 0
torch.ao.quantization.fx.convert 100.0% 100.0% 31 0
torch.ao.quantization.fx.custom_config 95.0% 95.0% 60 0
torch.ao.quantization.fx.fuse 100.0% 100.0% 5 0
torch.ao.quantization.fx.fuse_handler 91.7% 91.7% 24 0
torch.ao.quantization.fx.graph_module 34.3% 34.3% 35 0
torch.ao.quantization.fx.lower_to_fbgemm 100.0% 100.0% 5 0
torch.ao.quantization.fx.lower_to_qnnpack 100.0% 100.0% 4 0
torch.ao.quantization.fx.pattern_utils 75.0% 75.0% 4 0
torch.ao.quantization.fx.prepare 100.0% 100.0% 22 0
torch.ao.quantization.fx.quantize_handler 26.7% 26.7% 15 0
torch.ao.quantization.fx.tracer 50.0% 50.0% 10 0
torch.ao.quantization.fx.utils 96.1% 86.3% 51 0
torch.ao.quantization.observer 29.8% 29.8% 285 8
torch.ao.quantization.pt2e._numeric_debugger 100.0% 100.0% 9 0
torch.ao.quantization.pt2e.duplicate_dq_pass 100.0% 100.0% 2 0
torch.ao.quantization.pt2e.export_utils 62.5% 62.5% 8 0
torch.ao.quantization.pt2e.graph_utils 63.6% 63.6% 11 0
torch.ao.quantization.pt2e.lowering 100.0% 100.0% 3 0
torch.ao.quantization.pt2e.port_metadata_pass 100.0% 100.0% 2 0
torch.ao.quantization.pt2e.prepare 80.0% 80.0% 5 0
torch.ao.quantization.pt2e.representation.rewrite 100.0% 100.0% 2 0
torch.ao.quantization.pt2e.utils 100.0% 100.0% 8 0
torch.ao.quantization.qconfig 6.8% 6.8% 44 0
torch.ao.quantization.qconfig_mapping 87.5% 87.5% 32 0
torch.ao.quantization.quant_type 0.0% 0.0% 0 0
torch.ao.quantization.quantization_mappings 100.0% 93.1% 29 0
torch.ao.quantization.quantize 0.0% 0.0% 47 0
torch.ao.quantization.quantize_fx 100.0% 100.0% 33 0
torch.ao.quantization.quantize_jit 0.0% 0.0% 37 0
torch.ao.quantization.quantize_pt2e 100.0% 100.0% 10 0
torch.ao.quantization.quantizer.composable_quantizer 83.3% 83.3% 12 0
torch.ao.quantization.quantizer.embedding_quantizer 100.0% 100.0% 11 0
torch.ao.quantization.quantizer.quantizer 90.9% 90.9% 11 0
torch.ao.quantization.quantizer.x86_inductor_quantizer 87.7% 87.7% 130 0
torch.ao.quantization.quantizer.xnnpack_quantizer 91.5% 91.5% 47 0
torch.ao.quantization.quantizer.xnnpack_quantizer_utils 66.7% 66.7% 12 0
torch.ao.quantization.quantizer.xpu_inductor_quantizer 80.0% 80.0% 10 0
torch.ao.quantization.stubs 70.6% 70.6% 17 0
torch.ao.quantization.utils 42.3% 39.4% 71 0
torch.autograd 100.0% 100.0% 17 1
torch.autograd.anomaly_mode 62.5% 62.5% 16 0
torch.autograd.forward_ad 26.7% 6.7% 15 0
torch.autograd.function 35.4% 18.5% 65 3
torch.autograd.functional 0.0% 0.0% 38 0
torch.autograd.grad_mode 83.0% 51.1% 47 0
torch.autograd.gradcheck 65.4% 65.4% 52 0
torch.autograd.graph 94.4% 94.4% 36 0
torch.autograd.profiler 20.2% 17.7% 124 1
torch.autograd.profiler_legacy 6.5% 6.5% 46 0
torch.autograd.profiler_util 34.4% 34.4% 160 0
torch.autograd.variable 0.0% 0.0% 3 1
torch.backends 0.0% 0.0% 24 0
torch.backends.cpu 100.0% 100.0% 1 0
torch.backends.cuda 38.4% 37.0% 73 0
torch.backends.cudnn 16.1% 16.1% 31 1
torch.backends.cudnn.rnn 0.0% 0.0% 14 1
torch.backends.cusparselt 100.0% 100.0% 3 1
torch.backends.kleidiai 0.0% 0.0% 1 0
torch.backends.mha 100.0% 100.0% 3 0
torch.backends.miopen 16.7% 16.7% 6 0
torch.backends.mkl 0.0% 0.0% 9 0
torch.backends.mkldnn 9.7% 9.7% 31 0
torch.backends.mps 100.0% 100.0% 9 0
torch.backends.nnpack 0.0% 0.0% 5 0
torch.backends.openmp 0.0% 0.0% 1 0
torch.backends.opt_einsum 16.7% 8.3% 12 1
torch.backends.quantized 20.0% 20.0% 10 0
torch.backends.xeon.run_cpu 0.0% 0.0% 7 0
torch.backends.xnnpack 0.0% 0.0% 7 0
torch.compiler 38.8% 38.8% 49 2
torch.compiler.config 100.0% 100.0% 17 0
torch.cpu 74.3% 65.7% 35 0
torch.cpu.amp.autocast_mode 46.2% 23.1% 13 1
torch.cpu.amp.grad_scaler 100.0% 100.0% 6 0
torch.cuda 53.4% 47.5% 118 6
torch.cuda.amp.autocast_mode 33.3% 16.7% 18 1
torch.cuda.amp.common 0.0% 0.0% 1 0
torch.cuda.amp.grad_scaler 100.0% 100.0% 6 0
torch.cuda.gds 80.0% 80.0% 20 0
torch.cuda.graphs 86.5% 86.5% 37 0
torch.cuda.green_contexts 100.0% 100.0% 5 1
torch.cuda.memory 82.6% 82.6% 69 1
torch.cuda.nccl 56.7% 56.7% 30 0
torch.cuda.nvtx 15.4% 15.4% 13 1
torch.cuda.profiler 0.0% 0.0% 7 0
torch.cuda.random 100.0% 100.0% 15 0
torch.cuda.streams 19.5% 19.5% 41 0
torch.cuda.tunable 100.0% 100.0% 35 0
torch.distributed 50.0% 50.0% 6 1
torch.distributed.algorithms.ddp_comm_hooks 25.0% 25.0% 4 0
torch.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook 100.0% 100.0% 10 0
torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks 100.0% 66.7% 3 0
torch.distributed.algorithms.ddp_comm_hooks.default_hooks 100.0% 100.0% 13 0
torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook 18.8% 18.8% 16 0
torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook 25.0% 25.0% 40 0
torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks 85.7% 85.7% 7 0
torch.distributed.algorithms.join 66.7% 62.5% 24 0
torch.distributed.algorithms.model_averaging.averagers 21.4% 21.4% 14 0
torch.distributed.algorithms.model_averaging.hierarchical_model_averager 10.0% 10.0% 10 0
torch.distributed.algorithms.model_averaging.utils 62.5% 62.5% 8 0
torch.distributed.argparse_util 10.5% 10.5% 19 0
torch.distributed.autograd 100.0% 100.0% 6 0
torch.distributed.checkpoint.api 80.0% 80.0% 5 0
torch.distributed.checkpoint.default_planner 88.3% 85.7% 77 0
torch.distributed.checkpoint.filesystem 92.0% 87.4% 87 0
torch.distributed.checkpoint.format_utils 86.7% 86.7% 30 0
torch.distributed.checkpoint.hf_storage 90.2% 90.2% 41 0
torch.distributed.checkpoint.metadata 55.6% 55.6% 9 0
torch.distributed.checkpoint.optimizer 100.0% 100.0% 5 0
torch.distributed.checkpoint.planner 100.0% 100.0% 36 0
torch.distributed.checkpoint.planner_helpers 100.0% 100.0% 4 0
torch.distributed.checkpoint.quantized_hf_storage 90.9% 81.8% 33 0
torch.distributed.checkpoint.staging 81.0% 71.4% 21 0
torch.distributed.checkpoint.state_dict 100.0% 100.0% 31 0
torch.distributed.checkpoint.state_dict_loader 100.0% 100.0% 14 0
torch.distributed.checkpoint.state_dict_saver 100.0% 100.0% 25 1
torch.distributed.checkpoint.stateful 0.0% 0.0% 0 0
torch.distributed.checkpoint.storage 100.0% 84.2% 38 0
torch.distributed.checkpoint.utils 100.0% 83.3% 6 0
torch.distributed.collective_utils 100.0% 93.3% 15 0
torch.distributed.constants 100.0% 100.0% 2 0
torch.distributed.debug 100.0% 100.0% 4 0
torch.distributed.device_mesh 94.5% 94.5% 91 2
torch.distributed.distributed_c10d 63.2% 62.8% 261 0
torch.distributed.elastic.agent.server.api 73.3% 72.1% 86 0
torch.distributed.elastic.agent.server.health_check_server 100.0% 100.0% 13 0
torch.distributed.elastic.agent.server.local_elastic_agent 92.3% 92.3% 26 0
torch.distributed.elastic.control_plane 100.0% 100.0% 1 0
torch.distributed.elastic.events 100.0% 100.0% 15 0
torch.distributed.elastic.events.api 75.0% 75.0% 8 0
torch.distributed.elastic.events.handlers 100.0% 100.0% 2 0
torch.distributed.elastic.metrics 50.0% 50.0% 2 1
torch.distributed.elastic.metrics.api 51.4% 51.4% 37 0
torch.distributed.elastic.multiprocessing 100.0% 100.0% 11 0
torch.distributed.elastic.multiprocessing.api 81.9% 81.9% 105 0
torch.distributed.elastic.multiprocessing.errors 59.1% 59.1% 22 0
torch.distributed.elastic.multiprocessing.errors.error_handler 75.0% 75.0% 16 0
torch.distributed.elastic.multiprocessing.errors.handlers 100.0% 100.0% 1 0
torch.distributed.elastic.multiprocessing.redirects 25.0% 25.0% 8 0
torch.distributed.elastic.multiprocessing.subprocess_handler.handlers 100.0% 100.0% 8 0
torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler 86.7% 86.7% 15 0
torch.distributed.elastic.multiprocessing.tail_log 88.2% 88.2% 17 0
torch.distributed.elastic.rendezvous.api 78.2% 74.5% 55 0
torch.distributed.elastic.rendezvous.c10d_rendezvous_backend 84.2% 78.9% 19 0
torch.distributed.elastic.rendezvous.dynamic_rendezvous 89.9% 89.9% 79 0
torch.distributed.elastic.rendezvous.etcd_rendezvous 9.7% 9.7% 72 1
torch.distributed.elastic.rendezvous.etcd_rendezvous_backend 100.0% 100.0% 17 2
torch.distributed.elastic.rendezvous.etcd_server 73.9% 73.9% 23 1
torch.distributed.elastic.rendezvous.etcd_store 29.6% 29.6% 27 1
torch.distributed.elastic.rendezvous.registry 100.0% 100.0% 2 0
torch.distributed.elastic.rendezvous.static_tcp_rendezvous 56.5% 56.5% 23 0
torch.distributed.elastic.rendezvous.utils 100.0% 100.0% 3 0
torch.distributed.elastic.timer.api 68.2% 61.4% 44 0
torch.distributed.elastic.timer.debug_info_logging 66.7% 66.7% 3 0
torch.distributed.elastic.timer.file_based_local_timer 88.5% 88.5% 52 0
torch.distributed.elastic.timer.local_timer 60.0% 60.0% 25 0
torch.distributed.elastic.utils.api 100.0% 100.0% 6 0
torch.distributed.elastic.utils.data.cycling_iterator 83.3% 83.3% 6 0
torch.distributed.elastic.utils.data.elastic_distributed_sampler 60.0% 60.0% 10 0
torch.distributed.elastic.utils.distributed 72.7% 72.7% 11 0
torch.distributed.elastic.utils.log_level 100.0% 100.0% 1 0
torch.distributed.elastic.utils.logging 100.0% 100.0% 2 0
torch.distributed.elastic.utils.store 73.9% 73.9% 23 0
torch.distributed.flight_recorder.components.builder 100.0% 93.8% 16 2
torch.distributed.flight_recorder.components.config_manager 50.0% 50.0% 4 0
torch.distributed.flight_recorder.components.fr_logger 100.0% 90.9% 11 0
torch.distributed.flight_recorder.components.loader 100.0% 100.0% 5 0
torch.distributed.flight_recorder.components.types 59.1% 57.0% 93 2
torch.distributed.flight_recorder.components.utils 100.0% 100.0% 73 0
torch.distributed.flight_recorder.fr_trace 100.0% 100.0% 2 0
torch.distributed.fsdp._fully_shard._fsdp_api 0.0% 0.0% 0 0
torch.distributed.fsdp._fully_shard._fully_shard 92.1% 87.3% 63 1
torch.distributed.fsdp.api 0.0% 0.0% 0 0
torch.distributed.fsdp.fully_sharded_data_parallel 88.2% 84.9% 152 0
torch.distributed.fsdp.sharded_grad_scaler 91.3% 91.3% 23 0
torch.distributed.fsdp.wrap 79.5% 72.7% 44 2
torch.distributed.launch 0.0% 0.0% 6 0
torch.distributed.launcher.api 60.0% 60.0% 10 0
torch.distributed.nn.api.remote_module 80.0% 80.0% 5 19
torch.distributed.nn.functional 0.0% 0.0% 39 0
torch.distributed.nn.jit.instantiator 0.0% 0.0% 6 0
torch.distributed.nn.jit.templates.remote_module_template 50.0% 50.0% 2 0
torch.distributed.optim.optimizer 0.0% 0.0% 9 0
torch.distributed.optim.post_localSGD_optimizer 18.8% 18.8% 16 3
torch.distributed.optim.utils 14.3% 14.3% 7 0
torch.distributed.optim.zero_redundancy_optimizer 94.7% 76.3% 38 0
torch.distributed.pipelining._IR 68.8% 68.8% 48 2
torch.distributed.pipelining.microbatch 52.6% 52.6% 19 0
torch.distributed.pipelining.schedules 58.9% 58.9% 146 0
torch.distributed.pipelining.stage 92.6% 92.6% 27 0
torch.distributed.rendezvous 37.5% 37.5% 8 0
torch.distributed.rpc 14.3% 14.3% 7 0
torch.distributed.rpc.api 3.3% 3.3% 30 2
torch.distributed.rpc.backend_registry 0.0% 0.0% 16 2
torch.distributed.rpc.functions 0.0% 0.0% 2 0
torch.distributed.rpc.internal 0.0% 0.0% 5 0
torch.distributed.rpc.options 76.9% 76.9% 13 2
torch.distributed.rpc.rref_proxy 0.0% 0.0% 9 0
torch.distributed.run 33.3% 33.3% 21 0
torch.distributed.tensor._api 72.5% 71.6% 109 5
torch.distributed.tensor.debug._comm_mode 16.1% 16.1% 31 0
torch.distributed.tensor.debug._visualize_sharding 25.0% 25.0% 4 0
torch.distributed.tensor.experimental 100.0% 100.0% 1 0
torch.distributed.tensor.experimental._context_parallel._attention 100.0% 100.0% 5 0
torch.distributed.tensor.experimental._func_map 85.7% 85.7% 7 1
torch.distributed.tensor.experimental._register_sharding 50.0% 50.0% 2 0
torch.distributed.tensor.parallel.api 100.0% 100.0% 5 0
torch.distributed.tensor.parallel.fsdp 84.6% 80.8% 26 0
torch.distributed.tensor.parallel.input_reshard 100.0% 100.0% 4 0
torch.distributed.tensor.parallel.loss 0.0% 0.0% 1 0
torch.distributed.tensor.parallel.style 40.1% 39.5% 147 0
torch.distributed.tensor.placement_types 91.0% 91.0% 145 0
torch.distributions.bernoulli 42.3% 42.3% 26 0
torch.distributions.beta 57.1% 57.1% 21 0
torch.distributions.binomial 42.3% 42.3% 26 0
torch.distributions.categorical 38.5% 38.5% 26 0
torch.distributions.cauchy 47.4% 47.4% 19 0
torch.distributions.chi2 57.1% 57.1% 7 0
torch.distributions.constraint_registry 0.0% 0.0% 8 0
torch.distributions.constraints 6.5% 6.5% 31 11
torch.distributions.continuous_bernoulli 40.0% 40.0% 35 0
torch.distributions.dirichlet 47.4% 47.4% 19 0
torch.distributions.distribution 87.8% 87.8% 41 0
torch.distributions.exp_family 40.0% 40.0% 5 0
torch.distributions.exponential 45.5% 45.5% 22 0
torch.distributions.fishersnedecor 64.3% 64.3% 14 0
torch.distributions.gamma 47.6% 47.6% 21 0
torch.distributions.generalized_pareto 0.0% 0.0% 27 0
torch.distributions.geometric 52.9% 52.9% 17 0
torch.distributions.gumbel 57.1% 57.1% 14 0
torch.distributions.half_cauchy 44.4% 44.4% 18 0
torch.distributions.half_normal 44.4% 44.4% 18 0
torch.distributions.independent 53.8% 53.8% 26 2
torch.distributions.inverse_gamma 71.4% 71.4% 14 0
torch.distributions.kl 50.0% 50.0% 6 0
torch.distributions.kumaraswamy 63.6% 63.6% 11 0
torch.distributions.laplace 50.0% 50.0% 20 0
torch.distributions.lkj_cholesky 33.3% 33.3% 12 0
torch.distributions.log_normal 71.4% 71.4% 14 0
torch.distributions.logistic_normal 70.0% 70.0% 10 0
torch.distributions.lowrank_multivariate_normal 59.1% 59.1% 22 1
torch.distributions.mixture_same_family 37.5% 37.5% 24 0
torch.distributions.multinomial 47.8% 47.8% 23 0
torch.distributions.multivariate_normal 58.3% 58.3% 24 0
torch.distributions.negative_binomial 50.0% 50.0% 24 0
torch.distributions.normal 42.3% 42.3% 26 0
torch.distributions.one_hot_categorical 50.0% 50.0% 26 0
torch.distributions.pareto 100.0% 100.0% 12 0
torch.distributions.poisson 43.8% 43.8% 16 0
torch.distributions.relaxed_bernoulli 61.3% 61.3% 31 0
torch.distributions.relaxed_categorical 61.3% 61.3% 31 0
torch.distributions.studentT 62.5% 62.5% 16 0
torch.distributions.transformed_distribution 33.3% 33.3% 24 1
torch.distributions.transforms 18.2% 18.2% 292 10
torch.distributions.uniform 45.5% 45.5% 22 0
torch.distributions.utils 100.0% 95.5% 22 0
torch.distributions.von_mises 58.8% 58.8% 17 1
torch.distributions.weibull 58.3% 58.3% 12 0
torch.distributions.wishart 46.9% 46.9% 32 0
torch.export 100.0% 100.0% 37 0
torch.export.custom_obj 0.0% 0.0% 0 0
torch.export.custom_ops 0.0% 0.0% 1 0
torch.export.decomp_utils 29.6% 29.6% 27 1
torch.export.dynamic_shapes 37.9% 37.9% 58 4
torch.export.exported_program 48.1% 39.5% 81 0
torch.export.graph_signature 72.4% 72.4% 29 0
torch.export.passes 100.0% 100.0% 3 0
torch.export.pt2_archive._package 89.2% 83.8% 37 2
torch.export.pt2_archive.constants 100.0% 100.0% 23 0
torch.export.unflatten 25.7% 25.7% 70 0
torch.fft 0.0% 0.0% 18 1
torch.functional 43.8% 42.2% 64 0
torch.futures 94.1% 94.1% 17 0
torch.fx._symbolic_trace 67.3% 61.2% 49 1
torch.fx.annotate 0.0% 0.0% 3 0
torch.fx.config 0.0% 0.0% 0 0
torch.fx.experimental.accelerator_partitioner 83.8% 83.8% 74 0
torch.fx.experimental.const_fold 57.1% 57.1% 21 0
torch.fx.experimental.debug 100.0% 100.0% 2 0
torch.fx.experimental.graph_gradual_typechecker 27.5% 27.5% 102 1
torch.fx.experimental.merge_matmul 72.7% 72.7% 11 0
torch.fx.experimental.meta_tracer 15.4% 15.4% 78 2
torch.fx.experimental.migrate_gradual_types.constraint 1.0% 1.0% 210 0
torch.fx.experimental.migrate_gradual_types.constraint_generator 17.2% 17.2% 238 0
torch.fx.experimental.migrate_gradual_types.constraint_transformation 37.4% 37.4% 123 0
torch.fx.experimental.migrate_gradual_types.operation 0.0% 0.0% 0 0
torch.fx.experimental.migrate_gradual_types.transform_to_z3 0.0% 0.0% 33 1
torch.fx.experimental.migrate_gradual_types.util 100.0% 100.0% 11 0
torch.fx.experimental.migrate_gradual_types.z3_types 0.0% 0.0% 5 1
torch.fx.experimental.normalize 77.3% 72.7% 22 0
torch.fx.experimental.optimization 67.9% 67.9% 53 0
torch.fx.experimental.partitioner_utils 71.9% 71.9% 32 0
torch.fx.experimental.proxy_tensor 90.1% 88.9% 81 1
torch.fx.experimental.recording 64.0% 60.0% 25 0
torch.fx.experimental.refinement_types 66.7% 66.7% 9 0
torch.fx.experimental.rewriter 44.4% 44.4% 9 0
torch.fx.experimental.schema_type_annotation 68.2% 68.2% 22 0
torch.fx.experimental.sym_node 28.2% 28.2% 177 3
torch.fx.experimental.symbolic_shapes 99.0% 98.3% 405 7
torch.fx.experimental.unification.core 0.0% 0.0% 7 8
torch.fx.experimental.unification.dispatch 0.0% 0.0% 1 2
torch.fx.experimental.unification.match 0.0% 0.0% 30 3
torch.fx.experimental.unification.more 0.0% 0.0% 9 0
torch.fx.experimental.unification.multipledispatch.conflict 0.0% 0.0% 19 0
torch.fx.experimental.unification.multipledispatch.core 60.0% 40.0% 5 1
torch.fx.experimental.unification.multipledispatch.dispatcher 0.0% 0.0% 71 1
torch.fx.experimental.unification.multipledispatch.utils 0.0% 0.0% 12 1
torch.fx.experimental.unification.multipledispatch.variadic 0.0% 0.0% 9 0
torch.fx.experimental.unification.unification_tools 0.0% 0.0% 56 0
torch.fx.experimental.unification.utils 0.0% 0.0% 14 1
torch.fx.experimental.unification.variable 0.0% 0.0% 13 2
torch.fx.experimental.unify_refinements 0.0% 0.0% 17 1
torch.fx.experimental.validator 82.4% 78.4% 102 1
torch.fx.graph 78.4% 74.6% 134 0
torch.fx.graph_module 43.3% 43.3% 67 0
torch.fx.immutable_collections 100.0% 100.0% 4 3
torch.fx.interpreter 77.0% 63.5% 74 0
torch.fx.node 100.0% 97.5% 79 0
torch.fx.operator_schemas 80.0% 72.0% 25 0
torch.fx.passes.annotate_getitem_nodes 100.0% 100.0% 2 0
torch.fx.passes.backends.cudagraphs 33.3% 33.3% 6 0
torch.fx.passes.dialect.common.cse_pass 33.3% 33.3% 6 0
torch.fx.passes.fake_tensor_prop 40.0% 40.0% 10 0
torch.fx.passes.graph_drawer 84.4% 82.2% 45 0
torch.fx.passes.graph_manipulation 92.9% 85.7% 14 0
torch.fx.passes.graph_transform_observer 40.0% 40.0% 30 0
torch.fx.passes.infra.partitioner 63.9% 63.9% 36 0
torch.fx.passes.infra.pass_base 72.7% 72.7% 11 0
torch.fx.passes.infra.pass_manager 65.5% 65.5% 29 0
torch.fx.passes.net_min_base 0.0% 0.0% 0 0
torch.fx.passes.operator_support 94.7% 94.7% 19 0
torch.fx.passes.param_fetch 100.0% 100.0% 7 0
torch.fx.passes.pass_manager 45.7% 45.7% 35 0
torch.fx.passes.regional_inductor 0.0% 0.0% 3 0
torch.fx.passes.reinplace 0.0% 0.0% 3 0
torch.fx.passes.runtime_assert 100.0% 80.0% 5 0
torch.fx.passes.shape_prop 20.0% 10.0% 10 0
torch.fx.passes.split_module 86.4% 86.4% 22 0
torch.fx.passes.split_utils 46.2% 46.2% 13 0
torch.fx.passes.splitter_base 35.4% 35.4% 48 1
torch.fx.passes.tools_common 68.2% 68.2% 22 0
torch.fx.passes.utils.common 80.0% 80.0% 10 0
torch.fx.passes.utils.fuser_utils 100.0% 100.0% 23 1
torch.fx.passes.utils.matcher_utils 81.6% 76.3% 38 0
torch.fx.passes.utils.matcher_with_name_node_map_utils 90.0% 90.0% 10 0
torch.fx.passes.utils.source_matcher_utils 100.0% 100.0% 7 2
torch.fx.proxy 51.8% 49.1% 112 0
torch.fx.subgraph_rewriter 100.0% 100.0% 12 0
torch.fx.tensor_type 0.0% 0.0% 14 1
torch.fx.traceback 53.3% 53.3% 45 0
torch.hub 42.1% 42.1% 38 1
torch.jit 36.8% 21.1% 19 0
torch.jit._async 0.0% 0.0% 6 0
torch.jit._freeze 71.4% 71.4% 7 0
torch.jit._fuser 50.0% 50.0% 2 0
torch.jit._script 55.6% 48.1% 27 0
torch.jit._serialization 11.1% 11.1% 9 0
torch.jit._trace 0.0% 0.0% 26 1
torch.jit.annotations 5.1% 5.1% 39 0
torch.jit.frontend 2.1% 2.1% 195 0
torch.jit.generate_bytecode 33.3% 33.3% 3 0
torch.jit.mobile 9.1% 9.1% 11 0
torch.jit.quantized 0.0% 0.0% 29 0
torch.jit.unsupported_tensor_ops 25.0% 25.0% 4 0
torch.library 35.8% 34.6% 81 0
torch.linalg 0.0% 0.0% 41 0
torch.masked._ops 100.0% 100.0% 115 0
torch.masked.maskedtensor.core 4.4% 2.2% 45 4
torch.masked.maskedtensor.creation 100.0% 100.0% 7 0
torch.monitor 100.0% 100.0% 4 0
torch.mps 89.5% 89.5% 19 1
torch.mps.event 100.0% 100.0% 9 0
torch.mps.profiler 100.0% 100.0% 11 0
torch.mtia 80.0% 71.1% 45 1
torch.mtia.memory 100.0% 100.0% 8 0
torch.mtia.mtia_graph 81.8% 81.8% 22 0
torch.multiprocessing 0.0% 0.0% 4 0
torch.multiprocessing.pool 0.0% 0.0% 5 0
torch.multiprocessing.queue 0.0% 0.0% 12 0
torch.multiprocessing.reductions 0.9% 0.9% 114 1
torch.multiprocessing.spawn 23.9% 23.9% 46 1
torch.nested 69.0% 69.0% 29 1
torch.nn 0.0% 0.0% 2 0
torch.nn.attention 66.7% 66.7% 3 0
torch.nn.attention._registry 100.0% 100.0% 7 0
torch.nn.attention.bias 68.8% 68.8% 32 1
torch.nn.attention.flex_attention 83.5% 83.5% 91 0
torch.nn.attention.varlen 100.0% 100.0% 10 0
torch.nn.common_types 0.0% 0.0% 0 0
torch.nn.cpp 20.8% 20.8% 24 1
torch.nn.functional 99.0% 97.4% 625 0
torch.nn.grad 0.0% 0.0% 48 0
torch.nn.init 84.7% 84.7% 72 0
torch.nn.modules.activation 82.9% 82.9% 217 0
torch.nn.modules.adaptive 72.4% 72.4% 29 0
torch.nn.modules.batchnorm 54.8% 54.8% 31 6
torch.nn.modules.channelshuffle 100.0% 100.0% 6 0
torch.nn.modules.container 91.7% 85.4% 157 3
torch.nn.modules.conv 82.0% 82.0% 183 13
torch.nn.modules.distance 100.0% 100.0% 18 0
torch.nn.modules.dropout 100.0% 100.0% 12 0
torch.nn.modules.flatten 90.0% 90.0% 20 0
torch.nn.modules.fold 100.0% 100.0% 26 0
torch.nn.modules.instancenorm 66.7% 66.7% 18 3
torch.nn.modules.lazy 4.0% 4.0% 25 0
torch.nn.modules.linear 80.4% 76.1% 46 3
torch.nn.modules.loss 85.4% 85.4% 185 0
torch.nn.modules.module 74.2% 72.3% 213 0
torch.nn.modules.normalization 83.1% 83.1% 71 0
torch.nn.modules.padding 94.7% 94.7% 57 0
torch.nn.modules.pixelshuffle 100.0% 100.0% 12 0
torch.nn.modules.pooling 90.7% 90.7% 150 0
torch.nn.modules.rnn 77.2% 77.2% 158 2
torch.nn.modules.sparse 83.8% 83.8% 74 0
torch.nn.modules.transformer 69.4% 69.4% 157 0
torch.nn.modules.upsampling 91.3% 91.3% 23 0
torch.nn.modules.utils 100.0% 100.0% 3 0
torch.nn.parallel.comm 0.0% 0.0% 27 0
torch.nn.parallel.data_parallel 85.3% 61.8% 34 0
torch.nn.parallel.distributed 9.6% 7.3% 177 0
torch.nn.parallel.parallel_apply 100.0% 100.0% 7 0
torch.nn.parallel.replicate 100.0% 100.0% 4 0
torch.nn.parallel.scatter_gather 100.0% 84.6% 13 0
torch.nn.parameter 100.0% 100.0% 26 0
torch.nn.utils.clip_grad 100.0% 100.0% 26 0
torch.nn.utils.convert_parameters 100.0% 100.0% 5 0
torch.nn.utils.fusion 100.0% 100.0% 24 0
torch.nn.utils.init 0.0% 0.0% 4 0
torch.nn.utils.memory_format 100.0% 100.0% 6 0
torch.nn.utils.parametrizations 93.3% 93.3% 15 0
torch.nn.utils.parametrize 90.0% 90.0% 30 0
torch.nn.utils.prune 8.8% 8.8% 137 6
torch.nn.utils.rnn 100.0% 92.6% 54 0
torch.nn.utils.spectral_norm 67.2% 65.5% 58 0
torch.nn.utils.stateless 85.7% 85.7% 7 0
torch.nn.utils.weight_norm 95.7% 87.0% 23 0
torch.numa.binding 100.0% 100.0% 8 0
torch.onnx 100.0% 100.0% 30 0
torch.onnx._internal.exporter._onnx_program 76.7% 76.7% 30 0
torch.onnx._internal.exporter._testing 100.0% 100.0% 8 0
torch.onnx._internal.exporter._verification 100.0% 100.0% 10 0
torch.onnx.errors 100.0% 100.0% 7 0
torch.onnx.ops 100.0% 100.0% 39 0
torch.optim._adafactor 47.6% 47.6% 21 0
torch.optim._muon 88.2% 88.2% 17 0
torch.optim.adadelta 86.1% 86.1% 36 0
torch.optim.adagrad 74.4% 74.4% 39 0
torch.optim.adam 74.5% 74.5% 47 0
torch.optim.adamax 70.3% 70.3% 37 0
torch.optim.adamw 94.3% 94.3% 35 0
torch.optim.asgd 70.0% 70.0% 40 0
torch.optim.lbfgs 45.8% 45.8% 24 1
torch.optim.lr_scheduler 68.3% 68.3% 240 3
torch.optim.nadam 72.1% 72.1% 43 0
torch.optim.optimizer 98.3% 98.3% 59 2
torch.optim.radam 71.8% 71.8% 39 0
torch.optim.rmsprop 71.4% 71.4% 42 0
torch.optim.rprop 68.6% 68.6% 35 1
torch.optim.sgd 74.3% 74.3% 35 0
torch.optim.sparse_adam 75.0% 75.0% 8 0
torch.optim.swa_utils 44.7% 44.7% 47 0
torch.overrides 50.0% 44.4% 18 0
torch.package.analyze.find_first_use_of_broken_modules 100.0% 100.0% 2 0
torch.package.analyze.is_from_package 100.0% 50.0% 2 0
torch.package.analyze.trace_dependencies 100.0% 100.0% 3 0
torch.package.file_structure_representation 64.7% 64.7% 17 0
torch.package.glob_group 55.6% 55.6% 18 0
torch.package.importer 81.0% 61.9% 21 0
torch.package.package_exporter 59.7% 58.8% 119 0
torch.package.package_importer 44.4% 40.7% 81 0
torch.profiler.itt 0.0% 0.0% 10 1
torch.profiler.profiler 69.9% 69.9% 73 0
torch.quantization 0.0% 0.0% 46 0
torch.quasirandom 33.3% 33.3% 24 0
torch.random 53.8% 53.8% 13 0
torch.return_types 100.0% 100.0% 292 0
torch.serialization 52.2% 42.0% 69 2
torch.signal.windows.windows 97.6% 97.6% 83 0
torch.sparse 20.0% 20.0% 25 1
torch.sparse.semi_structured 74.3% 73.0% 74 5
torch.special 0.0% 0.0% 56 1
torch.storage 10.6% 10.6% 179 2
torch.testing._comparison 91.7% 75.0% 12 1
torch.testing._creation 100.0% 100.0% 10 0
torch.torch_version 100.0% 40.0% 5 0
torch.types 100.0% 93.1% 29 0
torch.utils 0.0% 0.0% 7 0
torch.utils.backcompat 18.2% 18.2% 11 0
torch.utils.backend_registration 100.0% 100.0% 8 0
torch.utils.benchmark.op_fuzzers.binary 25.0% 25.0% 4 0
torch.utils.benchmark.op_fuzzers.sparse_binary 25.0% 25.0% 4 0
torch.utils.benchmark.op_fuzzers.sparse_unary 100.0% 100.0% 4 0
torch.utils.benchmark.op_fuzzers.spectral 80.0% 80.0% 5 0
torch.utils.benchmark.op_fuzzers.unary 25.0% 25.0% 4 0
torch.utils.benchmark.utils.common 100.0% 96.9% 32 0
torch.utils.benchmark.utils.compare 66.7% 66.7% 15 0
torch.utils.benchmark.utils.compile 85.7% 85.7% 14 1
torch.utils.benchmark.utils.cpp_jit 80.0% 80.0% 15 0
torch.utils.benchmark.utils.fuzzer 40.3% 40.3% 62 0
torch.utils.benchmark.utils.sparse_fuzzer 47.6% 47.6% 21 0
torch.utils.benchmark.utils.timer 100.0% 100.0% 45 0
torch.utils.benchmark.utils.valgrind_wrapper.timer_interface 100.0% 92.5% 40 1
torch.utils.bundled_inputs 68.2% 68.2% 22 0
torch.utils.checkpoint 41.5% 41.5% 53 1
torch.utils.collect_env 3.0% 3.0% 66 0
torch.utils.cpp_backtrace 33.3% 33.3% 3 0
torch.utils.cpp_extension 27.6% 27.6% 87 0
torch.utils.data._utils.collate 0.0% 0.0% 2 0
torch.utils.data._utils.worker 100.0% 100.0% 1 1
torch.utils.data.backward_compatibility 50.0% 50.0% 2 0
torch.utils.data.dataloader 60.4% 60.4% 53 0
torch.utils.data.datapipes._decorator 62.9% 45.7% 35 0
torch.utils.data.datapipes.dataframe.dataframe_wrapper 2.6% 2.6% 39 0
torch.utils.data.datapipes.dataframe.dataframes 21.3% 21.3% 164 1
torch.utils.data.datapipes.dataframe.datapipes 17.6% 17.6% 34 0
torch.utils.data.datapipes.dataframe.structures 100.0% 100.0% 2 0
torch.utils.data.datapipes.datapipe 88.1% 73.0% 126 0
torch.utils.data.datapipes.iter.callable 64.7% 64.7% 17 0
torch.utils.data.datapipes.iter.combinatorics 75.0% 75.0% 32 0
torch.utils.data.datapipes.iter.combining 75.0% 75.0% 32 1
torch.utils.data.datapipes.iter.filelister 100.0% 100.0% 15 0
torch.utils.data.datapipes.iter.fileopener 100.0% 100.0% 11 0
torch.utils.data.datapipes.iter.grouping 53.3% 53.3% 45 0
torch.utils.data.datapipes.iter.routeddecoder 90.0% 90.0% 10 0
torch.utils.data.datapipes.iter.selecting 75.0% 75.0% 12 0
torch.utils.data.datapipes.iter.sharding 33.3% 33.3% 15 0
torch.utils.data.datapipes.iter.streamreader 66.7% 66.7% 6 0
torch.utils.data.datapipes.iter.utils 71.4% 71.4% 7 0
torch.utils.data.datapipes.map.callable 70.0% 70.0% 10 0
torch.utils.data.datapipes.map.combinatorics 61.1% 61.1% 18 0
torch.utils.data.datapipes.map.combining 83.3% 83.3% 12 1
torch.utils.data.datapipes.map.grouping 83.3% 83.3% 12 0
torch.utils.data.datapipes.map.utils 100.0% 100.0% 7 0
torch.utils.data.datapipes.utils.common 53.2% 53.2% 47 0
torch.utils.data.datapipes.utils.decoder 10.6% 10.6% 47 0
torch.utils.data.dataset 68.0% 68.0% 50 0
torch.utils.data.distributed 55.0% 55.0% 20 0
torch.utils.data.graph 100.0% 100.0% 5 0
torch.utils.data.graph_settings 93.8% 87.5% 16 0
torch.utils.data.sampler 79.5% 79.5% 44 0
torch.utils.dlpack 100.0% 75.0% 4 0
torch.utils.file_baton 36.4% 36.4% 11 0
torch.utils.flop_counter 38.5% 38.5% 26 0
torch.utils.hipify.constants 0.0% 0.0% 0 0
torch.utils.hipify.cuda_to_hip_mappings 0.0% 0.0% 8 0
torch.utils.hipify.hipify_python 43.4% 43.4% 143 0
torch.utils.hipify.version 0.0% 0.0% 0 0
torch.utils.hooks 33.3% 22.9% 48 0
torch.utils.jit.log_extract 28.1% 28.1% 32 0
torch.utils.mkldnn 14.0% 14.0% 43 0
torch.utils.mobile_optimizer 85.7% 85.7% 7 0
torch.utils.model_dump 0.0% 0.0% 15 0
torch.utils.module_tracker 22.7% 22.7% 22 0
torch.utils.serialization.config 100.0% 100.0% 7 0
torch.utils.show_pickle 17.0% 17.0% 47 4
torch.utils.tensorboard 0.0% 0.0% 0 0
torch.utils.tensorboard.summary 4.0% 4.0% 101 0
torch.utils.tensorboard.writer 20.3% 20.3% 187 0
torch.utils.throughput_benchmark 19.2% 19.2% 26 0
torch.utils.weak 22.0% 22.0% 59 1
torch.version 100.0% 100.0% 4 0
torch.xpu 81.5% 68.5% 54 3
torch.xpu.memory 96.7% 96.7% 30 0
torch.xpu.random 100.0% 100.0% 15 0
torch.xpu.streams 34.5% 34.5% 29 0

Incomplete Annotations

torch._C._distributed_c10d (15 missing, 1 any)

Symbol Typable Typed Any
meth ProcessGroup.allreduce_coalesced 3 2 0
meth ProcessGroup._allgather_base 4 3 0
meth ProcessGroup.allgather_coalesced 4 3 0
meth ProcessGroup.allgather_into_tensor_coalesced 4 3 0
func get_debug_level 1 0 0
meth ProcessGroupGloo.create_device 4 1 0
meth ProcessGroupGloo.create_default_device 2 1 0
meth ProcessGroupGloo._set_default_timeout 2 1 0
meth ProcessGroupNCCL._set_default_timeout 2 1 0
meth ProcessGroupNCCL.get_build_nccl_version 2 1 0
meth ProcessGroupNCCL.get_runtime_nccl_version 2 1 0
meth Work.exception 1 1 1
meth Store.set 3 2 0
meth Store.set_timeout 2 1 0

torch (132 missing, 3 any)

Symbol Typable Typed Any
func sym_not 2 0 0
func sym_ite 4 0 0
meth SymInt.init 2 0 0
meth SymInt.bool 1 0 0
meth SymInt.int 1 0 0
meth SymInt.index 1 0 0
meth SymInt.round 2 0 0
meth SymInt.truediv 2 0 0
meth SymInt.rtruediv 2 0 0
meth SymInt.floordiv 2 0 0
meth SymInt.rfloordiv 2 0 0
meth SymInt.pow 2 0 0
meth SymInt.rpow 2 0 0
meth SymInt.lt 2 1 0
meth SymInt.gt 2 1 0
meth SymInt.le 2 1 0
meth SymInt.ge 2 1 0
meth SymInt.add 2 1 0
meth SymInt.radd 2 1 0
meth SymInt.rmul 2 1 0
meth SymInt.mul 2 1 0
meth SymInt.pow_by_natural 2 1 0
meth SymInt.rpow_by_natural 2 1 0
meth SymInt.int_truediv 2 1 0
meth SymInt.rint_truediv 2 1 0
meth SymInt.int_floordiv 2 1 0
meth SymInt.rint_floordiv 2 1 0
meth SymInt.sym_max 2 0 0
meth SymInt.sym_min 2 0 0
meth SymInt.sym_float 1 0 0
meth SymInt.neg 1 0 0
meth SymInt.and 2 1 0
meth SymInt.or 2 1 0
meth SymInt.repr 1 0 0
meth SymInt.sympy 1 0 0
attr SymInt.node 1 0 0
meth IntStorage.dtype 1 0 0
meth IntStorage.dtype 1 0 0
meth ByteStorage.dtype 1 0 0
meth ByteStorage._dtype 1 0 0
meth LongStorage.dtype 1 0 0
meth LongStorage._dtype 1 0 0
meth CharStorage.dtype 1 0 0
meth CharStorage._dtype 1 0 0
meth BoolStorage.dtype 1 0 0
meth BoolStorage._dtype 1 0 0
func typename 2 2 1
meth SymFloat.__init__ 2 0 0
meth SymFloat.__truediv__ 2 0 0
meth SymFloat.__rtruediv__ 2 0 0
meth SymFloat.__floordiv__ 2 0 0
meth SymFloat.__rfloordiv__ 2 0 0
meth SymFloat.__bool__ 1 0 0
meth SymFloat.__float__ 1 0 0
meth SymFloat.__int__ 1 0 0
meth SymFloat.__pow__ 2 0 0
meth SymFloat.__rpow__ 2 0 0
meth SymFloat.__lt__ 2 1 0
meth SymFloat.__gt__ 2 1 0
meth SymFloat.__le__ 2 1 0
meth SymFloat.__ge__ 2 1 0
meth SymFloat.__float_pow__ 2 1 0
meth SymFloat.__rfloat_pow__ 2 1 0
meth SymFloat.__float_truediv__ 2 1 0
meth SymFloat.__rfloat_truediv__ 2 1 0
meth SymFloat.__trunc__ 1 0 0
meth SymFloat.__sym_max__ 2 0 0
meth SymFloat.__sym_min__ 2 0 0
meth SymFloat.__sym_int__ 1 0 0
meth SymFloat.is_integer 1 0 0
meth SymFloat.__repr__ 1 0 0
meth SymFloat._sympy 1 0 0
meth SymFloat.hash 1 0 0
attr SymFloat.node 1 0 0
func sym_fresh_size 2 0 0
func sym_min 3 0 0
func get_device_module 2 1 0
func sym_int 2 0 0
meth FloatStorage.dtype 1 0 0
meth FloatStorage.dtype 1 0 0
meth ShortStorage.dtype 1 0 0
meth ShortStorage._dtype 1 0 0
func sym_float 2 0 0
func sym_max 3 0 0
meth SymBool.__init__ 2 0 0
meth SymBool.__bool__ 1 0 0
meth SymBool.__int__ 1 0 0
meth SymBool.__and__ 2 1 0
meth SymBool.__or__ 2 1 0
meth SymBool.__sym_ite__ 3 0 0
meth SymBool.__eq__ 2 1 0
meth SymBool.__repr__ 1 0 0
meth SymBool._sympy 1 0 0
meth SymBool.hash 1 0 0
meth SymBool.sym_float 1 0 0
attr SymBool.node 1 0 0
func is_tensor 2 2 1
func sym_sum 2 0 0
func is_storage 2 2 1
meth DoubleStorage.dtype 1 0 0
meth DoubleStorage._dtype 1 0 0

torch._functorch.apis (1 missing, 0 any)

Symbol Typable Typed Any
func vmap 6 5 0

torch._functorch.eager_transforms (12 missing, 2 any)

Symbol Typable Typed Any
func linearize 3 2 0
func hessian 3 0 0
func jacfwd 5 4 0
func jvp 6 5 2
func debug_unwrap 3 2 0
func jacrev 6 3 0
func vjp 4 2 0

torch._functorch.functional_call (1 missing, 0 any)

Symbol Typable Typed Any
func functional_call 7 6 0

torch._higher_order_ops.cond (0 missing, 1 any)

Symbol Typable Typed Any
func cond 5 5 1

torch._jit_internal (11 missing, 0 any)

Symbol Typable Typed Any
func ignore 3 0 0
func is_optional 2 0 0
attr BroadcastingList1 1 0 0
func is_dict 2 1 0
func is_union 2 0 0
func is_tuple 2 1 0
func is_list 2 1 0

torch._library.infer_schema (1 missing, 0 any)

Symbol Typable Typed Any
func infer_schema 4 3 0

torch._library.triton (0 missing, 1 any)

Symbol Typable Typed Any
func wrap_triton 2 2 1

torch._tensor (106 missing, 1 any)

Symbol Typable Typed Any
meth Tensor.clear_non_serializable_cached_data 1 0 0
meth Tensor.__deepcopy__ 2 0 0
meth Tensor.__reduce_ex__ 2 0 0
meth Tensor.storage 1 0 0
meth Tensor._typed_storage 1 0 0
meth Tensor._reduce_ex_internal 2 0 0
meth Tensor.__setstate__ 2 0 0
meth Tensor.__repr__ 2 0 0
meth Tensor.backward 5 0 0
meth Tensor.index 3 0 0
meth Tensor.register_hook 2 0 0
meth Tensor.register_post_accumulate_grad_hook 2 0 0
meth Tensor.reinforce 2 0 0
meth Tensor.is_shared 1 0 0
meth Tensor.share_memory 1 0 0
meth Tensor.module_load 3 0 0
meth Tensor.reversed 1 0 0
meth Tensor.norm 5 1 0
meth Tensor.solve 2 0 0
meth Tensor.lstsq 2 0 0
meth Tensor.eig 2 0 0
meth Tensor.symeig 2 0 0
meth Tensor.lu 3 0 0
meth Tensor.stft 11 10 0
meth Tensor.istft 10 9 0
meth Tensor.resize 2 0 0
meth Tensor.resize_as 2 0 0
meth Tensor.split 3 0 0
meth Tensor.unique 5 0 0
meth Tensor.unique_consecutive 4 0 0
meth Tensor.format 2 0 0
meth Tensor.len 1 0 0
meth Tensor.iter 1 0 0
meth Tensor.hash 1 0 0
meth Tensor.dir 1 0 0
meth Tensor.array 2 0 0
meth Tensor.array_wrap 2 0 0
meth Tensor.contains 2 2 1
meth Tensor.storage_type 1 0 0
meth Tensor.refine_names 2 0 0
meth Tensor.align_to 2 0 0
meth Tensor.unflatten 3 0 0
meth Tensor.rename_ 3 0 0
meth Tensor.rename 3 0 0
meth Tensor.to_sparse_coo 1 0 0
meth Tensor.dim_order 2 1 0
meth Tensor.update_names 3 0 0
meth Tensor.__torch_function__ 5 0 0
meth Tensor.__dlpack__ 5 4 0
prop Tensor.__cuda_array_interface__ 1 0 0
attr Tensor.detach 1 0 0
attr Tensor.detach 1 0 0
attr Tensor.pow 1 0 0
attr Tensor.ipow 1 0 0

torch._tensor_str (7 missing, 0 any)

Symbol Typable Typed Any
func set_printoptions 7 0 0

torch.accelerator (4 missing, 0 any)

Symbol Typable Typed Any
attr set_device_idx 1 0 0
attr current_device_idx 1 0 0
attr device_index.idx 1 0 0
attr device_index.prev_idx 1 0 0

torch.amp.autocast_mode (16 missing, 3 any)

Symbol Typable Typed Any
meth autocast.init 5 4 0
meth autocast.enter 1 0 0
meth autocast.exit 4 3 3
meth autocast.call 2 0 0
attr autocast.fast_dtype 1 0 0
attr autocast.device 1 0 0
attr autocast.custom_backend_name 1 0 0
attr autocast.custom_device_mod 1 0 0
func custom_fwd 4 2 0
func custom_bwd 3 1 0
func autocast_decorator 3 0 0

torch.amp.grad_scaler (0 missing, 4 any)

Symbol Typable Typed Any
meth GradScaler._maybe_opt_step 5 5 2
meth GradScaler.step 4 4 2

torch.ao.nn.intrinsic.modules.fused (64 missing, 0 any)

Symbol Typable Typed Any
meth ConvBnReLU1d.init 4 0 0
meth LinearTanh.init 3 0 0
meth LinearReLU.init 3 0 0
meth ConvReLU3d.init 3 0 0
meth ConvBnReLU2d.init 4 0 0
meth LinearBn1d.init 3 0 0
meth BNReLU2d.init 3 0 0
meth ConvBn3d.init 3 0 0
meth ConvReLU2d.init 3 0 0
meth ConvAdd2d.init 3 0 0
meth ConvAdd2d.forward 3 0 0
attr ConvAdd2d.add 1 0 0
meth ConvBnReLU3d.init 4 0 0
meth ConvAddReLU2d.init 4 0 0
meth ConvAddReLU2d.forward 3 0 0
attr ConvAddReLU2d.add 1 0 0
attr ConvAddReLU2d.relu 1 0 0
meth LinearLeakyReLU.init 3 0 0
meth BNReLU3d.init 3 0 0
meth ConvReLU1d.init 3 0 0
meth ConvBn2d.init 3 0 0
meth ConvBn1d.init 3 0 0

torch.ao.nn.intrinsic.qat.modules.conv_fused (115 missing, 0 any)

Symbol Typable Typed Any
meth ConvBnReLU1d.forward 2 0 0
meth ConvBnReLU1d.from_float 3 0 0
func update_bn_stats 2 0 0
meth ConvBn3d.init 14 0 0
meth ConvReLU2d.init 11 0 0
meth ConvReLU2d.forward 2 0 0
meth ConvReLU2d.from_float 3 0 0
attr ConvReLU2d.qconfig 1 0 0
attr ConvReLU2d.weight_fake_quant 1 0 0
meth ConvBnReLU3d.forward 2 0 0
meth ConvBnReLU3d.from_float 3 0 0
func freeze_bn_stats 2 0 0
meth ConvReLU3d.init 11 0 0
meth ConvReLU3d.forward 2 0 0
meth ConvReLU3d.from_float 3 0 0
attr ConvReLU3d.qconfig 1 0 0
attr ConvReLU3d.weight_fake_quant 1 0 0
meth ConvBnReLU2d.forward 2 0 0
meth ConvBnReLU2d.from_float 3 0 0
meth ConvReLU1d.init 11 0 0
meth ConvReLU1d.forward 2 0 0
meth ConvReLU1d.from_float 3 0 0
attr ConvReLU1d.qconfig 1 0 0
attr ConvReLU1d.weight_fake_quant 1 0 0
meth ConvBn2d.init 14 0 0
meth ConvBn1d.init 14 0 0

torch.ao.nn.intrinsic.qat.modules.linear_fused (25 missing, 0 any)

Symbol Typable Typed Any
meth LinearBn1d.init 8 0 0
meth LinearBn1d.reset_running_stats 1 0 0
meth LinearBn1d.reset_bn_parameters 1 0 0
meth LinearBn1d.update_bn_stats 1 0 0
meth LinearBn1d.freeze_bn_stats 1 0 0
meth LinearBn1d.forward 2 0 0
meth LinearBn1d.train 2 0 0
meth LinearBn1d.from_float 3 0 0
meth LinearBn1d.to_float 1 0 0
attr LinearBn1d.qconfig 1 0 0
attr LinearBn1d.freeze_bn 1 0 0
attr LinearBn1d.bn 1 0 0
attr LinearBn1d.weight_fake_quant 1 0 0
attr LinearBn1d.bias 1 0 0

torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu (0 missing, 1 any)

Symbol Typable Typed Any
meth LinearReLU.from_reference 2 2 1

torch.ao.nn.intrinsic.quantized.modules.bn_relu (32 missing, 0 any)

Symbol Typable Typed Any
meth BNReLU2d.init 6 0 0
meth BNReLU2d.forward 2 0 0
meth BNReLU2d._get_name 1 0 0
meth BNReLU2d.from_float 3 0 0
meth BNReLU2d.from_reference 4 0 0
meth BNReLU3d.init 6 0 0
meth BNReLU3d.forward 2 0 0
meth BNReLU3d._get_name 1 0 0
meth BNReLU3d.from_float 3 0 0
meth BNReLU3d.from_reference 4 0 0

torch.ao.nn.intrinsic.quantized.modules.conv_add (46 missing, 0 any)

Symbol Typable Typed Any
meth ConvAdd2d.init 12 0 0
meth ConvAdd2d.forward 3 0 0
meth ConvAdd2d._get_name 1 0 0
meth ConvAdd2d.from_float 3 0 0
meth ConvAdd2d.from_reference 4 0 0
meth ConvAddReLU2d.init 12 0 0
meth ConvAddReLU2d.forward 3 0 0
meth ConvAddReLU2d._get_name 1 0 0
meth ConvAddReLU2d.from_float 3 0 0
meth ConvAddReLU2d.from_reference 4 0 0

torch.ao.nn.intrinsic.quantized.modules.conv_relu (66 missing, 0 any)

Symbol Typable Typed Any
meth ConvReLU2d.init 12 0 0
meth ConvReLU2d.forward 2 0 0
meth ConvReLU2d._get_name 1 0 0
meth ConvReLU2d.from_float 3 0 0
meth ConvReLU2d.from_reference 4 0 0
meth ConvReLU3d.init 12 0 0
meth ConvReLU3d.forward 2 0 0
meth ConvReLU3d._get_name 1 0 0
meth ConvReLU3d.from_float 3 0 0
meth ConvReLU3d.from_reference 4 0 0
meth ConvReLU1d.init 12 0 0
meth ConvReLU1d.forward 2 0 0
meth ConvReLU1d._get_name 1 0 0
meth ConvReLU1d.from_float 3 0 0
meth ConvReLU1d.from_reference 4 0 0

torch.ao.nn.intrinsic.quantized.modules.linear_relu (41 missing, 0 any)

Symbol Typable Typed Any
meth LinearTanh.init 5 0 0
meth LinearTanh._get_name 1 0 0
meth LinearTanh.from_float 3 0 0
meth LinearTanh.from_reference 4 0 0
meth LinearReLU.init 5 0 0
meth LinearReLU._get_name 1 0 0
meth LinearReLU.from_float 3 0 0
meth LinearReLU.from_reference 4 0 0
meth LinearLeakyReLU.init 6 0 0
meth LinearLeakyReLU._get_name 1 0 0
meth LinearLeakyReLU.from_float 3 0 0
meth LinearLeakyReLU.from_reference 4 0 0
attr LinearLeakyReLU.negative_slope 1 0 0

torch.ao.nn.qat.modules.conv (22 missing, 0 any)

Symbol Typable Typed Any
meth Conv2d.init 13 10 0
meth Conv2d.forward 2 0 0
meth Conv2d.from_float 3 0 0
meth Conv3d.init 13 10 0
meth Conv3d.forward 2 0 0
meth Conv3d.from_float 3 0 0
meth Conv1d.init 13 10 0
meth Conv1d.from_float 3 0 0

torch.ao.nn.qat.modules.embedding_ops (40 missing, 0 any)

Symbol Typable Typed Any
meth Embedding.init 12 1 0
meth Embedding.forward 2 1 0
meth Embedding.from_float 3 0 0
meth Embedding.to_float 1 0 0
attr Embedding.qconfig 1 0 0
attr Embedding.weight_fake_quant 1 0 0
meth EmbeddingBag.init 14 1 0
meth EmbeddingBag.forward 4 1 0
meth EmbeddingBag.from_float 3 0 0
meth EmbeddingBag.to_float 1 0 0
attr EmbeddingBag.qconfig 1 0 0
attr EmbeddingBag.weight_fake_quant 1 0 0

torch.ao.nn.qat.modules.linear (14 missing, 0 any)

Symbol Typable Typed Any
meth Linear.init 7 1 0
meth Linear.forward 2 0 0
meth Linear.from_float 3 0 0
meth Linear.to_float 1 0 0
attr Linear.qconfig 1 0 0
attr Linear.weight_fake_quant 1 0 0

torch.ao.nn.quantizable.modules.activation (18 missing, 0 any)

Symbol Typable Typed Any
meth MultiheadAttention.init 12 10 0
meth MultiheadAttention._get_name 1 0 0
meth MultiheadAttention.from_float 2 0 0
meth MultiheadAttention.dequantize 1 0 0
meth MultiheadAttention.from_observed 2 0 0
attr MultiheadAttention.linear_Q 1 0 0
attr MultiheadAttention.linear_K 1 0 0
attr MultiheadAttention.linear_V 1 0 0
attr MultiheadAttention.out_proj 1 0 0
attr MultiheadAttention.q_scaling_product 1 0 0
attr MultiheadAttention.quant_attn_output 1 0 0
attr MultiheadAttention.quant_attn_output_weights 1 0 0
attr MultiheadAttention.dequant_q 1 0 0
attr MultiheadAttention.dequant_k 1 0 0
attr MultiheadAttention.dequant_v 1 0 0

torch.ao.nn.quantizable.modules.rnn (45 missing, 0 any)

Symbol Typable Typed Any
meth LSTM.init 11 9 0
meth LSTM.forward 3 2 0
meth LSTM._get_name 1 0 0
meth LSTM.from_float 4 0 0
meth LSTM.from_observed 2 0 0
attr LSTM.input_size 1 0 0
attr LSTM.hidden_size 1 0 0
attr LSTM.num_layers 1 0 0
attr LSTM.bias 1 0 0
attr LSTM.batch_first 1 0 0
attr LSTM.dropout 1 0 0
attr LSTM.bidirectional 1 0 0
attr LSTM.training 1 0 0
attr LSTM.layers 1 0 0
meth LSTMCell.init 7 4 0
meth LSTMCell._get_name 1 0 0
meth LSTMCell.from_params 6 0 0
meth LSTMCell.from_float 4 0 0
attr LSTMCell.input_size 1 0 0
attr LSTMCell.hidden_size 1 0 0
attr LSTMCell.bias 1 0 0
attr LSTMCell.split_gates 1 0 0
attr LSTMCell.input_gate 1 0 0
attr LSTMCell.forget_gate 1 0 0
attr LSTMCell.cell_gate 1 0 0
attr LSTMCell.output_gate 1 0 0
attr LSTMCell.fgate_cx 1 0 0
attr LSTMCell.igate_cgate 1 0 0
attr LSTMCell.fgate_cx_igate_cgate 1 0 0
attr LSTMCell.ogate_cy 1 0 0

torch.ao.nn.quantized.dynamic.modules.conv (73 missing, 0 any)

Symbol Typable Typed Any
meth Conv2d.init 12 0 0
meth Conv2d._get_name 1 0 0
meth ConvTranspose2d.init 13 0 0
meth ConvTranspose2d._get_name 1 0 0
meth Conv3d.init 12 0 0
meth Conv3d._get_name 1 0 0
meth ConvTranspose3d.init 13 0 0
meth ConvTranspose3d._get_name 1 0 0
meth ConvTranspose1d.init 13 0 0
meth ConvTranspose1d._get_name 1 0 0
meth Conv1d.init 13 9 0
meth Conv1d._get_name 1 0 0

torch.ao.nn.quantized.dynamic.modules.linear (23 missing, 0 any)

Symbol Typable Typed Any
meth Linear.init 5 0 0
meth Linear.forward 2 0 0
meth Linear._get_name 1 0 0
meth Linear.extra_repr 1 0 0
meth Linear._load_from_state_dict 8 0 0
meth Linear.from_float 3 0 0
meth Linear.from_reference 2 0 0
attr Linear.version 1 0 0

torch.ao.nn.quantized.dynamic.modules.rnn (148 missing, 0 any)

Symbol Typable Typed Any
meth GRU.init 3 0 0
meth GRU._get_name 1 0 0
meth GRU.forward 3 0 0
meth GRU.from_float 3 0 0
meth GRU.from_reference 2 0 0
meth LSTM.init 3 0 0
meth LSTM._get_name 1 0 0
meth LSTM.forward 3 0 0
meth LSTM.from_float 3 0 0
meth LSTM.from_reference 2 0 0
meth LSTMCell.init 3 0 0
meth LSTMCell._get_name 1 0 0
meth LSTMCell.from_float 3 0 0
meth RNNCell.init 6 0 0
meth RNNCell._get_name 1 0 0
meth RNNCell.from_float 3 0 0
attr RNNCell.nonlinearity 1 0 0
meth GRUCell.init 5 0 0
meth GRUCell._get_name 1 0 0
meth GRUCell.from_float 3 0 0
func pack_weight_bias 4 0 0
meth RNNBase.init 10 0 0
meth RNNBase._get_name 1 0 0
meth RNNBase.extra_repr 1 0 0
meth RNNBase.repr 1 0 0
meth RNNBase._load_from_state_dict 8 0 0
meth RNNBase.set_weight_bias 2 0 0
meth RNNBase.from_float 3 0 0
meth RNNBase._weight_bias 1 0 0
meth RNNBase.get_weight 1 0 0
meth RNNBase.get_bias 1 0 0
attr RNNBase.mode 1 0 0
attr RNNBase.input_size 1 0 0
attr RNNBase.hidden_size 1 0 0
attr RNNBase.num_layers 1 0 0
attr RNNBase.bias 1 0 0
attr RNNBase.batch_first 1 0 0
attr RNNBase.dropout 1 0 0
attr RNNBase.bidirectional 1 0 0
attr RNNBase.dtype 1 0 0
attr RNNBase.version 1 0 0
attr RNNBase.training 1 0 0
meth RNNCellBase.init 6 0 0
meth RNNCellBase._get_name 1 0 0
meth RNNCellBase.extra_repr 1 0 0
meth RNNCellBase.check_forward_input 2 0 0
meth RNNCellBase.from_float 3 0 0
meth RNNCellBase.from_reference 2 0 0
meth RNNCellBase._weight_bias 1 0 0
meth RNNCellBase.get_weight 1 0 0
meth RNNCellBase.get_bias 1 0 0
meth RNNCellBase.set_weight_bias 2 0 0
meth RNNCellBase._save_to_state_dict 4 0 0
meth RNNCellBase._load_from_state_dict 8 0 0
attr RNNCellBase.input_size 1 0 0
attr RNNCellBase.hidden_size 1 0 0
attr RNNCellBase.bias 1 0 0
attr RNNCellBase.weight_dtype 1 0 0
attr RNNCellBase.bias_ih 1 0 0
attr RNNCellBase.bias_hh 1 0 0
meth PackedParameter.init 2 0 0
meth PackedParameter._save_to_state_dict 4 0 0
meth PackedParameter._load_from_state_dict 8 0 0
attr PackedParameter.param 1 0 0

torch.ao.nn.quantized.functional (89 missing, 0 any)

Symbol Typable Typed Any
func avg_pool3d 8 0 0
func leaky_relu 6 5 0
func max_pool2d 8 0 0
func conv2d 12 0 0
func max_pool1d 8 0 0
func conv1d 12 0 0
func upsample_bilinear 4 0 0
func conv3d 12 0 0
func avg_pool2d 8 0 0
func upsample_nearest 4 0 0
func upsample 6 0 0
func interpolate 6 0 0

torch.ao.nn.quantized.modules (17 missing, 0 any)

Symbol Typable Typed Any
meth DeQuantize.forward 2 0 0
meth DeQuantize.from_float 3 0 0
meth Quantize.init 5 0 0
meth Quantize.forward 2 0 0
meth Quantize.from_float 3 0 0
meth Quantize.extra_repr 1 0 0
attr Quantize.dtype 1 0 0

torch.ao.nn.quantized.modules.activation (93 missing, 0 any)

Symbol Typable Typed Any
meth Sigmoid.init 3 2 0
meth Sigmoid.forward 2 0 0
meth Sigmoid.from_float 3 0 0
attr Sigmoid.output_scale 1 0 0
attr Sigmoid.output_zero_point 1 0 0
meth PReLU._get_name 1 0 0
meth PReLU.from_float 3 0 0
meth PReLU.from_reference 4 0 0
attr PReLU.num_parameters 1 0 0
attr PReLU.scale 1 0 0
attr PReLU.zero_point 1 0 0
meth Hardswish.init 5 0 0
meth Hardswish.forward 2 0 0
meth Hardswish._get_name 1 0 0
meth Hardswish.from_float 3 0 0
meth Hardswish.from_reference 4 0 0
meth LeakyReLU.init 7 5 0
meth LeakyReLU.forward 2 0 0
meth LeakyReLU._get_name 1 0 0
meth LeakyReLU.from_float 3 0 0
meth LeakyReLU.from_reference 4 0 0
meth ELU.init 4 0 0
meth ELU.forward 2 0 0
meth ELU._get_name 1 0 0
meth ELU.from_float 3 0 0
meth ELU.from_reference 4 0 0
attr ELU.scale 1 0 0
attr ELU.zero_point 1 0 0
meth Softmax.init 4 0 0
meth Softmax.forward 2 0 0
meth Softmax._get_name 1 0 0
meth Softmax.from_float 3 0 0
meth Softmax.from_reference 4 0 0
attr Softmax.dim 1 0 0
attr Softmax.scale 1 0 0
attr Softmax.zero_point 1 0 0
meth MultiheadAttention._get_name 1 0 0
meth MultiheadAttention.from_float 2 0 0
meth MultiheadAttention.from_observed 2 0 0
meth ReLU6.init 2 0 0
meth ReLU6.forward 2 0 0
meth ReLU6._get_name 1 0 0
meth ReLU6.from_float 3 0 0
attr ReLU6.inplace 1 0 0

torch.ao.nn.quantized.modules.batchnorm (23 missing, 0 any)

Symbol Typable Typed Any
meth BatchNorm2d.init 6 1 0
meth BatchNorm2d._get_name 1 0 0
meth BatchNorm2d._check_input_dim 2 0 0
meth BatchNorm2d.from_float 3 0 0
meth BatchNorm3d.init 6 0 0
meth BatchNorm3d._get_name 1 0 0
meth BatchNorm3d._check_input_dim 2 0 0
meth BatchNorm3d.from_float 3 0 0

torch.ao.nn.quantized.modules.conv (123 missing, 0 any)

Symbol Typable Typed Any
meth Conv2d.init 12 0 0
meth Conv2d._get_name 1 0 0
meth Conv2d._weight_bias 1 0 0
meth Conv2d.weight 1 0 0
meth Conv2d.bias 1 0 0
meth Conv2d.forward 2 0 0
meth Conv2d.from_float 3 0 0
meth Conv3d.init 12 0 0
meth Conv3d._get_name 1 0 0
meth Conv3d._weight_bias 1 0 0
meth Conv3d.weight 1 0 0
meth Conv3d.bias 1 0 0
meth Conv3d.forward 2 0 0
meth Conv3d.from_float 3 0 0
meth ConvTranspose3d.init 13 0 0
meth ConvTranspose3d._get_name 1 0 0
meth ConvTranspose3d._weight_bias 1 0 0
meth ConvTranspose3d.weight 1 0 0
meth ConvTranspose3d.bias 1 0 0
meth ConvTranspose3d.forward 2 0 0
meth ConvTranspose3d.from_reference 4 0 0
meth ConvTranspose1d.init 13 0 0
meth ConvTranspose1d._get_name 1 0 0
meth ConvTranspose1d._weight_bias 1 0 0
meth ConvTranspose1d.weight 1 0 0
meth ConvTranspose1d.bias 1 0 0
meth ConvTranspose1d.forward 2 0 0
meth ConvTranspose1d.from_reference 4 0 0
meth Conv1d.init 12 9 0
meth Conv1d._get_name 1 0 0
meth Conv1d._weight_bias 1 0 0
meth Conv1d.weight 1 0 0
meth Conv1d.bias 1 0 0
meth Conv1d.forward 2 0 0
meth Conv1d.from_float 3 0 0
meth ConvTranspose2d.init 13 0 0
meth ConvTranspose2d._get_name 1 0 0
meth ConvTranspose2d._weight_bias 1 0 0
meth ConvTranspose2d.weight 1 0 0
meth ConvTranspose2d.bias 1 0 0
meth ConvTranspose2d.forward 2 0 0
meth ConvTranspose2d.from_reference 4 0 0

torch.ao.nn.quantized.modules.dropout (10 missing, 0 any)

Symbol Typable Typed Any
meth Dropout.forward 2 0 0
meth Dropout._get_name 1 0 0
meth Dropout.from_float 3 0 0
meth Dropout.from_reference 4 0 0

torch.ao.nn.quantized.modules.embedding_ops (45 missing, 0 any)

Symbol Typable Typed Any
meth Embedding.init 10 9 0
meth Embedding._get_name 1 0 0
meth Embedding.repr 1 0 0
meth Embedding.extra_repr 1 0 0
meth Embedding.weight 1 0 0
meth Embedding.from_float 3 0 0
meth Embedding.from_reference 2 0 0
attr Embedding.num_embeddings 1 0 0
attr Embedding.embedding_dim 1 0 0
attr Embedding.dtype 1 0 0
meth EmbeddingBag.init 11 10 0
meth EmbeddingBag._get_name 1 0 0
meth EmbeddingBag.from_float 3 0 0
meth EmbeddingBag.from_reference 2 0 0
attr EmbeddingBag.mode 1 0 0
attr EmbeddingBag.pruned_weights 1 0 0
attr EmbeddingBag.include_last_offset 1 0 0
attr EmbeddingBag.dtype 1 0 0
meth EmbeddingPackedParams.init 4 0 0
meth EmbeddingPackedParams._weight 1 0 0
meth EmbeddingPackedParams.forward 2 0 0
meth EmbeddingPackedParams._save_to_state_dict 4 0 0
meth EmbeddingPackedParams._load_from_state_dict 8 0 0
meth EmbeddingPackedParams.repr 1 0 0
attr EmbeddingPackedParams.dtype 1 0 0

torch.ao.nn.quantized.modules.functional_modules (27 missing, 0 any)

Symbol Typable Typed Any
meth FloatFunctional.forward 2 0 0
attr FloatFunctional.activation_post_process 1 0 0
meth QFunctional._save_to_state_dict 4 0 0
meth QFunctional._load_from_state_dict 8 0 0
meth QFunctional._get_name 1 0 0
meth QFunctional.extra_repr 1 0 0
meth QFunctional.forward 2 0 0
meth QFunctional.from_float 3 0 0
attr QFunctional.scale 1 0 0
attr QFunctional.zero_point 1 0 0
attr QFunctional.activation_post_process 1 0 0
meth FXFloatFunctional.forward 2 0 0

torch.ao.nn.quantized.modules.linear (53 missing, 0 any)

Symbol Typable Typed Any
meth Linear.init 5 0 0
meth Linear._get_name 1 0 0
meth Linear.extra_repr 1 0 0
meth Linear.repr 1 0 0
meth Linear._save_to_state_dict 4 0 0
meth Linear._load_from_state_dict 8 0 0
meth Linear._weight_bias 1 0 0
meth Linear.weight 1 0 0
meth Linear.bias 1 0 0
meth Linear.from_float 3 0 0
meth Linear.from_reference 4 0 0
attr Linear.in_features 1 0 0
attr Linear.out_features 1 0 0
attr Linear.scale 1 0 0
attr Linear.zero_point 1 0 0
meth LinearPackedParams.init 2 0 0
meth LinearPackedParams._weight_bias 1 0 0
meth LinearPackedParams.forward 2 0 0
meth LinearPackedParams._save_to_state_dict 4 0 0
meth LinearPackedParams._load_from_state_dict 8 0 0
meth LinearPackedParams.repr 1 0 0
attr LinearPackedParams.dtype 1 0 0

torch.ao.nn.quantized.modules.normalization (108 missing, 0 any)

Symbol Typable Typed Any
meth GroupNorm.init 11 1 0
meth GroupNorm.forward 2 0 0
meth GroupNorm._get_name 1 0 0
meth GroupNorm.from_float 3 0 0
attr GroupNorm.weight 1 0 0
attr GroupNorm.bias 1 0 0
meth LayerNorm.init 10 1 0
meth LayerNorm.forward 2 0 0
meth LayerNorm._get_name 1 0 0
meth LayerNorm.from_float 3 0 0
meth LayerNorm.from_reference 4 0 0
attr LayerNorm.weight 1 0 0
attr LayerNorm.bias 1 0 0
meth InstanceNorm1d.init 12 1 0
meth InstanceNorm1d.forward 2 0 0
meth InstanceNorm1d._get_name 1 0 0
meth InstanceNorm1d.from_float 3 0 0
meth InstanceNorm1d.from_reference 4 0 0
attr InstanceNorm1d.weight 1 0 0
attr InstanceNorm1d.bias 1 0 0
meth InstanceNorm2d.init 12 1 0
meth InstanceNorm2d.forward 2 0 0
meth InstanceNorm2d._get_name 1 0 0
meth InstanceNorm2d.from_float 3 0 0
meth InstanceNorm2d.from_reference 4 0 0
attr InstanceNorm2d.weight 1 0 0
attr InstanceNorm2d.bias 1 0 0
meth InstanceNorm3d.init 12 1 0
meth InstanceNorm3d.forward 2 0 0
meth InstanceNorm3d._get_name 1 0 0
meth InstanceNorm3d.from_float 3 0 0
meth InstanceNorm3d.from_reference 4 0 0
attr InstanceNorm3d.weight 1 0 0
attr InstanceNorm3d.bias 1 0 0

torch.ao.nn.quantized.modules.rnn (0 missing, 2 any)

Symbol Typable Typed Any
meth LSTM.from_float 3 3 2

torch.ao.nn.quantized.modules.utils (4 missing, 0 any)

Symbol Typable Typed Any
meth WeightedQuantizedModule.from_reference 4 0 0

torch.ao.nn.quantized.reference.modules.conv (80 missing, 0 any)

Symbol Typable Typed Any
meth Conv2d.init 13 1 0
meth Conv2d._get_name 1 0 0
meth Conv2d.from_float 3 0 0
meth ConvTranspose2d.init 14 1 0
meth ConvTranspose2d._get_name 1 0 0
meth ConvTranspose2d.from_float 3 0 0
meth Conv3d.init 13 1 0
meth Conv3d._get_name 1 0 0
meth Conv3d.from_float 3 0 0
meth ConvTranspose3d.init 14 1 0
meth ConvTranspose3d._get_name 1 0 0
meth ConvTranspose3d.from_float 3 0 0
meth ConvTranspose1d.init 14 11 0
meth ConvTranspose1d._get_name 1 0 0
meth ConvTranspose1d.from_float 3 0 0
meth Conv1d.init 13 10 0
meth Conv1d._get_name 1 0 0
meth Conv1d.from_float 3 0 0

torch.ao.nn.quantized.reference.modules.rnn (64 missing, 0 any)

Symbol Typable Typed Any
meth GRU.init 3 0 0
meth GRU.get_quantized_weight_bias_dict 1 0 0
meth GRU.get_flat_weights 1 0 0
meth GRU.forward 3 0 0
meth GRU._get_name 1 0 0
meth GRU.from_float 3 0 0
meth LSTMCell.init 7 5 0
meth LSTMCell._get_name 1 0 0
meth LSTMCell.from_float 4 0 0
meth LSTM.init 3 0 0
meth LSTM.check_forward_args 4 3 0
meth LSTM.get_quantized_weight_bias_dict 1 0 0
meth LSTM.get_flat_weights 1 0 0
meth LSTM.forward 3 0 0
meth LSTM._get_name 1 0 0
meth LSTM.from_float 3 0 0
meth RNNCell.init 8 6 0
meth RNNCell._get_name 1 0 0
meth RNNCell.from_float 3 0 0
attr RNNCell.nonlinearity 1 0 0
meth GRUCell.init 7 5 0
meth GRUCell._get_name 1 0 0
meth GRUCell.from_float 3 0 0
func get_quantized_weight 3 0 0
meth RNNBase.init 13 11 0
meth RNNBase._init_weight_qparams_dict 3 0 0
meth RNNCellBase.init 8 5 0
meth RNNCellBase._init_weight_qparams_dict 3 0 0
meth RNNCellBase._get_name 1 0 0
meth RNNCellBase.get_quantized_weight_ih 1 0 0
meth RNNCellBase.get_quantized_weight_hh 1 0 0
meth RNNCellBase.get_weight_ih 1 0 0
meth RNNCellBase.get_weight_hh 1 0 0

torch.ao.nn.quantized.reference.modules.sparse (13 missing, 0 any)

Symbol Typable Typed Any
meth EmbeddingBag.init 14 12 0
meth EmbeddingBag._get_name 1 0 0
meth EmbeddingBag.from_float 4 0 0
meth Embedding.init 12 10 0
meth Embedding._get_name 1 0 0
meth Embedding.from_float 3 0 0

torch.ao.nn.quantized.reference.modules.utils (17 missing, 0 any)

Symbol Typable Typed Any
meth ReferenceQuantizedModule._init_weight_qparams 3 0 0
meth ReferenceQuantizedModule.get_weight 1 0 0
meth ReferenceQuantizedModule.get_quantized_weight 1 0 0
meth ReferenceQuantizedModule._save_to_state_dict 4 0 0
meth ReferenceQuantizedModule._load_from_state_dict 8 0 0

torch.ao.nn.sparse.quantized.dynamic.linear (30 missing, 0 any)

Symbol Typable Typed Any
meth Linear.init 7 0 0
meth Linear._get_name 1 0 0
meth Linear.extra_repr 1 0 0
meth Linear.repr 1 0 0
meth Linear._save_to_state_dict 4 0 0
meth Linear._load_from_state_dict 8 0 0
meth Linear._weight_bias 1 0 0
meth Linear.weight 1 0 0
meth Linear.bias 1 0 0
meth Linear.from_float 3 0 0
attr Linear.in_features 1 0 0
attr Linear.out_features 1 0 0

torch.ao.nn.sparse.quantized.linear (57 missing, 0 any)

Symbol Typable Typed Any
meth Linear.init 7 0 0
meth Linear._get_name 1 0 0
meth Linear.extra_repr 1 0 0
meth Linear.repr 1 0 0
meth Linear._save_to_state_dict 4 0 0
meth Linear._load_from_state_dict 8 0 0
meth Linear._weight_bias 1 0 0
meth Linear.weight 1 0 0
meth Linear.bias 1 0 0
meth Linear.from_float 3 0 0
attr Linear.in_features 1 0 0
attr Linear.out_features 1 0 0
attr Linear.scale 1 0 0
attr Linear.zero_point 1 0 0
meth LinearPackedParams.init 4 0 0
meth LinearPackedParams._get_name 1 0 0
meth LinearPackedParams._weight_bias 1 0 0
meth LinearPackedParams.forward 2 0 0
meth LinearPackedParams._save_to_state_dict 4 0 0
meth LinearPackedParams._load_from_state_dict 8 0 0
meth LinearPackedParams.getstate 1 0 0
meth LinearPackedParams.setstate 2 0 0
meth LinearPackedParams.repr 1 0 0
attr LinearPackedParams.dtype 1 0 0

torch.ao.nn.sparse.quantized.utils (2 missing, 0 any)

Symbol Typable Typed Any
meth LinearBlockSparsePattern.init 3 2 0
attr LinearBlockSparsePattern.rlock 1 0 0

torch.ao.ns.fx.n_shadows_utils (9 missing, 3 any)

Symbol Typable Typed Any
func group_results_by_subgraph 2 2 1
meth OutputProp.init 2 0 0
meth OutputProp.propagate 2 0 0
attr OutputProp.mod 1 0 0
attr OutputProp.graph 1 0 0
attr OutputProp.modules 1 0 0
func create_results_comparison 2 1 1
func create_one_transformed_and_logged_copy_of_subgraph 12 12 1
func print_n_shadows_summary 2 1 0

torch.ao.ns.fx.qconfig_multi_mapping (1 missing, 0 any)

Symbol Typable Typed Any
meth QConfigMultiMapping.repr 1 0 0

torch.ao.ns.fx.utils (2 missing, 0 any)

Symbol Typable Typed Any
func maybe_dequantize_first_two_tensor_args_and_handle_tuples 2 0 0

torch.ao.pruning.scheduler.base_scheduler (23 missing, 0 any)

Symbol Typable Typed Any
meth BaseScheduler.init 4 0 0
meth BaseScheduler.state_dict 1 0 0
meth BaseScheduler.load_state_dict 2 0 0
meth BaseScheduler.get_last_sl 1 0 0
meth BaseScheduler.get_sl 1 0 0
meth BaseScheduler.print_sl 5 0 0
meth BaseScheduler.repr 1 0 0
meth BaseScheduler.step 2 0 0
meth BaseScheduler._make_sure_a_list 2 0 0
attr BaseScheduler.sparsifier 1 0 0
attr BaseScheduler.base_sl 1 0 0
attr BaseScheduler.last_epoch 1 0 0
attr BaseScheduler.verbose 1 0 0

torch.ao.pruning.scheduler.cubic_scheduler (24 missing, 0 any)

Symbol Typable Typed Any
meth CubicSL.init 9 0 0
meth CubicSL.sparsity_compute_fn 8 0 0
meth CubicSL.get_sl 1 0 0
attr CubicSL.sparsifier 1 0 0
attr CubicSL.init_sl 1 0 0
attr CubicSL.init_t 1 0 0
attr CubicSL.delta_t 1 0 0
attr CubicSL.total_t 1 0 0
attr CubicSL.initially_zero 1 0 0

torch.ao.pruning.scheduler.lambda_scheduler (2 missing, 0 any)

Symbol Typable Typed Any
attr LambdaSL.sparsifier 1 0 0
attr LambdaSL.sl_lambdas 1 0 0

torch.ao.pruning.sparsifier.base_sparsifier (16 missing, 0 any)

Symbol Typable Typed Any
meth BaseSparsifier.init 2 1 0
meth BaseSparsifier.repr 1 0 0
meth BaseSparsifier.load_state_dict 3 2 0
meth BaseSparsifier.prepare 3 0 0
meth BaseSparsifier._prepare 3 0 0
meth BaseSparsifier.squash_mask 5 2 0
meth BaseSparsifier.convert 5 4 0
meth BaseSparsifier.update_mask 4 2 0
attr BaseSparsifier.enable_mask_update 1 0 0

torch.ao.pruning.sparsifier.nearly_diagonal_sparsifier (6 missing, 0 any)

Symbol Typable Typed Any
meth NearlyDiagonalSparsifier.init 2 1 0
meth NearlyDiagonalSparsifier.update_mask 5 0 0

torch.ao.pruning.sparsifier.utils (7 missing, 0 any)

Symbol Typable Typed Any
meth FakeSparsity.init 2 0 0
meth FakeSparsity.forward 2 0 0
meth FakeSparsity.state_dict 3 0 0

torch.ao.pruning.sparsifier.weight_norm_sparsifier (28 missing, 0 any)

Symbol Typable Typed Any
meth WeightNormSparsifier.init 5 4 0
meth WeightNormSparsifier._scatter_fold_block_mask 8 0 0
meth WeightNormSparsifier._make_tensor_mask 6 0 0
meth WeightNormSparsifier._make_block_mask 5 0 0
meth WeightNormSparsifier.update_mask 7 0 0
attr WeightNormSparsifier.norm_fn 1 0 0

torch.ao.quantization (6 missing, 0 any)

Symbol Typable Typed Any
attr Pattern 1 0 0
attr get_combined_dict 1 0 0
attr MatchAllNode 1 0 0
func default_eval_fn 3 0 0

torch.ao.quantization.backend_config.backend_config (7 missing, 0 any)

Symbol Typable Typed Any
meth DTypeConfig.init 6 5 0
meth BackendPatternConfig.init 2 1 0
meth BackendPatternConfig.repr 1 0 0
attr BackendPatternConfig.observation_type 1 0 0
meth BackendConfig.init 2 1 0
meth BackendConfig.repr 1 0 0
attr BackendConfig.name 1 0 0

torch.ao.quantization.backend_config.native (9 missing, 0 any)

Symbol Typable Typed Any
func get_native_backend_config_dict 1 0 0
attr default_op_fp16_dtype_config 1 0 0
func get_test_only_legacy_native_backend_config_dict 1 0 0
attr weight_only_quint4x2_dtype_config 1 0 0
attr default_op_quint8_dtype_config 1 0 0
attr weight_only_quint8_dtype_config 1 0 0
attr input_output_only_quint8_dtype_config 1 0 0
attr default_dynamic_int8_dtype_config 1 0 0
attr default_dynamic_float16_dtype_config 1 0 0

torch.ao.quantization.backend_config.tensorrt (1 missing, 0 any)

Symbol Typable Typed Any
func get_tensorrt_backend_config_dict 1 0 0

torch.ao.quantization.backend_config.utils (3 missing, 2 any)

Symbol Typable Typed Any
func remove_boolean_dispatch_from_name 2 1 1
func pattern_to_human_readable 2 1 1
func entry_to_pretty_str 2 1 0

torch.ao.quantization.fake_quantize (65 missing, 2 any)

Symbol Typable Typed Any
meth FakeQuantizeBase.forward 2 0 0
meth FakeQuantizeBase.calculate_qparams 2 0 0
meth FakeQuantizeBase.disable_fake_quant 1 0 0
meth FakeQuantizeBase.disable_observer 1 0 0
meth FakeQuantizeBase.with_args 2 0 0
meth FakeQuantize.init 6 0 0
meth FakeQuantize.calculate_qparams 1 0 0
meth FakeQuantize.forward 2 0 0
meth FakeQuantize.extra_repr 1 0 0
meth FakeQuantize._save_to_state_dict 4 0 0
meth FakeQuantize._load_from_state_dict 8 0 0
attr FakeQuantize.activation_post_process 1 0 0
attr FakeQuantize.quant_min 1 0 0
attr FakeQuantize.quant_max 1 0 0
attr FakeQuantize.is_dynamic 1 0 0
attr FakeQuantize.dtype 1 0 0
attr FakeQuantize.qscheme 1 0 0
attr FakeQuantize.ch_axis 1 0 0
attr FakeQuantize.is_per_channel 1 0 0
attr default_weight_fake_quant 1 0 0
func disable_observer 2 0 0
meth FixedQParamsFakeQuantize.init 2 0 0
meth FixedQParamsFakeQuantize.calculate_qparams 1 0 0
meth FixedQParamsFakeQuantize.extra_repr 1 0 0
attr default_embedding_fake_quant 1 0 0
attr fused_per_channel_wt_fake_quant_range_neg_127_to_127 1 0 0
attr default_fixed_qparams_range_0to1_fake_quant 1 0 0
attr default_fixed_qparams_range_neg1to1_fake_quant 1 0 0
attr default_histogram_fake_quant 1 0 0
attr default_embedding_fake_quant_4bit 1 0 0
attr default_fake_quant 1 0 0
attr fused_wt_fake_quant_range_neg_127_to_127 1 0 0
attr default_fused_per_channel_wt_fake_quant 1 0 0
attr default_dynamic_fake_quant 1 0 0
func disable_fake_quant 2 0 0
attr default_per_channel_weight_fake_quant 1 0 0
func enable_observer 2 0 0
func enable_fake_quant 2 0 0
attr default_fused_wt_fake_quant 1 0 0
meth FusedMovingAvgObsFakeQuantize.init 5 5 2
attr FusedMovingAvgObsFakeQuantize.is_symmetric_quant 1 0 0
attr default_fused_act_fake_quant 1 0 0

torch.ao.quantization.fuse_modules (16 missing, 0 any)

Symbol Typable Typed Any
func fuse_modules_qat 6 0 0
func fuse_modules 6 0 0
func fuse_known_modules 4 0 0

torch.ao.quantization.fuser_method_mappings (21 missing, 0 any)

Symbol Typable Typed Any
func fuse_linear_bn 4 0 0
func fuse_conv_bn_relu 5 0 0
func get_fuser_method 3 0 0
func fuse_convtranspose_bn 4 0 0
func get_fuser_method_new 3 2 0
func fuse_conv_bn 4 0 0

torch.ao.quantization.fx.custom_config (3 missing, 0 any)

Symbol Typable Typed Any
meth ConvertCustomConfig.repr 1 0 0
meth PrepareCustomConfig.repr 1 0 0
meth FuseCustomConfig.repr 1 0 0

torch.ao.quantization.fx.fuse_handler (2 missing, 0 any)

Symbol Typable Typed Any
meth FuseHandler.init 2 1 0
meth DefaultFuseHandler.init 2 1 0

torch.ao.quantization.fx.graph_module (23 missing, 0 any)

Symbol Typable Typed Any
meth QuantizedGraphModule.init 4 3 0
meth QuantizedGraphModule._load_from_state_dict 8 0 0
meth QuantizedGraphModule.deepcopy 2 0 0
attr QuantizedGraphModule.preserved_attr_names 1 0 0
meth ObservedGraphModule.init 4 3 0
meth ObservedGraphModule.deepcopy 2 0 0
attr ObservedGraphModule.preserved_attr_names 1 0 0
meth ObservedStandaloneGraphModule.init 4 3 0
meth ObservedStandaloneGraphModule.deepcopy 2 0 0
meth FusedGraphModule.init 4 3 0
meth FusedGraphModule.deepcopy 2 0 0
attr FusedGraphModule.preserved_attr_names 1 0 0

torch.ao.quantization.fx.pattern_utils (1 missing, 0 any)

Symbol Typable Typed Any
func get_default_output_activation_post_process_map 2 1 0

torch.ao.quantization.fx.quantize_handler (11 missing, 0 any)

Symbol Typable Typed Any
meth QuantizeHandler.init 6 3 0
meth QuantizeHandler.is_custom_module 1 0 0
meth QuantizeHandler.is_standalone_module 1 0 0
attr QuantizeHandler.node_pattern 1 0 0
attr QuantizeHandler.modules 1 0 0
attr QuantizeHandler.root_node 1 0 0
attr QuantizeHandler.is_custom_module_ 1 0 0
attr QuantizeHandler.is_standalone_module_ 1 0 0
attr QuantizeHandler.num_tensor_args 1 0 0

torch.ao.quantization.fx.tracer (5 missing, 0 any)

Symbol Typable Typed Any
meth QuantizationTracer.init 3 2 0
attr QuantizationTracer.skipped_module_names 1 0 0
attr QuantizationTracer.skipped_module_classes 1 0 0
attr QuantizationTracer.scope 1 0 0
attr QuantizationTracer.record_stack_traces 1 0 0

torch.ao.quantization.fx.utils (2 missing, 5 any)

Symbol Typable Typed Any
func node_arg_is_weight 3 3 1
func create_getattr_from_value 6 6 1
func assert_and_get_unique_device 2 2 1
func node_arg_is_bias 3 3 1
func maybe_get_next_module 5 5 1
func get_linear_prepack_op_for_dtype 2 0 0

torch.ao.quantization.observer (200 missing, 0 any)

Symbol Typable Typed Any
func load_observer_state_dict 3 0 0
attr default_per_channel_weight_observer 1 0 0
meth FixedQParamsObserver.init 9 0 0
meth FixedQParamsObserver.forward 2 0 0
meth FixedQParamsObserver.calculate_qparams 1 0 0
attr FixedQParamsObserver.quant_min 1 0 0
attr FixedQParamsObserver.quant_max 1 0 0
attr FixedQParamsObserver.dtype 1 0 0
attr FixedQParamsObserver.qscheme 1 0 0
attr default_float_qparams_observer 1 0 0
meth UniformQuantizationObserverBase.init 10 1 0
meth UniformQuantizationObserverBase._load_from_state_dict 8 0 0
meth UniformQuantizationObserverBase.reset_min_max_vals 1 0 0
attr UniformQuantizationObserverBase.qscheme 1 0 0
attr UniformQuantizationObserverBase.reduce_range 1 0 0
attr UniformQuantizationObserverBase.has_customized_qrange 1 0 0
attr per_channel_weight_observer_range_neg_127_to_127 1 0 0
attr default_histogram_observer 1 0 0
meth ObserverBase.init 3 1 0
meth ObserverBase.forward 2 0 0
meth ObserverBase.calculate_qparams 2 0 0
attr ObserverBase.dtype 1 0 0
attr ObserverBase.is_dynamic 1 0 0
attr ObserverBase.with_args 1 0 0
attr ObserverBase.with_callable_args 1 0 0
meth ReuseInputObserver.forward 2 0 0
meth ReuseInputObserver.calculate_qparams 1 0 0
func get_observer_state_dict 2 0 0
meth MovingAverageMinMaxObserver.init 10 1 0
meth MovingAverageMinMaxObserver.forward 2 0 0
attr MovingAverageMinMaxObserver.averaging_constant 1 0 0
attr default_weight_observer 1 0 0
meth NoopObserver.init 3 1 0
meth NoopObserver.forward 2 0 0
meth NoopObserver.calculate_qparams 1 0 0
attr NoopObserver.dtype 1 0 0
attr NoopObserver.custom_op 1 0 0
attr default_float_qparams_observer_4bit 1 0 0
meth AffineQuantizedObserverBase.init 12 10 0
meth AffineQuantizedObserverBase.convert 3 2 0
attr AffineQuantizedObserverBase.with_args 1 0 0
attr AffineQuantizedObserverBase.mapping_type 1 0 0
attr AffineQuantizedObserverBase.target_dtype 1 0 0
attr AffineQuantizedObserverBase.granularity 1 0 0
attr AffineQuantizedObserverBase.quant_min 1 0 0
attr AffineQuantizedObserverBase.quant_max 1 0 0
attr AffineQuantizedObserverBase.eps 1 0 0
attr AffineQuantizedObserverBase.scale_dtype 1 0 0
attr AffineQuantizedObserverBase.zero_point_dtype 1 0 0
attr AffineQuantizedObserverBase.preserve_zero 1 0 0
attr AffineQuantizedObserverBase.zero_point_domain 1 0 0
attr AffineQuantizedObserverBase.block_size 1 0 0
attr AffineQuantizedObserverBase.original_dtype 1 0 0
attr default_dynamic_quant_observer 1 0 0
meth HistogramObserver.init 11 3 0
meth HistogramObserver._compute_quantization_error 3 2 0
meth HistogramObserver._upscale_histogram 6 5 0
meth HistogramObserver.calculate_qparams 1 0 0
meth HistogramObserver._save_to_state_dict 4 0 0
meth HistogramObserver._load_from_state_dict 8 0 0
meth HistogramObserver.extra_repr 1 0 0
attr HistogramObserver.bins 1 0 0
attr HistogramObserver.dst_nbins 1 0 0
attr HistogramObserver.upsample_rate 1 0 0
attr weight_observer_range_neg_127_to_127 1 0 0
meth PlaceholderObserver.init 9 1 0
meth PlaceholderObserver.forward 2 0 0
meth PlaceholderObserver.extra_repr 1 0 0
meth PlaceholderObserver.calculate_qparams 1 0 0
attr PlaceholderObserver.dtype 1 0 0
attr PlaceholderObserver.qscheme 1 0 0
attr PlaceholderObserver.quant_min 1 0 0
attr PlaceholderObserver.quant_max 1 0 0
attr PlaceholderObserver.eps 1 0 0
attr PlaceholderObserver.custom_op 1 0 0
attr default_fixed_qparams_range_0to1_observer 1 0 0
attr default_observer 1 0 0
meth RecordingObserver.init 2 0 0
meth RecordingObserver.forward 2 0 0
meth RecordingObserver.calculate_qparams 1 0 0
meth RecordingObserver.get_tensor_value 1 0 0
attr RecordingObserver.tensor_val 1 0 0
meth MinMaxObserver.init 10 1 0
meth MinMaxObserver.forward 2 0 0
meth MinMaxObserver.calculate_qparams 1 0 0
meth MinMaxObserver.extra_repr 1 0 0
meth MinMaxObserver.reset_min_max_vals 1 0 0
meth PerChannelMinMaxObserver.init 11 1 0
meth PerChannelMinMaxObserver.forward 2 0 0
meth PerChannelMinMaxObserver._forward 2 0 0
meth PerChannelMinMaxObserver.calculate_qparams 1 0 0
meth PerChannelMinMaxObserver.extra_repr 1 0 0
meth PerChannelMinMaxObserver._load_from_state_dict 8 7 0
meth PerChannelMinMaxObserver._load_from_state_dict_script 8 7 0
meth PerChannelMinMaxObserver.reset_min_max_vals 1 0 0
attr PerChannelMinMaxObserver.ch_axis 1 0 0
meth MovingAveragePerChannelMinMaxObserver.init 11 1 0
meth MovingAveragePerChannelMinMaxObserver.forward 2 0 0
attr MovingAveragePerChannelMinMaxObserver.averaging_constant 1 0 0
attr default_fixed_qparams_range_neg1to1_observer 1 0 0

torch.ao.quantization.pt2e.export_utils (3 missing, 0 any)

Symbol Typable Typed Any
func _allow_exported_model_train_eval 2 1 0
func _move_exported_model_to_eval 2 1 0
func _move_exported_model_to_train 2 1 0

torch.ao.quantization.pt2e.graph_utils (4 missing, 0 any)

Symbol Typable Typed Any
func update_equivalent_types_dict 2 0 0
func find_sequential_partitions 5 3 0

torch.ao.quantization.pt2e.prepare (1 missing, 0 any)

Symbol Typable Typed Any
func prepare 5 4 0

torch.ao.quantization.qconfig (41 missing, 0 any)

Symbol Typable Typed Any
func get_default_qconfig 3 0 0
meth QConfig.new 3 0 0
func get_default_qat_qconfig 3 0 0
func get_default_qat_qconfig_dict 3 0 0
func qconfig_equals 3 2 0
func get_default_qconfig_dict 3 0 0
meth QConfigDynamic.new 3 0 0
attr default_symmetric_qnnpack_qconfig 1 0 0
attr default_per_channel_symmetric_qnnpack_qconfig 1 0 0
attr default_qat_qconfig 1 0 0
attr default_quint8_weight_qconfig 1 0 0
attr default_symmetric_qnnpack_qat_qconfig 1 0 0
attr default_qat_qconfig_v2 1 0 0
attr default_per_channel_symmetric_qnnpack_qat_qconfig 1 0 0
attr float16_static_qconfig 1 0 0
attr default_dynamic_qconfig 1 0 0
attr float_qparams_weight_only_qconfig_4bit 1 0 0
attr per_channel_dynamic_qconfig 1 0 0
attr default_dynamic_qat_qconfig 1 0 0
attr default_qconfig 1 0 0
attr float_qparams_weight_only_qconfig 1 0 0
attr default_weight_only_qconfig 1 0 0
attr float16_dynamic_qconfig 1 0 0
attr default_debug_qconfig 1 0 0
attr default_per_channel_qconfig 1 0 0
attr default_activation_only_qconfig 1 0 0
attr default_embedding_qat_qconfig_4bit 1 0 0
attr default_embedding_qat_qconfig 1 0 0
attr default_reuse_input_qconfig 1 0 0

torch.ao.quantization.qconfig_mapping (4 missing, 0 any)

Symbol Typable Typed Any
func get_default_qconfig_mapping 3 1 0
func get_default_qat_qconfig_mapping 3 1 0

torch.ao.quantization.quantization_mappings (0 missing, 2 any)

Symbol Typable Typed Any
func get_dynamic_quant_module_class 3 3 1
func get_static_quant_module_class 4 4 1

torch.ao.quantization.quantize (47 missing, 0 any)

Symbol Typable Typed Any
func add_quant_dequant 2 0 0
func prepare 6 0 0
func get_default_custom_config_dict 1 0 0
func prepare_qat 4 0 0
func quantize_qat 5 0 0
func quantize 6 0 0
func propagate_qconfig_ 4 0 0
func convert 8 0 0
func quantize_dynamic 6 0 0
func swap_module 5 0 0

torch.ao.quantization.quantize_jit (37 missing, 0 any)

Symbol Typable Typed Any
func convert_jit 5 0 0
func quantize_jit 7 0 0
func script_qconfig_dict 2 0 0
func prepare_dynamic_jit 4 0 0
func quantize_dynamic_jit 5 0 0
func convert_dynamic_jit 5 0 0
func script_qconfig 2 0 0
func prepare_jit 4 0 0
func fuse_conv_bn_jit 3 0 0

torch.ao.quantization.quantizer.composable_quantizer (2 missing, 0 any)

Symbol Typable Typed Any
meth ComposableQuantizer.init 2 1 0
attr ComposableQuantizer.quantizers 1 0 0

torch.ao.quantization.quantizer.quantizer (1 missing, 0 any)

Symbol Typable Typed Any
meth QuantizationSpec.post_init 1 0 0

torch.ao.quantization.quantizer.x86_inductor_quantizer (16 missing, 0 any)

Symbol Typable Typed Any
meth X86InductorQuantizer.set_global 2 1 0
meth X86InductorQuantizer.get_global_quantization_config 1 0 0
meth X86InductorQuantizer.set_module_name_qconfig 3 2 0
meth X86InductorQuantizer._get_input_idx_for_binary_node 3 2 0
meth X86InductorQuantizer._annotate_qat_conv2d_fusion_pattern 4 3 0
meth X86InductorQuantizer._annotate_conv2d_fusion_pattern 4 3 0
meth X86InductorQuantizer._annotate_linear_fusion_pattern 4 3 0
meth X86InductorQuantizer._annotate_matmul 4 3 0
meth X86InductorQuantizer._annotate_propagation_quantizable_pattern_entry 4 3 0
meth X86InductorQuantizer._annotate_propagation_quantizable_pattern 4 2 0
meth X86InductorQuantizer._annotate_output_share_observer_as_input 3 2 0
meth X86InductorQuantizer._annotate_output_for_int8_in_int8_out_pattern_entry 2 1 0
attr X86InductorQuantizer.module_function_to_aten_operator_type 1 0 0
func get_default_x86_inductor_quantization_config 4 3 0
func get_x86_inductor_linear_dynamic_fp16_config 1 0 0

torch.ao.quantization.quantizer.xnnpack_quantizer (4 missing, 0 any)

Symbol Typable Typed Any
func get_symmetric_quantization_config 8 7 0
meth XNNPACKQuantizer.set_module_type 3 2 0
meth XNNPACKQuantizer.set_module_name 3 2 0
attr XNNPACKQuantizer.supported_config_and_operators 1 0 0

torch.ao.quantization.quantizer.xnnpack_quantizer_utils (4 missing, 0 any)

Symbol Typable Typed Any
func get_weight_qspec 2 1 0
func get_bias_qspec 2 1 0
func get_output_act_qspec 2 1 0
func get_input_act_qspec 2 1 0

torch.ao.quantization.quantizer.xpu_inductor_quantizer (2 missing, 0 any)

Symbol Typable Typed Any
meth XPUInductorQuantizer._annotate_qat_conv2d_fusion_pattern 4 3 0
func get_default_xpu_inductor_quantization_config 1 0 0

torch.ao.quantization.stubs (5 missing, 0 any)

Symbol Typable Typed Any
meth QuantWrapper.init 2 1 0
meth DeQuantStub.init 2 1 0
attr DeQuantStub.qconfig 1 0 0
meth QuantStub.init 2 1 0
attr QuantStub.qconfig 1 0 0

torch.ao.quantization.utils (41 missing, 2 any)

Symbol Typable Typed Any
func weight_is_statically_quantized 2 0 0
func get_qparam_dict 2 0 0
func get_qconfig_dtypes 2 0 0
func to_underlying_dtype 2 0 0
func activation_is_int32_quantized 2 0 0
func weight_dtype 2 0 0
func get_quant_type 2 0 0
func is_per_channel 2 0 0
func activation_dtype 2 0 0
func getattr_from_fqn 3 3 2
func get_combined_dict 3 0 0
func op_is_int8_dynamically_quantized 2 1 0
func check_node 3 0 0
func is_per_tensor 2 0 0
func has_no_children_ignoring_parametrizations 2 0 0
func weight_is_quantized 2 0 0
func activation_is_dynamically_quantized 2 0 0
func get_swapped_custom_module_class 4 0 0
func activation_is_statically_quantized 2 0 0
func activation_is_int8_quantized 2 0 0

torch.autograd.anomaly_mode (6 missing, 0 any)

Symbol Typable Typed Any
attr set_detect_anomaly.prev 1 0 0
attr set_detect_anomaly.prev_check_nan 1 0 0
meth detect_anomaly.init 2 1 0
attr detect_anomaly.prev 1 0 0
attr detect_anomaly.check_nan 1 0 0
attr detect_anomaly.prev_check_nan 1 0 0

torch.autograd.forward_ad (11 missing, 3 any)

Symbol Typable Typed Any
meth dual_level.enter 1 0 0
meth dual_level.exit 4 4 3
func make_dual 4 0 0
func unpack_dual 3 0 0
func exit_dual_level 2 0 0
func enter_dual_level 1 0 0

torch.autograd.function (42 missing, 11 any)

Symbol Typable Typed Any
meth Function.init 3 0 0
meth Function.call 3 0 0
meth Function.vmap 4 0 0
meth Function.apply 3 0 0
meth Function._compiled_autograd_key 2 0 0
meth NestedIOFunction._do_forward 2 0 0
meth NestedIOFunction._do_backward 3 0 0
meth NestedIOFunction.backward 2 2 2
meth NestedIOFunction.call 2 0 0
meth NestedIOFunction.forward 2 2 2
meth NestedIOFunction.save_for_backward 2 2 1
meth NestedIOFunction.mark_dirty 3 3 2
meth NestedIOFunction.mark_non_differentiable 3 3 2
meth NestedIOFunction.forward_extended 2 2 1
meth NestedIOFunction.backward_extended 2 2 1
prop NestedIOFunction.saved_tensors 1 0 0
meth FunctionCtx.save_for_backward 2 1 0
meth FunctionCtx.save_for_forward 2 1 0
meth FunctionCtx.mark_dirty 2 1 0
meth FunctionCtx.mark_shared_storage 2 0 0
meth FunctionCtx.mark_non_differentiable 2 1 0
meth FunctionCtx.set_materialize_grads 2 1 0
meth FunctionMeta.init 4 0 0
meth BackwardCFunction.apply 2 0 0
meth BackwardCFunction.apply_jvp 2 0 0
meth BackwardCFunction._compiled_autograd_key 1 0 0
meth InplaceFunction.init 2 0 0
attr InplaceFunction.inplace 1 0 0

torch.autograd.functional (38 missing, 0 any)

Symbol Typable Typed Any
func hvp 6 0 0
func jvp 6 0 0
func jacobian 7 0 0
func vjp 6 0 0
func hessian 7 0 0
func vhp 6 0 0

torch.autograd.grad_mode (8 missing, 15 any)

Symbol Typable Typed Any
meth enable_grad.exit 4 4 3
meth no_grad.exit 4 4 3
attr no_grad.prev 1 0 0
meth inference_mode.new 2 0 0
meth inference_mode.exit 4 4 3
attr inference_mode.mode 1 0 0
meth set_grad_enabled.exit 4 4 3
attr set_grad_enabled.prev 1 0 0
attr set_grad_enabled.mode 1 0 0
meth set_multithreading_enabled.exit 4 4 3
attr set_multithreading_enabled.prev 1 0 0
attr set_multithreading_enabled.mode 1 0 0

torch.autograd.gradcheck (18 missing, 0 any)

Symbol Typable Typed Any
func get_numerical_jacobian 6 0 0
func get_numerical_jacobian_wrt_specific_input 8 1 0
func get_analytical_jacobian 5 0 0

torch.autograd.graph (2 missing, 0 any)

Symbol Typable Typed Any
attr saved_tensors_hooks.pack_hook 1 0 0
attr saved_tensors_hooks.unpack_hook 1 0 0

torch.autograd.profiler (99 missing, 3 any)

Symbol Typable Typed Any
meth EnforceUnique.init 1 0 0
meth EnforceUnique.see 2 0 0
attr EnforceUnique.seen 1 0 0
meth record_function.init 3 2 0
meth record_function.enter 1 0 0
meth record_function.exit 4 3 3
attr record_function.record 1 0 0
meth KinetoStepTracker.init_step_count 2 1 0
meth emit_nvtx.init 3 0 0
meth emit_nvtx.enter 1 0 0
meth emit_nvtx.exit 4 0 0
attr emit_nvtx.enabled 1 0 0
attr emit_nvtx.entered 1 0 0
attr emit_nvtx.record_shapes 1 0 0
func load_nvprof 2 0 0
meth profile.init 14 0 0
meth profile.default_trace_id 1 0 0
meth profile.create_trace_id 1 0 0
meth profile.config 2 0 0
meth profile.enter 1 0 0
meth profile._prepare_trace 1 0 0
meth profile._start_trace 1 0 0
meth profile.exit 4 0 0
meth profile.repr 1 0 0
meth profile.str 1 0 0
meth profile._ensure_function_events 1 0 0
meth profile.table 8 0 0
meth profile.export_chrome_trace 2 0 0
meth profile.export_stacks 3 2 0
meth profile.toggle_collection_dynamic 3 2 0
meth profile.key_averages 4 0 0
meth profile.total_average 1 0 0
meth profile._parse_kineto_results 2 1 0
prop profile.function_events 1 0 0
prop profile.self_cpu_time_total 1 0 0
attr profile.use_cuda 1 0 0
attr profile.entered 1 0 0
attr profile.record_shapes 1 0 0
attr profile.with_flops 1 0 0
attr profile.profile_memory 1 0 0
attr profile.with_stack 1 0 0
attr profile.with_modules 1 0 0
attr profile.use_cpu 1 0 0
attr profile.acc_events 1 0 0
attr profile.experimental_config 1 0 0
attr profile.profiling_start_time_ns 1 0 0
attr profile.profiling_end_time_ns 1 0 0
attr profile.custom_trace_id_callback 1 0 0
attr profile.trace_id 1 0 0
attr profile.kineto_activities 1 0 0
attr profile.profiler_kind 1 0 0
meth emit_itt.init 3 0 0
meth emit_itt.enter 1 0 0
meth emit_itt.exit 4 0 0
attr emit_itt.enabled 1 0 0
attr emit_itt.entered 1 0 0
attr emit_itt.record_shapes 1 0 0
func parse_nvprof_trace 2 0 0

torch.autograd.profiler_legacy (43 missing, 0 any)

Symbol Typable Typed Any
meth profile.init 8 0 0
meth profile.config 1 0 0
meth profile.enter 1 0 0
meth profile._start_trace 1 0 0
meth profile.exit 4 0 0
meth profile.repr 1 0 0
meth profile.str 1 0 0
meth profile._check_finish 1 0 0
meth profile.table 8 0 0
meth profile.export_chrome_trace 2 0 0
meth profile.export_stacks 3 2 0
meth profile.key_averages 3 0 0
meth profile.total_average 1 0 0
prop profile.self_cpu_time_total 1 0 0
attr profile.use_cuda 1 0 0
attr profile.function_events 1 0 0
attr profile.entered 1 0 0
attr profile.record_shapes 1 0 0
attr profile.with_flops 1 0 0
attr profile.profile_memory 1 0 0
attr profile.with_stack 1 0 0
attr profile.with_modules 1 0 0
attr profile.profiler_kind 1 0 0

torch.autograd.profiler_util (105 missing, 0 any)

Symbol Typable Typed Any
meth EventList.init 3 0 0
meth EventList._build_tree 1 0 0
meth EventList.str 1 0 0
meth EventList._remove_dup_nodes 1 0 0
meth EventList._populate_cpu_children 1 0 0
meth EventList._set_backward_stacktraces 1 0 0
meth EventList.table 9 0 0
meth EventList.export_chrome_trace 2 0 0
meth EventList.supported_export_stacks_metrics 1 0 0
meth EventList.export_stacks 3 2 0
meth EventList.key_averages 4 0 0
meth EventList.total_average 1 0 0
prop EventList.self_cpu_time_total 1 0 0
meth MemRecordsAcc.init 2 0 0
meth MemRecordsAcc.in_interval 3 0 0
meth FunctionEvent.init 28 0 0
meth FunctionEvent.append_kernel 4 0 0
meth FunctionEvent.append_cpu_child 2 0 0
meth FunctionEvent.set_cpu_parent 2 0 0
meth FunctionEvent.repr 1 0 0
prop FunctionEvent.self_cpu_memory_usage 1 0 0
prop FunctionEvent.self_device_memory_usage 1 0 0
prop FunctionEvent.self_cuda_memory_usage 1 0 0
prop FunctionEvent.cpu_time_total 1 0 0
prop FunctionEvent.self_cpu_time_total 1 0 0
prop FunctionEvent.device_time_total 1 0 0
prop FunctionEvent.cuda_time_total 1 0 0
prop FunctionEvent.self_device_time_total 1 0 0
prop FunctionEvent.self_cuda_time_total 1 0 0
prop FunctionEvent.key 1 0 0
attr FunctionEvent.self_cpu_percent 1 0 0
attr FunctionEvent.total_cpu_percent 1 0 0
attr FunctionEvent.total_device_percent 1 0 0
attr FunctionEvent.metadata_json 1 0 0
meth FunctionEventAvg.add 2 0 0
meth FunctionEventAvg.iadd 2 0 0
meth FunctionEventAvg.repr 1 0 0
prop FormattedTimesMixin.cpu_time 1 0 0
prop FormattedTimesMixin.device_time 1 0 0
prop FormattedTimesMixin.cuda_time 1 0 0
attr FormattedTimesMixin.cpu_time_str 1 0 0
attr FormattedTimesMixin.device_time_str 1 0 0
attr FormattedTimesMixin.cpu_time_total_str 1 0 0
attr FormattedTimesMixin.device_time_total_str 1 0 0
attr FormattedTimesMixin.self_cpu_time_total_str 1 0 0
attr FormattedTimesMixin.self_device_time_total_str 1 0 0
meth StringTable.missing 2 0 0
meth Interval.init 3 0 0
meth Interval.elapsed_us 1 0 0
attr Interval.start 1 0 0
attr Interval.end 1 0 0

torch.autograd.variable (3 missing, 0 any)

Symbol Typable Typed Any
attr Variable._execution_engine 1 0 0
meth VariableMeta.instancecheck 2 0 0

torch.backends (24 missing, 0 any)

Symbol Typable Typed Any
func set_flags 2 0 0
meth ContextProp.init 3 0 0
meth ContextProp.get 3 0 0
meth ContextProp.set 3 0 0
attr ContextProp.getter 1 0 0
attr ContextProp.setter 1 0 0
func disable_global_flags 1 0 0
attr GenericModule.fp32_precision 1 0 0
func flags 2 0 0
meth PropModule.init 3 0 0
meth PropModule.getattr 2 0 0
attr PropModule.m 1 0 0
func flags_frozen 1 0 0

torch.backends.cuda (45 missing, 1 any)

Symbol Typable Typed Any
func enable_mem_efficient_sdp 2 1 0
func fp16_bf16_reduction_math_sdp_allowed 1 0 0
attr matmul 1 0 0
meth cuFFTPlanCacheAttrContextProp.init 3 0 0
meth cuFFTPlanCacheAttrContextProp.get 3 0 0
meth cuFFTPlanCacheAttrContextProp.set 3 0 0
attr cuFFTPlanCacheAttrContextProp.getter 1 0 0
attr cuFFTPlanCacheAttrContextProp.setter 1 0 0
func math_sdp_enabled 1 0 0
func allow_fp16_bf16_reduction_math_sdp 2 1 0
func is_built 1 0 0
meth cuFFTPlanCache.init 2 0 0
meth cuFFTPlanCache.clear 1 0 0
attr cuFFTPlanCache.device_index 1 0 0
attr cuFFTPlanCache.size 1 0 0
attr cuFFTPlanCache.max_size 1 0 0
func enable_math_sdp 2 1 0
func cudnn_sdp_enabled 1 0 0
attr cufft_plan_cache 1 0 0
meth cuFFTPlanCacheManager.init 1 0 0
meth cuFFTPlanCacheManager.getitem 2 0 0
meth cuFFTPlanCacheManager.getattr 2 0 0
meth cuFFTPlanCacheManager.setattr 3 0 0
attr cuFFTPlanCacheManager.caches 1 0 0
func enable_cudnn_sdp 2 1 0
func flash_sdp_enabled 1 0 0
func mem_efficient_sdp_enabled 1 0 0
func sdp_kernel 5 4 0
func enable_flash_sdp 2 1 0
meth cuBLASModule._parse_reduction_setting 3 3 1
meth cuBLASModule.getattr 2 0 0
meth cuBLASModule.setattr 3 0 0

torch.backends.cudnn (26 missing, 0 any)

Symbol Typable Typed Any
func set_flags 7 0 0
func is_acceptable 2 0 0
func version 1 0 0
attr CudnnModule.enabled 1 0 0
attr CudnnModule.deterministic 1 0 0
attr CudnnModule.benchmark 1 0 0
attr CudnnModule.benchmark_limit 1 0 0
attr CudnnModule.allow_tf32 1 0 0
attr CudnnModule.conv 1 0 0
attr CudnnModule.rnn 1 0 0
attr CudnnModule.fp32_precision 1 0 0
func is_available 1 0 0
func flags 7 0 0

torch.backends.cudnn.rnn (14 missing, 0 any)

Symbol Typable Typed Any
meth Unserializable.init 2 0 0
meth Unserializable.get 1 0 0
meth Unserializable.getstate 1 0 0
meth Unserializable.setstate 2 0 0
attr Unserializable.inner 1 0 0
func get_cudnn_mode 2 0 0
func init_dropout_state 5 0 0

torch.backends.kleidiai (1 missing, 0 any)

Symbol Typable Typed Any
func is_available 1 0 0

torch.backends.miopen (5 missing, 0 any)

Symbol Typable Typed Any
func set_flags 2 0 0
attr MiopenModule.immediate 1 0 0
func flags 2 0 0

torch.backends.mkl (9 missing, 0 any)

Symbol Typable Typed Any
func is_available 1 0 0
meth verbose.init 2 0 0
meth verbose.enter 1 0 0
meth verbose.exit 4 0 0
attr verbose.enable 1 0 0

torch.backends.mkldnn (28 missing, 0 any)

Symbol Typable Typed Any
func set_flags 5 0 0
meth verbose.init 2 0 0
meth verbose.enter 1 0 0
meth verbose.exit 4 0 0
attr verbose.level 1 0 0
func is_acl_available 1 0 0
meth MkldnnModule.is_available 1 0 0
attr MkldnnModule.enabled 1 0 0
attr MkldnnModule.deterministic 1 0 0
attr MkldnnModule.allow_tf32 1 0 0
attr MkldnnModule.matmul 1 0 0
attr MkldnnModule.conv 1 0 0
attr MkldnnModule.rnn 1 0 0
attr MkldnnModule.fp32_precision 1 0 0
func is_available 1 0 0
func flags 5 0 0

torch.backends.nnpack (5 missing, 0 any)

Symbol Typable Typed Any
func is_available 1 0 0
func set_flags 2 0 0
func flags 2 0 0

torch.backends.openmp (1 missing, 0 any)

Symbol Typable Typed Any
func is_available 1 0 0

torch.backends.opt_einsum (10 missing, 1 any)

Symbol Typable Typed Any
func set_flags 3 0 0
attr strategy 1 0 0
func get_opt_einsum 1 1 1
attr enabled 1 0 0
attr OptEinsumModule.enabled 1 0 0
attr OptEinsumModule.strategy 1 0 0
func flags 3 0 0

torch.backends.quantized (8 missing, 0 any)

Symbol Typable Typed Any
meth QuantizedEngine.init 3 0 0
meth QuantizedEngine.getattr 2 0 0
attr QuantizedEngine.m 1 0 0
attr QuantizedEngine.engine 1 0 0
attr QuantizedEngine.supported_engines 1 0 0

torch.backends.xeon.run_cpu (7 missing, 0 any)

Symbol Typable Typed Any
attr parser 1 0 0
attr args 1 0 0
func create_args 2 0 0
attr logger 1 0 0
func main 2 0 0

torch.backends.xnnpack (7 missing, 0 any)

Symbol Typable Typed Any
meth XNNPACKEngine.init 3 0 0
meth XNNPACKEngine.getattr 2 0 0
attr XNNPACKEngine.m 1 0 0
attr XNNPACKEngine.enabled 1 0 0

torch.compiler (30 missing, 0 any)

Symbol Typable Typed Any
func assume_constant_result 2 0 0
func wrap_numpy 2 0 0
func keep_tensor_guards_unsafe 3 0 0
func set_enable_guard_collectives 2 1 0
func skip_guard_on_inbuilt_nn_modules_unsafe 2 0 0
func disable 4 0 0
func skip_all_guards_unsafe 2 0 0
func allow_in_graph 2 0 0
func set_stance 4 3 0
func cudagraph_mark_step_begin 1 0 0
func skip_guard_on_all_nn_modules_unsafe 2 0 0
func compile 3 0 0
func list_backends 2 1 0
func skip_guard_on_globals_unsafe 2 0 0
func nested_compile_region 2 0 0

torch.cpu (9 missing, 3 any)

Symbol Typable Typed Any
meth StreamContext.init 2 0 0
meth StreamContext.enter 1 0 0
meth StreamContext.exit 4 4 3
attr StreamContext.stream 1 0 0
attr StreamContext.prev_stream 1 0 0
meth Event.record 2 1 0
meth Event.wait 2 1 0
meth Stream.wait_stream 2 1 0
meth Stream.wait_event 2 1 0

torch.cpu.amp.autocast_mode (7 missing, 3 any)

Symbol Typable Typed Any
meth autocast.init 4 3 0
meth autocast.enter 1 0 0
meth autocast.exit 4 3 3
meth autocast.call 2 0 0
attr autocast.device 1 0 0
attr autocast.fast_dtype 1 0 0

torch.cuda (55 missing, 7 any)

Symbol Typable Typed Any
func cudart 1 0 0
attr ByteTensor 1 0 0
attr BoolTensor 1 0 0
meth device.init 2 1 1
meth device.enter 1 0 0
meth device.exit 4 3 3
attr device.idx 1 0 0
attr device.prev_idx 1 0 0
attr CharTensor 1 0 0
meth IntStorage.dtype 1 0 0
meth IntStorage._dtype 1 0 0
meth LongStorage.dtype 1 0 0
meth LongStorage._dtype 1 0 0
meth ByteStorage.dtype 1 0 0
meth ByteStorage._dtype 1 0 0
attr LongTensor 1 0 0
func current_blas_handle 1 0 0
meth CharStorage.dtype 1 0 0
meth CharStorage._dtype 1 0 0
func is_initialized 1 0 0
meth BoolStorage.dtype 1 0 0
meth BoolStorage._dtype 1 0 0
meth StreamContext.init 2 1 0
meth StreamContext.enter 1 0 0
meth StreamContext.exit 4 3 3
attr StreamContext.stream 1 0 0
attr StreamContext.idx 1 0 0
attr StreamContext.src_prev_stream 1 0 0
attr StreamContext.dst_prev_stream 1 0 0
attr FloatTensor 1 0 0
meth BFloat16Storage.dtype 1 0 0
meth BFloat16Storage._dtype 1 0 0
meth ComplexDoubleStorage.dtype 1 0 0
meth ComplexDoubleStorage._dtype 1 0 0
meth FloatStorage.dtype 1 0 0
meth FloatStorage._dtype 1 0 0
meth ShortStorage.dtype 1 0 0
meth ShortStorage._dtype 1 0 0
func ipc_collect 1 0 0
attr BFloat16Tensor 1 0 0
meth ComplexFloatStorage.dtype 1 0 0
meth ComplexFloatStorage._dtype 1 0 0
func set_stream 2 1 0
meth DoubleStorage.dtype 1 0 0
meth DoubleStorage._dtype 1 0 0
attr DoubleTensor 1 0 0
meth device_of.init 2 0 0
attr ShortTensor 1 0 0
attr IntTensor 1 0 0
attr HalfTensor 1 0 0
meth HalfStorage.dtype 1 0 0
meth HalfStorage._dtype 1 0 0
func init 1 0 0
func is_bf16_supported 2 1 0

torch.cuda.amp.autocast_mode (12 missing, 3 any)

Symbol Typable Typed Any
func custom_fwd 3 0 0
func custom_bwd 2 0 0
meth autocast.init 4 3 0
meth autocast.enter 1 0 0
meth autocast.exit 4 3 3
meth autocast.call 2 0 0
attr autocast.device 1 0 0
attr autocast.fast_dtype 1 0 0

torch.cuda.amp.common (1 missing, 0 any)

Symbol Typable Typed Any
func amp_definitely_not_available 1 0 0

torch.cuda.gds (4 missing, 0 any)

Symbol Typable Typed Any
meth GdsFile.init 3 2 0
attr GdsFile.filename 1 0 0
attr GdsFile.flags 1 0 0
attr GdsFile.fd 1 0 0

torch.cuda.graphs (5 missing, 0 any)

Symbol Typable Typed Any
meth graph.init 5 4 0
attr graph.capture_stream 1 0 0
attr graph.stream_ctx 1 0 0
attr graph.cuda_graph 1 0 0
attr graph.capture_error_mode 1 0 0

torch.cuda.memory (12 missing, 0 any)

Symbol Typable Typed Any
func memory_snapshot 2 0 0
meth CUDAPluggableAllocator.init 4 3 0
func use_mem_pool 3 2 0
meth MemPool.init 4 3 0
meth MemPool.snapshot 1 0 0
func caching_allocator_alloc 4 1 0
func caching_allocator_delete 2 0 0
func set_per_process_memory_fraction 3 2 0

torch.cuda.nccl (13 missing, 0 any)

Symbol Typable Typed Any
func reduce_scatter 6 4 0
func all_gather 5 3 0
func reduce 8 7 0
func all_reduce 6 0 0
func broadcast 5 3 0

torch.cuda.nvtx (11 missing, 0 any)

Symbol Typable Typed Any
func range_pop 1 0 0
func range_start 2 1 0
func range 4 0 0
func range_push 2 0 0
func mark 2 0 0
func range_end 2 1 0

torch.cuda.profiler (7 missing, 0 any)

Symbol Typable Typed Any
func init 4 0 0
func start 1 0 0
func profile 1 0 0
func stop 1 0 0

torch.cuda.streams (33 missing, 0 any)

Symbol Typable Typed Any
meth Event.new 5 0 0
meth Event.from_ipc_handle 3 0 0
meth Event.record 2 0 0
meth Event.wait 2 1 0
meth Event.query 1 0 0
meth Event.elapsed_time 2 0 0
meth Event.ipc_handle 1 0 0
prop Event.as_parameter 1 0 0
meth ExternalStream.new 4 0 0
meth Stream.new 4 0 0
meth Stream.wait_event 2 1 0
meth Stream.wait_stream 2 1 0
meth Stream.record_event 2 0 0
meth Stream.eq 2 1 0
meth Stream.hash 1 0 0
meth Stream.repr 1 0 0
meth Stream.cuda_stream 1 0 0
prop Stream.as_parameter 1 0 0

torch.distributed (3 missing, 0 any)

Symbol Typable Typed Any
attr log 1 0 0
func breakpoint 4 2 0

torch.distributed.algorithms.ddp_comm_hooks (3 missing, 0 any)

Symbol Typable Typed Any
func register_ddp_comm_hook 4 1 0

torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks (0 missing, 1 any)

Symbol Typable Typed Any
func noop_hook 3 3 1

torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook (13 missing, 0 any)

Symbol Typable Typed Any
meth PostLocalSGDState.init 5 0 0
meth PostLocalSGDState.maybe_increase_iter 2 0 0
attr PostLocalSGDState.process_group 1 0 0
attr PostLocalSGDState.subgroup 1 0 0
attr PostLocalSGDState.start_localSGD_iter 1 0 0
attr PostLocalSGDState.post_local_gradient_allreduce 1 0 0
attr PostLocalSGDState.iter 1 0 0
attr logger 1 0 0

torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook (30 missing, 0 any)

Symbol Typable Typed Any
meth PowerSGDState.init 11 1 0
meth PowerSGDState.getstate 1 0 0
meth PowerSGDState.setstate 2 0 0
meth PowerSGDState.maybe_increase_iter 2 0 0
meth PowerSGDState.compression_stats 1 0 0
attr PowerSGDState.process_group 1 0 0
attr PowerSGDState.matrix_approximation_rank 1 0 0
attr PowerSGDState.start_powerSGD_iter 1 0 0
attr PowerSGDState.min_compression_rate 1 0 0
attr PowerSGDState.use_error_feedback 1 0 0
attr PowerSGDState.warm_start 1 0 0
attr PowerSGDState.orthogonalization_epsilon 1 0 0
attr PowerSGDState.rng 1 0 0
attr PowerSGDState.iter 1 0 0
attr PowerSGDState.total_numel_before_compression 1 0 0
attr PowerSGDState.total_numel_after_compression 1 0 0
attr PowerSGDState.compression_stats_logging_frequency 1 0 0
attr PowerSGDState.next_stats_report 1 0 0
attr PowerSGDState.batch_tensors_with_same_shape 1 0 0

torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks (1 missing, 0 any)

Symbol Typable Typed Any
func quantization_perchannel_hook 4 3 0

torch.distributed.algorithms.join (8 missing, 1 any)

Symbol Typable Typed Any
meth Join.init 5 3 0
meth Join.enter 1 0 0
meth Join.exit 4 3 0
meth Join._get_num_nonjoined_procs 1 0 0
meth Join._notify_procs_to_terminate 1 0 0
meth Join.notify_join_context 2 1 0
meth Joinable.join_hook 2 1 0
prop Joinable.join_process_group 1 1 1

torch.distributed.algorithms.model_averaging.averagers (11 missing, 0 any)

Symbol Typable Typed Any
meth PeriodicModelAverager.init 4 1 0
meth PeriodicModelAverager.average_parameters 2 1 0
attr PeriodicModelAverager.warmup_steps 1 0 0
attr PeriodicModelAverager.period 1 0 0
meth ModelAverager.init 2 1 0
meth ModelAverager.average_parameters 2 0 0
attr ModelAverager.process_group 1 0 0
attr ModelAverager.step 1 0 0

torch.distributed.algorithms.model_averaging.hierarchical_model_averager (9 missing, 0 any)

Symbol Typable Typed Any
meth HierarchicalModelAverager.init 4 0 0
meth HierarchicalModelAverager._find_process_group 1 0 0
meth HierarchicalModelAverager.average_parameters 2 1 0
attr HierarchicalModelAverager.period_process_group_dict 1 0 0
attr HierarchicalModelAverager.warmup_steps 1 0 0
attr logger 1 0 0

torch.distributed.algorithms.model_averaging.utils (3 missing, 0 any)

Symbol Typable Typed Any
func average_parameters_or_parameter_groups 3 2 0
func get_params_to_average 2 1 0
func average_parameters 3 2 0

torch.distributed.argparse_util (17 missing, 0 any)

Symbol Typable Typed Any
meth env.init 5 1 0
meth env.call 5 0 0
meth check_env.init 4 1 0
meth check_env.call 5 0 0

torch.distributed.checkpoint.api (1 missing, 0 any)

Symbol Typable Typed Any
meth CheckpointException.init 3 2 0

torch.distributed.checkpoint.default_planner (9 missing, 2 any)

Symbol Typable Typed Any
meth DefaultLoadPlanner.resolve_tensor 2 1 0
meth DefaultLoadPlanner.transform_tensor 3 2 0
attr DefaultLoadPlanner.flatten_state_dict 1 0 0
attr DefaultLoadPlanner.flatten_sharded_tensors 1 0 0
attr DefaultLoadPlanner.allow_partial_load 1 0 0
meth DefaultSavePlanner.lookup_object 2 2 1
meth DefaultSavePlanner.transform_object 3 2 1
attr DefaultSavePlanner.flatten_state_dict 1 0 0
attr DefaultSavePlanner.flatten_sharded_tensors 1 0 0
attr DefaultSavePlanner.dedup_save_to_lowest_rank 1 0 0

torch.distributed.checkpoint.filesystem (7 missing, 4 any)

Symbol Typable Typed Any
meth FileSystemReader._slice_file 3 2 0
meth FileSystemReader.read_metadata 3 3 2
meth FileSystemReader.set_up_storage_reader 5 5 2
attr FileSystemReader.fs 1 0 0
attr FileSystemReader.path 1 0 0
attr FileSystemReader.load_id 1 0 0
attr FileSystemReader.transforms 1 0 0
attr FileSystemReader.rank 1 0 0
attr FileSystemReader.use_collectives 1 0 0

torch.distributed.checkpoint.format_utils (4 missing, 0 any)

Symbol Typable Typed Any
func dcp_to_torch_save 3 2 0
attr BroadcastingTorchSaveReader.checkpoint_id 1 0 0
attr BroadcastingTorchSaveReader.coordinator_rank 1 0 0
func torch_save_to_dcp 3 2 0

torch.distributed.checkpoint.hf_storage (4 missing, 0 any)

Symbol Typable Typed Any
meth HuggingFaceStorageReader._process_read_request 4 3 0
attr HuggingFaceStorageReader.thread_count 1 0 0
attr HuggingFaceStorageWriter.thread_count_consolidation 1 0 0
attr HuggingFaceStorageWriter.path 1 0 0

torch.distributed.checkpoint.metadata (4 missing, 0 any)

Symbol Typable Typed Any
meth TensorProperties.getstate 1 0 0
meth TensorProperties.setstate 2 0 0
meth MetadataIndex.init 4 3 0

torch.distributed.checkpoint.quantized_hf_storage (3 missing, 3 any)

Symbol Typable Typed Any
meth QuantizedHuggingFaceStorageReader.init 5 4 0
meth QuantizedHuggingFaceStorageReader.read_metadata 1 1 1
meth QuantizedHuggingFaceStorageReader._load_quantization_metadata 1 0 0
meth QuantizedHuggingFaceStorageReader._build_weight_scale_mapping 2 1 0
meth QuantizedHuggingFaceStorageReader._process_read_request 4 4 1
meth QuantizedHuggingFaceStorageReader._read_quantized_tensor_with_block_alignment 3 3 1

torch.distributed.checkpoint.staging (4 missing, 2 any)

Symbol Typable Typed Any
meth DefaultStager.init 2 1 0
meth DefaultStager.stage 3 3 1
meth DefaultStager._stage 3 3 1
meth BlockingAsyncStager.init 3 2 0
attr BlockingAsyncStager.cache_staged_state_dict 1 0 0
attr BlockingAsyncStager.type_check 1 0 0

torch.distributed.checkpoint.storage (0 missing, 6 any)

Symbol Typable Typed Any
meth StorageReader.read_metadata 3 3 2
meth StorageReader.set_up_storage_reader 5 5 2
meth StorageWriter.set_up_storage_writer 4 4 2

torch.distributed.checkpoint.utils (0 missing, 1 any)

Symbol Typable Typed Any
func find_state_dict_object 3 3 1

torch.distributed.collective_utils (0 missing, 1 any)

Symbol Typable Typed Any
func all_gather_object_enforce_type 5 5 1

torch.distributed.device_mesh (5 missing, 0 any)

Symbol Typable Typed Any
meth DeviceMesh._setup_world_group_and_device 1 0 0
meth DeviceMesh.exit 4 1 0
meth DeviceMesh.hash 1 0 0

torch.distributed.distributed_c10d (96 missing, 1 any)

Symbol Typable Typed Any
func new_subgroups 7 0 0
func all_to_all_single 7 1 0
func broadcast_object_list 6 5 0
func reduce_scatter_tensor 6 0 0
func destroy_process_group 2 1 0
func scatter_object_list 6 5 0
meth BackendConfig.init 2 1 0
meth BackendConfig.repr 1 0 0
attr reduce_op 1 0 0
func all_gather 5 0 0
func all_to_all 5 1 0
meth P2POp.init 7 6 0
meth P2POp.new 7 6 0
meth P2POp.repr 1 0 0
attr P2POp.op 1 0 0
attr P2POp.tensor 1 0 0
attr P2POp.group 1 0 0
attr P2POp.peer 1 0 0
attr P2POp.tag 1 0 0
attr P2POp.group_peer 1 0 0
func all_reduce 5 1 0
func all_gather_object 4 0 0
func new_subgroups_by_enumeration 6 0 0
func scatter 7 6 0
func new_group 8 2 0
func monitored_barrier 4 2 0
func all_gather_into_tensor 5 0 0
func all_reduce_coalesced 5 1 0
func barrier 4 2 0
func all_gather_coalesced 5 1 0
func broadcast 6 5 0
func send_object_list 7 6 0
func gather 7 6 0
func recv_object_list 7 6 0
func reduce 7 5 0
meth Backend.new 2 1 0
meth Backend.register_backend 5 3 0
func reduce_scatter 6 1 0
func gather_object 6 5 1

torch.distributed.elastic.agent.server.api (23 missing, 1 any)

Symbol Typable Typed Any
meth WorkerGroup.init 2 1 0
attr WorkerGroup.spec 1 0 0
attr WorkerGroup.workers 1 0 0
attr WorkerGroup.store 1 0 0
attr WorkerGroup.group_rank 1 0 0
attr WorkerGroup.group_world_size 1 0 0
attr WorkerGroup.master_addr 1 0 0
attr WorkerGroup.master_port 1 0 0
attr WorkerGroup.state 1 0 0
meth SimpleElasticAgent.init 3 2 0
meth SimpleElasticAgent._assign_worker_ranks 5 4 0
meth SimpleElasticAgent.record_duration 2 1 0
meth SimpleElasticAgent._record_metrics 2 1 0
meth SimpleElasticAgent._record_metric_with_condition 3 0 0
meth SimpleElasticAgent._record_flakiness_metric 2 1 0
meth SimpleElasticAgent._exit_barrier 1 0 0
meth WorkerSpec.post_init 1 0 0
meth WorkerSpec.get_entrypoint_name 1 0 0
meth Worker.init 6 5 0
meth Worker.str 1 0 0
meth Worker.repr 1 0 0
attr Worker.id 1 1 1

torch.distributed.elastic.agent.server.local_elastic_agent (2 missing, 0 any)

Symbol Typable Typed Any
meth LocalElasticAgent.init 6 4 0

torch.distributed.elastic.events.api (2 missing, 0 any)

Symbol Typable Typed Any
meth RdzvEvent.str 1 0 0
meth Event.str 1 0 0

torch.distributed.elastic.metrics (1 missing, 0 any)

Symbol Typable Typed Any
func initialize_metrics 2 1 0

torch.distributed.elastic.metrics.api (18 missing, 0 any)

Symbol Typable Typed Any
meth MetricHandler.emit 2 1 0
meth ConsoleMetricHandler.emit 2 1 0
func configure 3 2 0
func get_elapsed_time_ms 2 1 0
func prof 3 1 0
meth MetricStream.init 3 2 0
meth MetricStream.add_value 3 2 0
attr MetricStream.group_name 1 0 0
attr MetricStream.handler 1 0 0
func publish_metric 4 3 0
func profile 2 0 0
func put_metric 4 3 0
func getStream 2 1 0
meth NullMetricHandler.emit 2 1 0
meth MetricsConfig.init 2 1 0
attr MetricsConfig.params 1 0 0

torch.distributed.elastic.multiprocessing.api (19 missing, 0 any)

Symbol Typable Typed Any
attr SignalException.sigval 1 0 0
meth SubprocessContext.init 10 9 0
meth SubprocessContext._start 1 0 0
meth SubprocessContext._capture_process_failures 2 1 0
meth DefaultLogsSpecs._make_log_dir 3 2 0
meth PContext.init 9 8 0
attr PContext.name 1 0 0
attr PContext.entrypoint 1 0 0
attr PContext.args 1 0 0
attr PContext.envs 1 0 0
attr PContext.stdouts 1 0 0
attr PContext.stderrs 1 0 0
attr PContext.error_files 1 0 0
attr PContext.nprocs 1 0 0
meth MultiprocessContext.init 11 10 0
meth MultiprocessContext._start 1 0 0
attr MultiprocessContext.start_method 1 0 0
func get_std_cm 3 1 0

torch.distributed.elastic.multiprocessing.errors (9 missing, 0 any)

Symbol Typable Typed Any
meth ProcessFailure.post_init 1 0 0
meth ProcessFailure._set_no_reply_file 1 0 0
meth ProcessFailure.timestamp_isoformat 1 0 0
meth ChildFailedError.init 3 2 0
meth ChildFailedError.format_msg 3 0 0
attr ChildFailedError.name 1 0 0
attr ChildFailedError.failures 1 0 0

torch.distributed.elastic.multiprocessing.errors.error_handler (4 missing, 0 any)

Symbol Typable Typed Any
meth ErrorHandler.override_error_code_in_rootcause_data 4 3 0
meth ErrorHandler.dump_error_file 3 2 0
meth ErrorHandler._rm 2 0 0

torch.distributed.elastic.multiprocessing.redirects (6 missing, 0 any)

Symbol Typable Typed Any
func get_libc 1 0 0
attr libc 1 0 0
attr redirect_stderr 1 0 0
attr redirect_stdout 1 0 0
func redirect 3 2 0
attr logger 1 0 0

torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler (2 missing, 0 any)

Symbol Typable Typed Any
meth SubprocessHandler.init 8 7 0
attr SubprocessHandler.local_rank_id 1 0 0

torch.distributed.elastic.multiprocessing.tail_log (2 missing, 0 any)

Symbol Typable Typed Any
func tail_logfile 7 6 0
meth TailLog.init 7 6 0

torch.distributed.elastic.rendezvous.api (12 missing, 2 any)

Symbol Typable Typed Any
attr rendezvous_handler_registry 1 0 0
meth RendezvousParameters.init 8 6 0
meth RendezvousParameters.get 3 3 2
attr RendezvousParameters.backend 1 0 0
attr RendezvousParameters.endpoint 1 0 0
attr RendezvousParameters.run_id 1 0 0
attr RendezvousParameters.min_nodes 1 0 0
attr RendezvousParameters.max_nodes 1 0 0
attr RendezvousParameters.config 1 0 0
attr RendezvousParameters.local_addr 1 0 0
meth RendezvousHandler.set_closed 1 0 0
meth RendezvousInfo.init 5 4 0

torch.distributed.elastic.rendezvous.c10d_rendezvous_backend (3 missing, 1 any)

Symbol Typable Typed Any
attr logger 1 0 0
meth C10dRendezvousBackend._call_store 4 2 1

torch.distributed.elastic.rendezvous.dynamic_rendezvous (8 missing, 0 any)

Symbol Typable Typed Any
meth DynamicRendezvousHandler.from_backend 10 9 0
meth DynamicRendezvousHandler._create_tcp_store_server 3 1 0
meth DynamicRendezvousHandler._keep_alive_weak 2 1 0
attr DynamicRendezvousHandler._node_desc_generator 1 0 0
meth RendezvousTimeout._set_timeouts 2 1 0
attr RendezvousTimeout._ZERO 1 0 0
attr RendezvousTimeout._DEFAULT_TIMEOUTS 1 0 0

torch.distributed.elastic.rendezvous.etcd_rendezvous (65 missing, 0 any)

Symbol Typable Typed Any
meth EtcdRendezvousHandler.init 3 2 0
meth EtcdRendezvousHandler.del 1 0 0
meth EtcdRendezvousHandler.next_rendezvous 1 0 0
meth EtcdRendezvousHandler.is_closed 1 0 0
meth EtcdRendezvousHandler.set_closed 1 0 0
meth EtcdRendezvousHandler.num_nodes_waiting 1 0 0
meth EtcdRendezvous.init 8 0 0
meth EtcdRendezvous.del 1 0 0
meth EtcdRendezvous.rendezvous_barrier 1 0 0
meth EtcdRendezvous.init_phase 1 0 0
meth EtcdRendezvous.join_phase 2 0 0
meth EtcdRendezvous.confirm_phase 3 0 0
meth EtcdRendezvous.handle_existing_rendezvous 2 0 0
meth EtcdRendezvous.try_create_rendezvous 1 0 0
meth EtcdRendezvous.join_rendezvous 2 0 0
meth EtcdRendezvous.wait_for_peers 2 0 0
meth EtcdRendezvous.confirm_membership 3 0 0
meth EtcdRendezvous.wait_for_final 2 0 0
meth EtcdRendezvous.announce_self_waiting 2 0 0
meth EtcdRendezvous.wait_for_rendezvous_to_free 2 0 0
meth EtcdRendezvous.handle_join_last_call 3 0 0
meth EtcdRendezvous.set_closed 1 0 0
meth EtcdRendezvous.get_rdzv_state 1 0 0
meth EtcdRendezvous.try_wait_for_state_change 3 0 0
meth EtcdRendezvous.get_path 2 0 0
meth EtcdRendezvous.create_path_if_not_exists 3 0 0
meth EtcdRendezvous.setup_lease_renewal 3 0 0
meth EtcdRendezvous.store_extra_data 4 0 0
meth EtcdRendezvous.load_extra_data 4 0 0
meth EtcdRendezvous.setup_kv_store 2 0 0
attr EtcdRendezvous.client 1 0 0

torch.distributed.elastic.rendezvous.etcd_server (6 missing, 0 any)

Symbol Typable Typed Any
func find_free_port 1 0 0
attr logger 1 0 0
func stop_etcd 3 1 0
meth EtcdServer.init 2 1 0
meth EtcdServer.get_client 1 0 0

torch.distributed.elastic.rendezvous.etcd_store (19 missing, 0 any)

Symbol Typable Typed Any
meth EtcdStore.init 4 1 0
meth EtcdStore.set 3 0 0
meth EtcdStore.get 2 1 0
meth EtcdStore.add 3 2 0
meth EtcdStore.wait 3 1 0
meth EtcdStore.check 2 1 0
meth EtcdStore._encode 2 1 0
meth EtcdStore._decode 2 1 0
meth EtcdStore._try_wait_get 3 0 0
attr EtcdStore.client 1 0 0
attr EtcdStore.prefix 1 0 0
func cas_delay 1 0 0

torch.distributed.elastic.rendezvous.static_tcp_rendezvous (10 missing, 0 any)

Symbol Typable Typed Any
meth StaticTCPRendezvous.init 7 6 0
meth StaticTCPRendezvous.is_closed 1 0 0
meth StaticTCPRendezvous.set_closed 1 0 0
meth StaticTCPRendezvous.num_nodes_waiting 1 0 0
attr StaticTCPRendezvous.master_addr 1 0 0
attr StaticTCPRendezvous.master_port 1 0 0
attr StaticTCPRendezvous.rank 1 0 0
attr StaticTCPRendezvous.world_size 1 0 0
attr StaticTCPRendezvous.run_id 1 0 0
attr StaticTCPRendezvous.timeout 1 0 0

torch.distributed.elastic.timer.api (14 missing, 3 any)

Symbol Typable Typed Any
func configure 2 1 0
meth TimerServer.init 4 3 0
meth TimerServer._reap_worker 2 2 1
meth TimerServer._reap_worker_no_throw 2 2 1
meth TimerServer._watchdog_loop 1 0 0
meth TimerServer._run_watchdog 1 0 0
meth TimerServer._get_scopes 2 0 0
meth TimerClient.release 2 1 0
func expires 4 3 0
meth TimerRequest.init 4 3 1
meth TimerRequest.eq 2 0 0
attr TimerRequest.worker_id 1 0 0
attr TimerRequest.scope_id 1 0 0
attr TimerRequest.expiration_time 1 0 0

torch.distributed.elastic.timer.debug_info_logging (1 missing, 0 any)

Symbol Typable Typed Any
func log_debug_info_for_expired_timers 3 2 0

torch.distributed.elastic.timer.file_based_local_timer (6 missing, 0 any)

Symbol Typable Typed Any
meth FileTimerRequest.eq 2 1 0
attr FileTimerRequest.version 1 0 0
attr FileTimerRequest.signal 1 0 0
meth FileTimerClient.init 3 2 0
attr FileTimerClient.signal 1 0 0
meth FileTimerServer.is_process_running 2 1 0

torch.distributed.elastic.timer.local_timer (10 missing, 0 any)

Symbol Typable Typed Any
meth LocalTimerServer.init 4 3 0
meth MultiprocessingRequestQueue.init 2 1 0
meth MultiprocessingRequestQueue.get 3 2 0
meth LocalTimerClient.init 2 0 0
meth LocalTimerClient.acquire 3 0 0
meth LocalTimerClient.release 2 0 0

torch.distributed.elastic.utils.data.cycling_iterator (1 missing, 0 any)

Symbol Typable Typed Any
meth CyclingIterator.init 4 3 0

torch.distributed.elastic.utils.data.elastic_distributed_sampler (4 missing, 0 any)

Symbol Typable Typed Any
meth ElasticDistributedSampler.init 5 4 0
attr ElasticDistributedSampler.start_index 1 0 0
attr ElasticDistributedSampler.num_samples 1 0 0
attr ElasticDistributedSampler.total_size 1 0 0

torch.distributed.elastic.utils.distributed (3 missing, 0 any)

Symbol Typable Typed Any
func get_free_port 1 0 0
func create_c10d_store 9 7 0

torch.distributed.elastic.utils.store (6 missing, 0 any)

Symbol Typable Typed Any
func get_all 5 3 0
func store_timeout 3 1 0
func barrier 8 7 0
func synchronize 7 6 0

torch.distributed.flight_recorder.components.builder (0 missing, 1 any)

Symbol Typable Typed Any
func build_groups_memberships 2 2 1

torch.distributed.flight_recorder.components.config_manager (2 missing, 0 any)

Symbol Typable Typed Any
meth JobConfig.init 1 0 0
attr JobConfig.parser 1 0 0

torch.distributed.flight_recorder.components.fr_logger (0 missing, 1 any)

Symbol Typable Typed Any
meth FlightRecorderLogger.new 1 1 1

torch.distributed.flight_recorder.components.types (38 missing, 2 any)

Symbol Typable Typed Any
attr MatchInfo.culprit 1 0 0
attr MatchStateRecord.expected_ranks 1 0 0
attr MatchStateRecord.other_ranks 1 0 0
attr MatchStateRecord.entry_state 1 0 0
attr MatchStateRecord.candidate_ranks 1 0 0
attr MatchStateRecord.candidate_idx 1 0 0
attr MatchStateRecord.found_ranks 1 0 0
attr MatchStateRecord.found_idx 1 0 0
attr MatchStateRecord.errors 1 0 0
attr MatchStateRecord.has_undecided_case 1 0 0
meth Op.init 4 3 0
attr Op.profiling_name 1 0 0
attr Op.state 1 0 0
attr Op.pg_name 1 0 0
attr Op.type 1 0 0
attr Op.pg_size 1 0 0
attr Op.collective_seq_id 1 0 0
attr Op.stack_id 1 0 0
attr Op.p2p_seq_id 1 0 0
attr Op.input_dtypes 1 0 0
attr Op.output_dtypes 1 0 0
attr Op.time_created_ns 1 0 0
attr Op.collective_frames 1 0 0
attr Op.is_verbose 1 0 0
attr Op.input_sizes 1 0 0
attr Op.output_sizes 1 0 0
meth EntryState.log 7 7 1
meth EntryState.to_nccl_call 5 5 1
attr EntryState.pg_name 1 0 0
attr EntryState.desc 1 0 0
attr EntryState.pg_desc 1 0 0
attr EntryState.profiling_name 1 0 0
attr EntryState.collective_seq_id 1 0 0
attr EntryState.p2p_seq_id 1 0 0
attr EntryState.record_id 1 0 0
attr EntryState.input_sizes 1 0 0
attr EntryState.output_sizes 1 0 0
attr EntryState.collective_state 1 0 0
attr EntryState.collective_frames 1 0 0
attr EntryState.expected_ranks 1 0 0

torch.distributed.fsdp._fully_shard._fully_shard (5 missing, 3 any)

Symbol Typable Typed Any
meth FSDPModule.new 3 0 0
meth FSDPModule.set_all_reduce_hook 3 2 0
meth FSDPModule._set_unshard_async_op 2 1 0
meth FSDPModule._apply 3 3 3

torch.distributed.fsdp.fully_sharded_data_parallel (18 missing, 5 any)

Symbol Typable Typed Any
meth FullyShardedDataParallel.init 17 16 0
meth FullyShardedDataParallel.getattr 2 2 1
meth FullyShardedDataParallel.getitem 2 2 1
meth FullyShardedDataParallel.forward 3 3 3
meth FullyShardedDataParallel._deregister_orig_params_ctx 1 0 0
meth FullyShardedDataParallel._apply 3 0 0
meth FullyShardedDataParallel.named_buffers 3 1 0
meth FullyShardedDataParallel.named_parameters 3 1 0
meth FullyShardedDataParallel._warn_optim_input 3 1 0
meth FullyShardedDataParallel._is_using_optim_input 3 1 0
meth FullyShardedDataParallel._warn_legacy_optim_state_dict 4 3 0
meth FullyShardedDataParallel.register_comm_hook 3 2 0
meth FullyShardedDataParallel._unshard 2 1 0
meth FullyShardedDataParallel._wait_unshard_streams_on_current_stream 1 0 0
meth FullyShardedDataParallel._use_training_state 3 2 0

torch.distributed.fsdp.sharded_grad_scaler (2 missing, 0 any)

Symbol Typable Typed Any
attr ShardedGradScaler.process_group 1 0 0
attr logger 1 0 0

torch.distributed.fsdp.wrap (9 missing, 3 any)

Symbol Typable Typed Any
meth ModuleWrapPolicy.init 2 1 0
meth ModuleWrapPolicy.call 5 0 0
func wrap 3 3 1
func enable_wrap 3 3 2
meth CustomPolicy.init 2 1 0
func always_wrap_policy 3 1 0

torch.distributed.launch (6 missing, 0 any)

Symbol Typable Typed Any
func parse_args 2 0 0
func launch 2 0 0
func main 2 0 0

torch.distributed.launcher.api (4 missing, 0 any)

Symbol Typable Typed Any
meth LaunchConfig.post_init 1 0 0
meth elastic_launch.init 3 2 0
meth elastic_launch.call 2 0 0

torch.distributed.nn.api.remote_module (1 missing, 0 any)

Symbol Typable Typed Any
meth RemoteModule.init 5 4 0

torch.distributed.nn.functional (39 missing, 0 any)

Symbol Typable Typed Any
func all_gather 3 0 0
func all_to_all 4 0 0
func gather 4 0 0
func scatter 4 0 0
func reduce 5 0 0
func all_to_all_single 6 0 0
func reduce_scatter 5 0 0
func all_reduce 4 0 0
func broadcast 4 0 0

torch.distributed.nn.jit.instantiator (6 missing, 0 any)

Symbol Typable Typed Any
func instantiate_scriptable_remote_module_template 3 0 0
func get_arg_return_types_from_interface 2 0 0
func instantiate_non_scriptable_remote_module_template 1 0 0

torch.distributed.nn.jit.templates.remote_module_template (1 missing, 0 any)

Symbol Typable Typed Any
func get_remote_module_template 2 1 0

torch.distributed.optim.optimizer (9 missing, 0 any)

Symbol Typable Typed Any
meth DistributedOptimizer.init 5 0 0
meth DistributedOptimizer.step 2 0 0
attr DistributedOptimizer.is_functional_optim 1 0 0
attr DistributedOptimizer.remote_optimizers 1 0 0

torch.distributed.optim.post_localSGD_optimizer (13 missing, 0 any)

Symbol Typable Typed Any
meth PostLocalSGDOptimizer.init 3 2 0
meth PostLocalSGDOptimizer.repr 1 0 0
meth PostLocalSGDOptimizer.state_dict 1 0 0
meth PostLocalSGDOptimizer.load_state_dict 2 0 0
meth PostLocalSGDOptimizer.step 1 0 0
meth PostLocalSGDOptimizer.zero_grad 2 1 0
meth PostLocalSGDOptimizer.add_param_group 2 0 0
prop PostLocalSGDOptimizer.state 1 0 0
attr PostLocalSGDOptimizer.optim 1 0 0
attr PostLocalSGDOptimizer.param_groups 1 0 0
attr PostLocalSGDOptimizer.averager 1 0 0

torch.distributed.optim.utils (6 missing, 0 any)

Symbol Typable Typed Any
func as_functional_optim 4 1 0
func register_functional_optim 3 0 0

torch.distributed.optim.zero_redundancy_optimizer (2 missing, 7 any)

Symbol Typable Typed Any
meth ZeroRedundancyOptimizer.init 7 7 2
meth ZeroRedundancyOptimizer._local_step 4 4 1
meth ZeroRedundancyOptimizer.join_hook 2 0 0
meth ZeroRedundancyOptimizer.join_process_group 1 1 1
attr ZeroRedundancyOptimizer.functional_optim_map 1 1 1
attr ZeroRedundancyOptimizer.process_group 1 1 1
attr ZeroRedundancyOptimizer.optim 1 1 1

torch.distributed.pipelining._IR (15 missing, 0 any)

Symbol Typable Typed Any
meth Pipe.init 5 3 0
meth Pipe.forward 3 0 0
meth Pipe._number_and_count_forward_stages 2 1 0
meth Pipe._from_traced 6 4 0
meth Pipe.print_readable 1 0 0
meth Pipe.from_tracing 5 4 0
meth Pipe.str 1 0 0
meth Pipe.repr 1 0 0
attr Pipe.has_loss_and_backward 1 0 0
attr Pipe.loss_spec 1 0 0
func pipe_split 1 0 0

torch.distributed.pipelining.microbatch (9 missing, 0 any)

Symbol Typable Typed Any
func merge_chunks 3 1 0
meth TensorChunkSpec.init 2 0 0
meth TensorChunkSpec.repr 1 0 0
meth TensorChunkSpec.str 1 0 0
meth TensorChunkSpec.from_tuple 2 1 0
meth TensorChunkSpec.from_dict 2 1 0
attr TensorChunkSpec.split_dim 1 0 0

torch.distributed.pipelining.schedules (60 missing, 0 any)

Symbol Typable Typed Any
meth ScheduleInterleavedZeroBubble.init 9 8 0
meth ScheduleInterleavedZeroBubble._calculate_single_rank_operations 2 1 0
meth ScheduleInterleavedZeroBubble._add_bubbles_to_actions 2 0 0
attr ScheduleInterleavedZeroBubble.pp_group_size 1 0 0
attr ScheduleInterleavedZeroBubble.n_local_stages 1 0 0
attr ScheduleInterleavedZeroBubble.rank 1 0 0
attr ScheduleInterleavedZeroBubble.number_of_rounds 1 0 0
attr ScheduleInterleavedZeroBubble.microbatches_per_round 1 0 0
meth ScheduleInterleaved1F1B.init 9 8 0
meth ScheduleInterleaved1F1B._calculate_single_rank_operations 2 1 0
attr ScheduleInterleaved1F1B.pp_group_size 1 0 0
attr ScheduleInterleaved1F1B.n_local_stages 1 0 0
attr ScheduleInterleaved1F1B.rank 1 0 0
attr ScheduleInterleaved1F1B.number_of_rounds 1 0 0
attr ScheduleInterleaved1F1B.microbatches_per_round 1 0 0
meth ScheduleLoopedBFS.init 7 6 0
meth ScheduleLoopedBFS._calculate_single_rank_operations 2 0 0
meth ScheduleGPipe._step_microbatches 6 5 0
meth ScheduleZBVZeroBubble.init 9 8 0
meth ScheduleZBVZeroBubble._calculate_single_rank_operations 2 1 0
attr ScheduleZBVZeroBubble.pp_group_size 1 0 0
attr ScheduleZBVZeroBubble.stage_index_to_group_rank 1 0 0
attr ScheduleZBVZeroBubble.n_local_stages 1 0 0
attr ScheduleZBVZeroBubble.rank 1 0 0
attr ScheduleZBVZeroBubble.num_stages 1 0 0
meth ScheduleDualPipeV.init 9 8 0
meth ScheduleDualPipeV._calculate_single_rank_operations 2 1 0
attr ScheduleDualPipeV.pp_group_size 1 0 0
attr ScheduleDualPipeV.stage_index_to_group_rank 1 0 0
attr ScheduleDualPipeV.n_local_stages 1 0 0
attr ScheduleDualPipeV.rank 1 0 0
attr ScheduleDualPipeV.num_stages 1 0 0
meth Schedule1F1B._step_microbatches 6 5 0
func get_schedule_class 2 1 0
meth PipelineScheduleMulti.init 10 9 0
meth PipelineScheduleMulti._initialize_stages 3 1 0
meth PipelineScheduleMulti._dump_csv 2 0 0
meth PipelineScheduleMulti._load_csv 3 0 0
meth PipelineScheduleMulti.step 6 2 0
meth PipelineScheduleMulti._step_microbatches 6 5 0
attr PipelineScheduleMulti.pp_group_size 1 0 0
attr PipelineScheduleMulti.rank 1 0 0
attr PipelineScheduleMulti.stage_index_to_group_rank 1 0 0
meth PipelineScheduleSingle.init 8 7 0
meth PipelineScheduleSingle._initialize_stage 3 0 0
meth PipelineScheduleSingle.step 6 2 0

torch.distributed.pipelining.stage (2 missing, 0 any)

Symbol Typable Typed Any
meth PipelineStage.init 9 8 0
meth PipelineStage._shape_inference 3 2 0

torch.distributed.rendezvous (5 missing, 0 any)

Symbol Typable Typed Any
func register_rendezvous_handler 3 0 0
func rendezvous 5 3 0

torch.distributed.rpc (6 missing, 0 any)

Symbol Typable Typed Any
func init_rpc 6 0 0

torch.distributed.rpc.api (29 missing, 0 any)

Symbol Typable Typed Any
func remote 6 0 0
func rpc_async 6 0 0
attr new_method 1 0 0
func shutdown 3 0 0
meth AllGatherStates.init 1 0 0
attr AllGatherStates.gathered_objects 1 0 0
attr AllGatherStates.proceed_signal 1 0 0
func method_factory 3 0 0
func rpc_sync 6 1 0
func get_worker_info 2 0 0

torch.distributed.rpc.backend_registry (16 missing, 0 any)

Symbol Typable Typed Any
attr BackendType 1 0 0
func register_backend 4 0 0
func init_backend 4 0 0
func construct_rpc_backend_options 5 0 0
func backend_registered 2 0 0

torch.distributed.rpc.functions (2 missing, 0 any)

Symbol Typable Typed Any
func async_execution 2 0 0

torch.distributed.rpc.internal (5 missing, 0 any)

Symbol Typable Typed Any
func deserialize 3 0 0
func serialize 2 0 0

torch.distributed.rpc.options (3 missing, 0 any)

Symbol Typable Typed Any
meth TensorPipeRpcBackendOptions.init 8 7 0
meth TensorPipeRpcBackendOptions.set_device_map 3 2 0
meth TensorPipeRpcBackendOptions.set_devices 2 1 0

torch.distributed.rpc.rref_proxy (9 missing, 0 any)

Symbol Typable Typed Any
meth RRefProxy.init 4 0 0
meth RRefProxy.getattr 2 0 0
attr RRefProxy.rref 1 0 0
attr RRefProxy.rpc_api 1 0 0
attr RRefProxy.rpc_timeout 1 0 0

torch.distributed.run (14 missing, 0 any)

Symbol Typable Typed Any
attr logger 1 0 0
func determine_local_world_size 2 1 0
func run_script_path 3 2 0
func get_use_env 2 1 0
func parse_args 2 0 0
func get_rdzv_endpoint 2 0 0
func run 2 0 0
func config_from_args 2 1 0
func main 2 0 0
func parse_min_max_nnodes 2 1 0

torch.distributed.tensor._api (30 missing, 1 any)

Symbol Typable Typed Any
func zeros 7 6 0
func rand 7 6 0
func empty 7 6 0
func ones 7 6 0
func full 8 6 0
func randn 7 6 0
meth DTensor.init 3 0 0
meth DTensor.repr 1 0 0
meth DTensor.tensor_flatten 1 0 0
meth DTensor.tensor_unflatten 5 0 0
meth DTensor.coerce_tangent_metadata 1 0 0
meth DTensor.coerce_same_metadata_as_tangent 3 0 0
meth DTensor.torch_dispatch 5 0 0
meth DTensor.create_write_items 3 2 1
meth DTensor.create_chunk_list 1 0 0
meth DTensor.get_tensor_shard 2 0 0

torch.distributed.tensor.debug._comm_mode (26 missing, 0 any)

Symbol Typable Typed Any
meth CommDebugMode.init 1 0 0
meth CommDebugMode.generate_json_dump 3 0 0
meth CommDebugMode.generate_comm_debug_tracing_table 2 0 0
meth CommDebugMode._get_operations_list 2 0 0
meth CommDebugMode.enter 1 0 0
meth CommDebugMode.exit 2 0 0
meth CommDebugMode.log_comm_debug_tracing_table_to_file 3 0 0
meth CommDebugMode._set_noise_parameters 2 0 0
meth CommDebugMode.torch_dispatch 5 0 0
meth CommDebugMode.repr 1 0 0
attr CommDebugMode.comm_module_counts 1 0 0
attr CommDebugMode.comm_module_operation_counts 1 0 0
attr CommDebugMode.comm_registry 1 0 0
attr CommDebugMode.advanced_module_tracker 1 0 0

torch.distributed.tensor.debug._visualize_sharding (3 missing, 0 any)

Symbol Typable Typed Any
func visualize_sharding 4 1 0

torch.distributed.tensor.experimental._func_map (1 missing, 0 any)

Symbol Typable Typed Any
func local_map 7 6 0

torch.distributed.tensor.experimental._register_sharding (1 missing, 0 any)

Symbol Typable Typed Any
func register_sharding 2 1 0

torch.distributed.tensor.parallel.fsdp (4 missing, 1 any)

Symbol Typable Typed Any
meth DTensorExtensions.init 2 1 0
meth DTensorExtensions.post_unflatten_transform 3 3 1
attr DTensorExtensions.compute_stream 1 0 0
attr DTensorExtensions.device_handle 1 0 0
attr DTensorExtensions.post_unflatten_transform 1 0 0

torch.distributed.tensor.parallel.loss (1 missing, 0 any)

Symbol Typable Typed Any
func loss_parallel 1 0 0

torch.distributed.tensor.parallel.style (88 missing, 1 any)

Symbol Typable Typed Any
meth PrepareModuleOutput.init 4 3 0
meth PrepareModuleOutput._prepare_out_fn 3 0 0
attr PrepareModuleOutput.output_layouts 1 0 0
attr PrepareModuleOutput.desired_output_layouts 1 0 0
attr PrepareModuleOutput.use_local_output 1 0 0
meth PrepareModuleInputOutput.init 9 8 0
attr PrepareModuleInputOutput.prepare_module_input 1 0 0
attr PrepareModuleInputOutput.prepare_module_output 1 0 0
meth RowwiseParallel.init 4 3 0
meth RowwiseParallel._prepare_input_fn 6 0 0
meth RowwiseParallel._partition_linear_fn 4 0 0
meth RowwiseParallel._partition_embedding_fn 4 0 0
meth RowwiseParallel._prepare_output_fn 6 0 0
attr RowwiseParallel.input_layouts 1 0 0
attr RowwiseParallel.output_layouts 1 0 0
attr RowwiseParallel.use_local_output 1 0 0
meth PrepareModuleInput.init 6 5 0
meth PrepareModuleInput._prepare_input_arg 5 4 1
meth PrepareModuleInput._prepare_input_fn 3 0 0
meth PrepareModuleInput._prepare_input_kwarg_fn 4 0 0
attr PrepareModuleInput.input_layouts 1 0 0
attr PrepareModuleInput.desired_input_layouts 1 0 0
attr PrepareModuleInput.use_local_output 1 0 0
attr PrepareModuleInput.with_kwargs 1 0 0
attr PrepareModuleInput.input_kwarg_layouts 1 0 0
attr PrepareModuleInput.desired_input_kwarg_layouts 1 0 0
meth ColwiseParallel.init 4 3 0
meth ColwiseParallel._prepare_input_fn 6 0 0
meth ColwiseParallel._partition_linear_fn 4 0 0
meth ColwiseParallel._partition_embedding_fn 4 0 0
meth ColwiseParallel._prepare_output_fn 6 0 0
attr ColwiseParallel.input_layouts 1 0 0
attr ColwiseParallel.output_layouts 1 0 0
attr ColwiseParallel.desired_input_layouts 1 0 0
attr ColwiseParallel.use_local_output 1 0 0
meth SequenceParallel.init 3 2 0
meth SequenceParallel._replicate_module_fn 4 3 0
meth SequenceParallel._prepare_input_fn 5 0 0
meth SequenceParallel._prepare_output_fn 5 0 0
attr SequenceParallel.sequence_sharding 1 0 0
attr SequenceParallel.use_local_output 1 0 0

torch.distributed.tensor.placement_types (13 missing, 0 any)

Symbol Typable Typed Any
meth Shard._maybe_unpad_tensor_with_sizes 6 1 0
meth Shard._select_shard 3 2 0
meth MaskPartial.init 7 0 0

torch.distributions.bernoulli (15 missing, 0 any)

Symbol Typable Typed Any
meth Bernoulli.expand 3 0 0
meth Bernoulli._new 3 0 0
meth Bernoulli.sample 2 0 0
meth Bernoulli.log_prob 2 0 0
meth Bernoulli.entropy 1 0 0
meth Bernoulli.enumerate_support 2 0 0
meth Bernoulli._log_normalizer 2 0 0

torch.distributions.beta (9 missing, 0 any)

Symbol Typable Typed Any
meth Beta.expand 3 0 0
meth Beta.log_prob 2 0 0
meth Beta.entropy 1 0 0
meth Beta._log_normalizer 3 0 0

torch.distributions.binomial (15 missing, 0 any)

Symbol Typable Typed Any
meth Binomial.expand 3 0 0
meth Binomial._new 3 0 0
meth Binomial.support 1 0 0
meth Binomial.sample 2 0 0
meth Binomial.log_prob 2 0 0
meth Binomial.entropy 1 0 0
meth Binomial.enumerate_support 2 0 0
attr Binomial.total_count 1 0 0

torch.distributions.categorical (16 missing, 0 any)

Symbol Typable Typed Any
meth Categorical.expand 3 0 0
meth Categorical._new 3 0 0
meth Categorical.support 1 0 0
meth Categorical.sample 2 0 0
meth Categorical.log_prob 2 0 0
meth Categorical.entropy 1 0 0
meth Categorical.enumerate_support 2 0 0
attr Categorical.probs 1 0 0
attr Categorical.logits 1 0 0

torch.distributions.cauchy (10 missing, 0 any)

Symbol Typable Typed Any
meth Cauchy.expand 3 0 0
meth Cauchy.log_prob 2 0 0
meth Cauchy.cdf 2 0 0
meth Cauchy.icdf 2 0 0
meth Cauchy.entropy 1 0 0

torch.distributions.chi2 (3 missing, 0 any)

Symbol Typable Typed Any
meth Chi2.expand 3 0 0

torch.distributions.constraint_registry (8 missing, 0 any)

Symbol Typable Typed Any
attr biject_to 1 0 0
attr transform_to 1 0 0
meth ConstraintRegistry.init 1 0 0
meth ConstraintRegistry.register 3 0 0
meth ConstraintRegistry.call 2 0 0

torch.distributions.constraints (29 missing, 0 any)

Symbol Typable Typed Any
attr corr_cholesky 1 0 0
attr lower_cholesky 1 0 0
attr simplex 1 0 0
meth MixtureSameFamilyConstraint.init 2 0 0
meth MixtureSameFamilyConstraint.check 2 0 0
meth MixtureSameFamilyConstraint.repr 1 0 0
attr MixtureSameFamilyConstraint.base_constraint 1 0 0
attr boolean 1 0 0
attr symmetric 1 0 0
func is_dependent 2 0 0
attr one_hot 1 0 0
attr nonnegative 1 0 0
attr unit_interval 1 0 0
attr lower_triangular 1 0 0
attr dependent 1 0 0
meth Constraint.check 2 0 0
meth Constraint.repr 1 0 0
attr real_vector 1 0 0
attr nonnegative_integer 1 0 0
attr positive_definite 1 0 0
attr real 1 0 0
attr positive_semidefinite 1 0 0
attr square 1 0 0
attr positive 1 0 0
attr positive_integer 1 0 0

torch.distributions.continuous_bernoulli (21 missing, 0 any)

Symbol Typable Typed Any
meth ContinuousBernoulli.expand 3 0 0
meth ContinuousBernoulli._new 3 0 0
meth ContinuousBernoulli._outside_unstable_region 1 0 0
meth ContinuousBernoulli._cut_probs 1 0 0
meth ContinuousBernoulli._cont_bern_log_norm 1 0 0
meth ContinuousBernoulli.sample 2 0 0
meth ContinuousBernoulli.log_prob 2 0 0
meth ContinuousBernoulli.cdf 2 0 0
meth ContinuousBernoulli.icdf 2 0 0
meth ContinuousBernoulli.entropy 1 0 0
meth ContinuousBernoulli._log_normalizer 2 0 0
attr ContinuousBernoulli.probs 1 0 0

torch.distributions.dirichlet (10 missing, 0 any)

Symbol Typable Typed Any
meth Dirichlet.expand 3 0 0
meth Dirichlet.log_prob 2 0 0
meth Dirichlet.entropy 1 0 0
meth Dirichlet._log_normalizer 2 0 0
attr Dirichlet.arg_constraints 1 0 0
attr Dirichlet.concentration 1 0 0

torch.distributions.distribution (5 missing, 0 any)

Symbol Typable Typed Any
meth Distribution.expand 3 1 0
meth Distribution._get_checked_instance 3 0 0

torch.distributions.exp_family (3 missing, 0 any)

Symbol Typable Typed Any
meth ExponentialFamily._log_normalizer 2 0 0
meth ExponentialFamily.entropy 1 0 0

torch.distributions.exponential (12 missing, 0 any)

Symbol Typable Typed Any
meth Exponential.expand 3 0 0
meth Exponential.log_prob 2 0 0
meth Exponential.cdf 2 0 0
meth Exponential.icdf 2 0 0
meth Exponential.entropy 1 0 0
meth Exponential._log_normalizer 2 0 0

torch.distributions.fishersnedecor (5 missing, 0 any)

Symbol Typable Typed Any
meth FisherSnedecor.expand 3 0 0
meth FisherSnedecor.log_prob 2 0 0

torch.distributions.gamma (11 missing, 0 any)

Symbol Typable Typed Any
meth Gamma.expand 3 0 0
meth Gamma.log_prob 2 0 0
meth Gamma.entropy 1 0 0
meth Gamma._log_normalizer 3 0 0
meth Gamma.cdf 2 0 0

torch.distributions.generalized_pareto (27 missing, 0 any)

Symbol Typable Typed Any
meth GeneralizedPareto.init 5 0 0
meth GeneralizedPareto.expand 3 0 0
meth GeneralizedPareto.rsample 2 0 0
meth GeneralizedPareto.log_prob 2 0 0
meth GeneralizedPareto.log_survival_function 2 0 0
meth GeneralizedPareto.log_cdf 2 0 0
meth GeneralizedPareto.cdf 2 0 0
meth GeneralizedPareto.icdf 2 0 0
meth GeneralizedPareto._z 2 0 0
meth GeneralizedPareto.entropy 1 0 0
meth GeneralizedPareto.support 1 0 0
prop GeneralizedPareto.mean 1 0 0
prop GeneralizedPareto.variance 1 0 0
prop GeneralizedPareto.mode 1 0 0

torch.distributions.geometric (8 missing, 0 any)

Symbol Typable Typed Any
meth Geometric.expand 3 0 0
meth Geometric.sample 2 0 0
meth Geometric.log_prob 2 0 0
meth Geometric.entropy 1 0 0

torch.distributions.gumbel (6 missing, 0 any)

Symbol Typable Typed Any
meth Gumbel.expand 3 0 0
meth Gumbel.log_prob 2 0 0
meth Gumbel.entropy 1 0 0

torch.distributions.half_cauchy (10 missing, 0 any)

Symbol Typable Typed Any
meth HalfCauchy.expand 3 0 0
meth HalfCauchy.log_prob 2 0 0
meth HalfCauchy.cdf 2 0 0
meth HalfCauchy.icdf 2 0 0
meth HalfCauchy.entropy 1 0 0

torch.distributions.half_normal (10 missing, 0 any)

Symbol Typable Typed Any
meth HalfNormal.expand 3 0 0
meth HalfNormal.log_prob 2 0 0
meth HalfNormal.cdf 2 0 0
meth HalfNormal.icdf 2 0 0
meth HalfNormal.entropy 1 0 0

torch.distributions.independent (12 missing, 0 any)

Symbol Typable Typed Any
meth Independent.expand 3 0 0
meth Independent.support 1 0 0
meth Independent.sample 2 1 0
meth Independent.log_prob 2 0 0
meth Independent.entropy 1 0 0
meth Independent.enumerate_support 2 0 0
meth Independent.repr 1 0 0
attr Independent.reinterpreted_batch_ndims 1 0 0

torch.distributions.inverse_gamma (4 missing, 0 any)

Symbol Typable Typed Any
meth InverseGamma.expand 3 0 0
meth InverseGamma.entropy 1 0 0

torch.distributions.kl (3 missing, 0 any)

Symbol Typable Typed Any
func register_kl 3 0 0

torch.distributions.kumaraswamy (4 missing, 0 any)

Symbol Typable Typed Any
meth Kumaraswamy.expand 3 0 0
meth Kumaraswamy.entropy 1 0 0

torch.distributions.laplace (10 missing, 0 any)

Symbol Typable Typed Any
meth Laplace.expand 3 0 0
meth Laplace.log_prob 2 0 0
meth Laplace.cdf 2 0 0
meth Laplace.icdf 2 0 0
meth Laplace.entropy 1 0 0

torch.distributions.lkj_cholesky (8 missing, 0 any)

Symbol Typable Typed Any
meth LKJCholesky.expand 3 0 0
meth LKJCholesky.sample 2 0 0
meth LKJCholesky.log_prob 2 0 0
attr LKJCholesky.dim 1 0 0

torch.distributions.log_normal (4 missing, 0 any)

Symbol Typable Typed Any
meth LogNormal.expand 3 0 0
meth LogNormal.entropy 1 0 0

torch.distributions.logistic_normal (3 missing, 0 any)

Symbol Typable Typed Any
meth LogisticNormal.expand 3 0 0

torch.distributions.lowrank_multivariate_normal (9 missing, 0 any)

Symbol Typable Typed Any
meth LowRankMultivariateNormal.expand 3 0 0
meth LowRankMultivariateNormal.log_prob 2 0 0
meth LowRankMultivariateNormal.entropy 1 0 0
attr LowRankMultivariateNormal.arg_constraints 1 0 0
attr LowRankMultivariateNormal.loc 1 0 0
attr LowRankMultivariateNormal.cov_diag 1 0 0

torch.distributions.mixture_same_family (15 missing, 0 any)

Symbol Typable Typed Any
meth MixtureSameFamily.expand 3 0 0
meth MixtureSameFamily.support 1 0 0
meth MixtureSameFamily.cdf 2 0 0
meth MixtureSameFamily.log_prob 2 0 0
meth MixtureSameFamily.sample 2 0 0
meth MixtureSameFamily._pad 2 0 0
meth MixtureSameFamily._pad_mixture_dimensions 2 0 0
meth MixtureSameFamily.repr 1 0 0

torch.distributions.multinomial (12 missing, 0 any)

Symbol Typable Typed Any
meth Multinomial.expand 3 0 0
meth Multinomial._new 3 0 0
meth Multinomial.support 1 0 0
meth Multinomial.sample 2 0 0
meth Multinomial.entropy 1 0 0
meth Multinomial.log_prob 2 0 0

torch.distributions.multivariate_normal (10 missing, 0 any)

Symbol Typable Typed Any
meth MultivariateNormal.expand 3 0 0
meth MultivariateNormal.log_prob 2 0 0
meth MultivariateNormal.entropy 1 0 0
attr MultivariateNormal.loc 1 0 0
attr MultivariateNormal.scale_tril 1 0 0
attr MultivariateNormal.covariance_matrix 1 0 0
attr MultivariateNormal.precision_matrix 1 0 0

torch.distributions.negative_binomial (12 missing, 0 any)

Symbol Typable Typed Any
meth NegativeBinomial.expand 3 0 0
meth NegativeBinomial._new 3 0 0
meth NegativeBinomial.sample 2 0 0
meth NegativeBinomial.log_prob 2 0 0
attr NegativeBinomial.arg_constraints 1 0 0
attr NegativeBinomial.total_count 1 0 0

torch.distributions.normal (15 missing, 0 any)

Symbol Typable Typed Any
meth Normal.expand 3 0 0
meth Normal.sample 2 0 0
meth Normal.log_prob 2 0 0
meth Normal.cdf 2 0 0
meth Normal.icdf 2 0 0
meth Normal.entropy 1 0 0
meth Normal._log_normalizer 3 0 0

torch.distributions.one_hot_categorical (13 missing, 0 any)

Symbol Typable Typed Any
meth OneHotCategorical.expand 3 0 0
meth OneHotCategorical._new 3 0 0
meth OneHotCategorical.sample 2 0 0
meth OneHotCategorical.log_prob 2 0 0
meth OneHotCategorical.entropy 1 0 0
meth OneHotCategorical.enumerate_support 2 0 0

torch.distributions.poisson (9 missing, 0 any)

Symbol Typable Typed Any
meth Poisson.expand 3 0 0
meth Poisson.sample 2 0 0
meth Poisson.log_prob 2 0 0
meth Poisson._log_normalizer 2 0 0

torch.distributions.relaxed_bernoulli (12 missing, 0 any)

Symbol Typable Typed Any
meth RelaxedBernoulli.expand 3 0 0
meth LogitRelaxedBernoulli.expand 3 0 0
meth LogitRelaxedBernoulli._new 3 0 0
meth LogitRelaxedBernoulli.log_prob 2 0 0
attr LogitRelaxedBernoulli.temperature 1 0 0

torch.distributions.relaxed_categorical (12 missing, 0 any)

Symbol Typable Typed Any
meth RelaxedOneHotCategorical.expand 3 0 0
meth ExpRelaxedCategorical.expand 3 0 0
meth ExpRelaxedCategorical._new 3 0 0
meth ExpRelaxedCategorical.log_prob 2 0 0
attr ExpRelaxedCategorical.temperature 1 0 0

torch.distributions.studentT (6 missing, 0 any)

Symbol Typable Typed Any
meth StudentT.expand 3 0 0
meth StudentT.log_prob 2 0 0
meth StudentT.entropy 1 0 0

torch.distributions.transformed_distribution (16 missing, 0 any)

Symbol Typable Typed Any
meth TransformedDistribution.expand 3 0 0
meth TransformedDistribution.support 1 0 0
meth TransformedDistribution.sample 2 0 0
meth TransformedDistribution.log_prob 2 0 0
meth TransformedDistribution._monotonize_cdf 2 0 0
meth TransformedDistribution.cdf 2 0 0
meth TransformedDistribution.icdf 2 0 0
attr TransformedDistribution.base_dist 1 0 0
attr TransformedDistribution.transforms 1 0 0

torch.distributions.transforms (239 missing, 0 any)

Symbol Typable Typed Any
meth AbsTransform.eq 2 0 0
meth AbsTransform._call 2 0 0
meth AbsTransform._inverse 2 0 0
meth SigmoidTransform.eq 2 0 0
meth SigmoidTransform._call 2 0 0
meth SigmoidTransform._inverse 2 0 0
meth SigmoidTransform.log_abs_det_jacobian 3 0 0
meth TanhTransform.eq 2 0 0
meth TanhTransform._call 2 0 0
meth TanhTransform._inverse 2 0 0
meth TanhTransform.log_abs_det_jacobian 3 0 0
attr TanhTransform.codomain 1 0 0
meth StackTransform.with_cache 2 0 0
meth StackTransform._slice 2 0 0
meth StackTransform._call 2 0 0
meth StackTransform._inverse 2 0 0
meth StackTransform.log_abs_det_jacobian 3 0 0
meth StackTransform.domain 1 0 0
meth StackTransform.codomain 1 0 0
attr StackTransform.dim 1 0 0
meth SoftplusTransform.eq 2 0 0
meth SoftplusTransform._call 2 0 0
meth SoftplusTransform._inverse 2 0 0
meth SoftplusTransform.log_abs_det_jacobian 3 0 0
meth StickBreakingTransform.eq 2 0 0
meth StickBreakingTransform._call 2 0 0
meth StickBreakingTransform._inverse 2 0 0
meth StickBreakingTransform.log_abs_det_jacobian 3 0 0
meth StickBreakingTransform.forward_shape 2 0 0
meth StickBreakingTransform.inverse_shape 2 0 0
meth ReshapeTransform.domain 1 0 0
meth ReshapeTransform.codomain 1 0 0
meth ReshapeTransform.with_cache 2 0 0
meth ReshapeTransform._call 2 0 0
meth ReshapeTransform._inverse 2 0 0
meth ReshapeTransform.log_abs_det_jacobian 3 0 0
meth ReshapeTransform.forward_shape 2 0 0
meth ReshapeTransform.inverse_shape 2 0 0
attr ReshapeTransform.in_shape 1 0 0
attr ReshapeTransform.out_shape 1 0 0
meth LowerCholeskyTransform.eq 2 0 0
meth LowerCholeskyTransform._call 2 0 0
meth LowerCholeskyTransform._inverse 2 0 0
attr LowerCholeskyTransform.domain 1 0 0
attr identity_transform 1 0 0
meth SoftmaxTransform.eq 2 0 0
meth SoftmaxTransform._call 2 0 0
meth SoftmaxTransform._inverse 2 0 0
meth SoftmaxTransform.forward_shape 2 0 0
meth SoftmaxTransform.inverse_shape 2 0 0
meth PowerTransform.with_cache 2 0 0
meth PowerTransform.eq 2 0 0
meth PowerTransform._call 2 0 0
meth PowerTransform._inverse 2 0 0
meth PowerTransform.log_abs_det_jacobian 3 0 0
meth PowerTransform.forward_shape 2 0 0
meth PowerTransform.inverse_shape 2 0 0
meth CumulativeDistributionTransform._call 2 0 0
meth CumulativeDistributionTransform._inverse 2 0 0
meth CumulativeDistributionTransform.log_abs_det_jacobian 3 0 0
meth CumulativeDistributionTransform.with_cache 2 0 0
attr CumulativeDistributionTransform.distribution 1 0 0
meth IndependentTransform.with_cache 2 0 0
meth IndependentTransform.domain 1 0 0
meth IndependentTransform.codomain 1 0 0
meth IndependentTransform._call 2 0 0
meth IndependentTransform._inverse 2 0 0
meth IndependentTransform.log_abs_det_jacobian 3 0 0
meth IndependentTransform.repr 1 0 0
meth IndependentTransform.forward_shape 2 0 0
meth IndependentTransform.inverse_shape 2 0 0
attr IndependentTransform.base_transform 1 0 0
attr IndependentTransform.reinterpreted_batch_ndims 1 0 0
meth ExpTransform.eq 2 0 0
meth ExpTransform._call 2 0 0
meth ExpTransform._inverse 2 0 0
meth ExpTransform.log_abs_det_jacobian 3 0 0
meth ComposeTransform.eq 2 0 0
meth ComposeTransform.domain 1 0 0
meth ComposeTransform.codomain 1 0 0
meth ComposeTransform.with_cache 2 0 0
meth ComposeTransform.call 2 0 0
meth ComposeTransform.log_abs_det_jacobian 3 0 0
meth ComposeTransform.forward_shape 2 0 0
meth ComposeTransform.inverse_shape 2 0 0
meth ComposeTransform.repr 1 0 0
attr ComposeTransform.parts 1 0 0
meth CatTransform.with_cache 2 0 0
meth CatTransform._call 2 0 0
meth CatTransform._inverse 2 0 0
meth CatTransform.log_abs_det_jacobian 3 0 0
meth CatTransform.domain 1 0 0
meth CatTransform.codomain 1 0 0
attr CatTransform.lengths 1 0 0
attr CatTransform.dim 1 0 0
meth CorrCholeskyTransform._call 2 0 0
meth CorrCholeskyTransform._inverse 2 0 0
meth CorrCholeskyTransform.log_abs_det_jacobian 4 0 0
meth CorrCholeskyTransform.forward_shape 2 0 0
meth CorrCholeskyTransform.inverse_shape 2 0 0
meth Transform.getstate 1 0 0
meth Transform.with_cache 2 0 0
meth Transform.eq 2 0 0
meth Transform.ne 2 0 0
meth Transform.call 2 0 0
meth Transform._inv_call 2 0 0
meth Transform._call 2 0 0
meth Transform._inverse 2 0 0
meth Transform.log_abs_det_jacobian 3 0 0
meth Transform.repr 1 0 0
meth Transform.forward_shape 2 0 0
meth Transform.inverse_shape 2 0 0
meth PositiveDefiniteTransform.eq 2 0 0
meth PositiveDefiniteTransform._call 2 0 0
meth PositiveDefiniteTransform._inverse 2 0 0
attr PositiveDefiniteTransform.domain 1 0 0
meth AffineTransform.domain 1 0 0
meth AffineTransform.codomain 1 0 0
meth AffineTransform.with_cache 2 0 0
meth AffineTransform.eq 2 0 0
meth AffineTransform._call 2 0 0
meth AffineTransform._inverse 2 0 0
meth AffineTransform.log_abs_det_jacobian 3 0 0
meth AffineTransform.forward_shape 2 0 0
meth AffineTransform.inverse_shape 2 0 0
attr AffineTransform.loc 1 0 0
attr AffineTransform.scale 1 0 0

torch.distributions.uniform (12 missing, 0 any)

Symbol Typable Typed Any
meth Uniform.expand 3 0 0
meth Uniform.support 1 0 0
meth Uniform.log_prob 2 0 0
meth Uniform.cdf 2 0 0
meth Uniform.icdf 2 0 0
meth Uniform.entropy 1 0 0
prop Uniform.arg_constraints 1 0 0

torch.distributions.utils (0 missing, 1 any)

Symbol Typable Typed Any
meth lazy_property.get 3 3 1

torch.distributions.von_mises (7 missing, 0 any)

Symbol Typable Typed Any
meth VonMises.log_prob 2 0 0
meth VonMises.sample 2 0 0
meth VonMises.expand 3 0 0

torch.distributions.weibull (5 missing, 0 any)

Symbol Typable Typed Any
meth Weibull.expand 3 0 0
meth Weibull.entropy 1 0 0
attr Weibull.concentration_reciprocal 1 0 0

torch.distributions.wishart (17 missing, 0 any)

Symbol Typable Typed Any
meth Wishart.expand 3 0 0
meth Wishart._bartlett_sampling 2 0 0
meth Wishart.rsample 3 2 0
meth Wishart.log_prob 2 0 0
meth Wishart.entropy 1 0 0
meth Wishart._log_normalizer 3 0 0
prop Wishart.arg_constraints 1 0 0
attr Wishart.df 1 0 0
attr Wishart.scale_tril 1 0 0
attr Wishart.covariance_matrix 1 0 0
attr Wishart.precision_matrix 1 0 0

torch.export.custom_ops (1 missing, 0 any)

Symbol Typable Typed Any
attr lib 1 0 0

torch.export.decomp_utils (19 missing, 0 any)

Symbol Typable Typed Any
meth CustomDecompTable.init 1 0 0
meth CustomDecompTable.getitem 2 0 0
meth CustomDecompTable.setitem 3 1 0
meth CustomDecompTable.keys 1 0 0
meth CustomDecompTable.delitem 2 1 0
meth CustomDecompTable.update 2 0 0
meth CustomDecompTable.missing 2 1 0
meth CustomDecompTable.contains 2 1 0
meth CustomDecompTable.iter 1 0 0
meth CustomDecompTable.reversed 1 0 0
meth CustomDecompTable.pop 2 0 0
meth CustomDecompTable.items 1 0 0
attr CustomDecompTable.decomp_table 1 0 0
attr CustomDecompTable.deleted_custom_ops 1 0 0
attr CustomDecompTable.has_materialized 1 0 0

torch.export.dynamic_shapes (36 missing, 0 any)

Symbol Typable Typed Any
meth ShapesCollection.init 1 0 0
meth ShapesCollection.setitem 3 0 0
meth ShapesCollection.getitem 2 0 0
meth ShapesCollection.len 1 0 0
meth ShapesCollection.dynamic_shapes 4 0 0
meth AdditionalInputs.init 1 0 0
meth AdditionalInputs.add 3 0 0
meth AdditionalInputs.dynamic_shapes 4 0 0
meth AdditionalInputs.verify 2 0 0
meth Dim.init 4 3 0
meth Dim.add 2 1 0
meth Dim.radd 2 1 0
meth Dim.sub 2 1 0
meth Dim.rsub 2 1 0
meth Dim.mul 2 1 0
meth Dim.rmul 2 1 0
meth Dim._derived_name 2 1 0
meth Dim._derive 2 1 0
meth Dim.repr 1 0 0
attr Dim.AUTO 1 0 0
attr Dim.DYNAMIC 1 0 0
attr Dim.STATIC 1 0 0
attr Dim.min 1 0 0
attr Dim.max 1 0 0

torch.export.exported_program (42 missing, 7 any)

Symbol Typable Typed Any
meth ExportedProgram.init 10 9 0
meth ExportedProgram._get_flat_args_with_check 3 0 0
meth ExportedProgram._graph_module_flat_inputs 3 3 3
meth ExportedProgram.call 3 3 3
meth ExportedProgram.module 2 1 0
meth ExportedProgram._num_lifted_params_buffers 1 0 0
meth ExportedProgram._check_input_constraints 2 0 0
meth ExportedProgram.validate 1 0 0
meth ExportedProgram._validate 1 0 0
meth ExportedProgram._update 6 1 0
prop ExportedProgram.graph_module 2 0 0
prop ExportedProgram.graph 2 0 0
prop ExportedProgram.graph_signature 2 0 0
prop ExportedProgram.state_dict 2 0 0
prop ExportedProgram.range_constraints 2 0 0
prop ExportedProgram.module_call_graph 2 0 0
prop ExportedProgram.example_inputs 2 0 0
prop ExportedProgram.call_spec 2 0 0
prop ExportedProgram.verifier 2 1 1
prop ExportedProgram.dialect 2 1 0
prop ExportedProgram.verifiers 2 0 0
prop ExportedProgram.tensor_constants 2 0 0
prop ExportedProgram.constants 2 0 0
meth ModuleCallSignature.replace_all_uses_with 3 0 0

torch.export.graph_signature (8 missing, 0 any)

Symbol Typable Typed Any
meth ExportGraphSignature.replace_all_uses 3 2 0
meth ExportGraphSignature.get_replace_hook 2 0 0
meth ExportGraphSignature.str 1 0 0
meth OutputSpec.post_init 1 0 0
meth OutputSpec.str 1 0 0
meth InputSpec.post_init 1 0 0
meth InputSpec.str 1 0 0

torch.export.pt2_archive._package (4 missing, 2 any)

Symbol Typable Typed Any
meth PT2ArchiveReader.init 2 1 0
meth PT2ArchiveReader.exit 2 2 1
attr PT2ArchiveReader.archive_file 1 0 0
meth PT2ArchiveWriter.init 2 1 0
meth PT2ArchiveWriter.exit 2 2 1
attr PT2ArchiveWriter.archive_file 1 0 0

torch.export.unflatten (52 missing, 0 any)

Symbol Typable Typed Any
meth UnflattenedModule.init 3 2 0
meth UnflattenedModule._print_graph 1 0 0
meth UnflattenedModule._adapt_flat_args 4 0 0
meth UnflattenedModule.process_forward_inputs 3 0 0
meth UnflattenedModule.forward 3 0 0
meth UnflattenedModule.finalize 1 0 0
meth UnflattenedModule._dispatch_modules 3 0 0
meth UnflattenedModule.print_readable 5 0 0
attr UnflattenedModule.graph_signature 1 0 0
attr UnflattenedModule.graph 1 0 0
attr UnflattenedModule.module_call_graph 1 0 0
attr UnflattenedModule.flat_args_adapter 1 0 0
attr UnflattenedModule.meta 1 0 0
attr UnflattenedModule.adapted 1 0 0
attr UnflattenedModule.ivals 1 0 0
attr UnflattenedModule.range_constraints 1 0 0
attr UnflattenedModule.input_placeholders 1 0 0
attr UnflattenedModule.check_input_constraints 1 0 0
meth InterpreterModuleDispatcher.init 3 2 0
meth InterpreterModuleDispatcher.forward 3 0 0
meth InterpreterModuleDispatcher.call_modules 1 0 0
meth InterpreterModuleDispatcher.print_readable 5 0 0
meth InterpreterModule.init 3 2 0
meth InterpreterModule.forward 3 0 0
meth InterpreterModule.finalize 1 0 0
meth InterpreterModule.print_readable 5 0 0
attr InterpreterModule.graph 1 0 0

torch.fft (18 missing, 0 any)

Symbol Typable Typed Any
attr fftn 1 0 0
attr ifft2 1 0 0
attr irfftn 1 0 0
attr rfftn 1 0 0
attr ifftn 1 0 0
attr ihfft 1 0 0
attr rfft2 1 0 0
attr fft 1 0 0
attr irfft2 1 0 0
attr rfft 1 0 0
attr ifftshift 1 0 0
attr fft2 1 0 0
attr irfft 1 0 0
attr hfft 1 0 0
attr fftfreq 1 0 0
attr rfftfreq 1 0 0
attr ifft 1 0 0
attr fftshift 1 0 0

torch.functional (36 missing, 1 any)

Symbol Typable Typed Any
func broadcast_shapes 2 0 0
attr istft 1 0 0
func atleast_2d 2 0 0
func atleast_1d 2 0 0
attr lu 1 0 0
func tensordot 5 2 0
func broadcast_tensors 2 0 0
func cdist 5 0 0
func block_diag 2 0 0
func atleast_3d 2 0 0
attr unique_consecutive 1 0 0
func chain_matmul 3 0 0
func align_tensors 2 0 0
func einsum 2 2 1
func norm 7 0 0
attr unique 1 0 0

torch.futures (1 missing, 0 any)

Symbol Typable Typed Any
meth Future.init 2 1 0

torch.fx._symbolic_trace (16 missing, 3 any)

Symbol Typable Typed Any
func wrap 2 1 0
meth Tracer.create_arg 2 2 1
meth Tracer.call_module 5 5 1
meth Tracer.getattr 4 3 1
meth Tracer.create_args_for_root 4 0 0
meth Tracer.deepcopy 2 0 0
meth Tracer._proxy_placeholder 5 0 0
attr Tracer.param_shapes_constant 1 0 0
attr Tracer.scope 1 0 0
attr Tracer.module_stack 1 0 0

torch.fx.annotate (3 missing, 0 any)

Symbol Typable Typed Any
func annotate 3 0 0

torch.fx.experimental.accelerator_partitioner (12 missing, 0 any)

Symbol Typable Typed Any
attr DAGNode.size_bytes 1 0 0
func check_dependency 2 0 0
meth Partitioner.find_single_partition 3 2 0
meth Partitioner.create_single_node_partition 2 0 0
meth Partitioner.aot_based_partition 3 0 0
func get_device_to_partitions_mapping 3 2 0
func reset_partition_device 2 0 0

torch.fx.experimental.const_fold (9 missing, 0 any)

Symbol Typable Typed Any
meth FoldedGraphModule.init 6 5 0
meth FoldedGraphModule.call 3 0 0
meth FoldedGraphModule.run_folding 1 0 0
attr FoldedGraphModule.const_subgraph_module 1 0 0
attr FoldedGraphModule.has_folding_been_run 1 0 0
attr FoldedGraphModule.fx_const_folded_attrs_name 1 0 0
attr FoldedGraphModule.device_for_folded_attrs 1 0 0

torch.fx.experimental.graph_gradual_typechecker (74 missing, 0 any)

Symbol Typable Typed Any
func get_greatest_upper_bound 3 0 0
func relu_inference_rule 3 1 0
func flatten_refinement_rule 2 1 0
func adaptiveavgpool2d_inference_rule 3 1 0
func flatten_inference_rule 2 1 0
func broadcast_types 3 0 0
func flatten_check 4 0 0
func element_wise_eq 2 1 0
func first_two_eq 2 1 0
func calculate_out_dimension 4 0 0
func conv_rule 3 1 0
func get_attr_inference_rule 3 1 0
func transpose_inference_rule 2 1 0
func linear_refinement_rule 2 1 0
func expand_to_tensor_dim 3 0 0
func maxpool2d_check 3 0 0
meth GraphTypeChecker.init 3 0 0
meth GraphTypeChecker.type_check 1 0 0
meth GraphTypeChecker.type_check_node 2 1 0
attr GraphTypeChecker.env 1 0 0
attr GraphTypeChecker.traced 1 0 0
func all_eq 2 1 0
func linear_check 3 0 0
func conv2d_inference_rule 3 1 0
func maxpool2d_inference_rule 3 1 0
meth Refine.init 2 0 0
meth Refine.refine 1 0 0
meth Refine.symbolic_relations 1 0 0
meth Refine.replace_dyn_with_fresh_var 2 0 0
meth Refine.convert_to_sympy_symbols 2 0 0
meth Refine.refine_node 2 1 0
meth Refine.infer_symbolic_relations 2 1 0
attr Refine.constraints 1 0 0
attr Refine.traced 1 0 0
attr Refine.symbol_iter 1 0 0
func add_inference_rule 2 1 0
func get_parameter 3 1 0
func linear_inference_rule 3 1 0
func adaptiveavgpool2d_check 3 0 0
func bn2d_inference_rule 3 1 0
func reshape_inference_rule 2 1 0
func conv_refinement_rule 2 1 0

torch.fx.experimental.merge_matmul (3 missing, 0 any)

Symbol Typable Typed Any
func are_nodes_independent 2 1 0
func merge_matmul 2 1 0
func may_depend_on 4 3 0

torch.fx.experimental.meta_tracer (66 missing, 0 any)

Symbol Typable Typed Any
func torch_nn_relu_override 3 0 0
func torch_abs_override 3 0 0
func torch_relu_override 2 0 0
func gen_constructor_wrapper 2 0 0
func embedding_override 3 0 0
meth MetaAttribute.init 3 1 0
meth MetaAttribute.call 3 0 0
prop MetaAttribute.node 1 0 0
attr MetaAttribute.root 1 0 0
attr MetaAttribute.attr 1 0 0
attr MetaAttribute.tracer 1 0 0
func torch_where_override 4 0 0
meth MetaTracer.create_proxy 8 0 0
meth MetaTracer.getattr 4 0 0
meth MetaTracer.call_module 5 0 0
meth MetaTracer.proxy 2 0 0
meth MetaTracer.trace 4 1 0
meth MetaProxy.install_tensor_meta 2 0 0
meth MetaProxy.size 2 0 0
meth MetaProxy.dim 1 0 0
meth MetaProxy.getattr 2 0 0
prop MetaProxy.shape 1 0 0
prop MetaProxy.dtype 1 0 0
prop MetaProxy.device 1 0 0
func nn_layernorm_override 3 0 0
func proxys_to_metas 2 0 0
func functional_relu_override 3 0 0

torch.fx.experimental.migrate_gradual_types.constraint (208 missing, 0 any)

Symbol Typable Typed Any
meth BinaryConstraint.init 4 0 0
meth BinaryConstraint.eq 2 0 0
meth BinaryConstraint.repr 1 0 0
attr BinaryConstraint.lhs 1 0 0
attr BinaryConstraint.rhs 1 0 0
attr BinaryConstraint.op 1 0 0
meth IndexSelect.init 6 0 0
meth IndexSelect.repr 1 0 0
meth IndexSelect.eq 2 0 0
attr IndexSelect.input_var 1 0 0
attr IndexSelect.tensor_size 1 0 0
attr IndexSelect.dim_replace 1 0 0
attr IndexSelect.index 1 0 0
attr IndexSelect.output 1 0 0
meth ApplyBroadcasting.init 5 0 0
meth ApplyBroadcasting.eq 2 0 0
meth ApplyBroadcasting.repr 1 0 0
attr ApplyBroadcasting.res1 1 0 0
attr ApplyBroadcasting.res2 1 0 0
attr ApplyBroadcasting.input1 1 0 0
attr ApplyBroadcasting.input2 1 0 0
meth BVar.init 2 0 0
meth BVar.repr 1 0 0
meth BVar.eq 2 0 0
attr BVar.c 1 0 0
meth CalcProduct.init 5 0 0
meth CalcProduct.eq 2 0 0
meth CalcProduct.repr 1 0 0
attr CalcProduct.start 1 0 0
attr CalcProduct.end 1 0 0
attr CalcProduct.dims_to_flatten 1 0 0
attr CalcProduct.flattened 1 0 0
meth Prod.init 2 0 0
meth Prod.eq 2 0 0
meth Prod.repr 1 0 0
attr Prod.products 1 0 0
func is_bool_expr 2 0 0
meth F.eq 2 0 0
meth F.repr 1 0 0
meth DVar.init 2 0 0
meth DVar.repr 1 0 0
meth DVar.eq 2 0 0
attr DVar.c 1 0 0
meth DGreatestUpperBound.init 4 0 0
meth DGreatestUpperBound.repr 1 0 0
meth DGreatestUpperBound.eq 2 0 0
attr DGreatestUpperBound.res 1 0 0
attr DGreatestUpperBound.rhs1 1 0 0
attr DGreatestUpperBound.rhs2 1 0 0
meth CanReshape.init 3 0 0
meth CanReshape.repr 1 0 0
meth CanReshape.eq 2 0 0
attr CanReshape.src 1 0 0
attr CanReshape.target 1 0 0
meth Conj.init 2 0 0
meth Conj.eq 2 0 0
meth Conj.repr 1 0 0
attr Conj.conjucts 1 0 0
meth TGreatestUpperBound.init 4 0 0
meth TGreatestUpperBound.repr 1 0 0
meth TGreatestUpperBound.eq 2 0 0
attr TGreatestUpperBound.res 1 0 0
attr TGreatestUpperBound.rhs1 1 0 0
attr TGreatestUpperBound.rhs2 1 0 0
meth GetItem.init 5 0 0
meth GetItem.repr 1 0 0
meth GetItem.eq 2 0 0
attr GetItem.res 1 0 0
attr GetItem.tensor_size 1 0 0
attr GetItem.index 1 0 0
attr GetItem.input_var 1 0 0
func is_algebraic_expression 2 0 0
meth T.eq 2 0 0
meth T.repr 1 0 0
meth BinConstraintD.init 4 0 0
meth TVar.init 2 0 0
meth TVar.repr 1 0 0
meth TVar.eq 2 0 0
attr TVar.tvar 1 0 0
func is_dim 2 0 0
meth GetItemTensor.init 5 0 0
meth GetItemTensor.repr 1 0 0
meth GetItemTensor.eq 2 0 0
attr GetItemTensor.res 1 0 0
attr GetItemTensor.tensor_size 1 0 0
attr GetItemTensor.index_tuple 1 0 0
attr GetItemTensor.input_var 1 0 0
meth Transpose.init 6 0 0
meth Transpose.repr 1 0 0
meth Transpose.eq 2 0 0
attr Transpose.input_var 1 0 0
attr Transpose.tensor_size 1 0 0
attr Transpose.index1 1 0 0
attr Transpose.index2 1 0 0
attr Transpose.output 1 0 0
meth CalcConv.init 9 0 0
meth CalcConv.repr 1 0 0
meth CalcConv.eq 2 0 0
attr CalcConv.conv_result 1 0 0
attr CalcConv.input_var 1 0 0
attr CalcConv.c_out 1 0 0
attr CalcConv.kernel 1 0 0
attr CalcConv.padding 1 0 0
attr CalcConv.stride 1 0 0
attr CalcConv.dilation 1 0 0
attr CalcConv.matching_constraint 1 0 0
meth CalcMaxPool.init 8 0 0
meth CalcMaxPool.repr 1 0 0
meth CalcMaxPool.eq 2 0 0
attr CalcMaxPool.maxpool_result 1 0 0
attr CalcMaxPool.input_var 1 0 0
attr CalcMaxPool.kernel 1 0 0
attr CalcMaxPool.padding 1 0 0
attr CalcMaxPool.stride 1 0 0
attr CalcMaxPool.dilation 1 0 0
attr CalcMaxPool.matching_constraint 1 0 0
meth BinConstraintT.init 4 0 0
meth Disj.init 2 0 0
meth Disj.eq 2 0 0
meth Disj.repr 1 0 0
attr Disj.disjuncts 1 0 0

torch.fx.experimental.migrate_gradual_types.constraint_generator (197 missing, 0 any)

Symbol Typable Typed Any
func relu_inference_rule 6 1 0
func gen_embedding_rules 5 1 0
func layer_norm_functional 5 1 0
func flatten_inference_rule 5 1 0
func view_inference_rule 5 1 0
func torch_dim_inference_rule 5 1 0
func maxpool_inference_rule 6 1 0
func linear_constraints 6 1 0
func cumsum_inference_rule 5 1 0
meth ConstraintGenerator.init 3 0 0
meth ConstraintGenerator.generate_constraints 2 0 0
meth ConstraintGenerator.generate_constraints_node 3 1 0
attr ConstraintGenerator.traced 1 0 0
attr ConstraintGenerator.traced_params 1 0 0
attr ConstraintGenerator.constraints 1 0 0
attr ConstraintGenerator.symbol_dict 1 0 0
attr ConstraintGenerator.graph 1 0 0
func batchnorm_inference_rule 6 1 0
func get_attr_inference_rule 5 1 0
func layer_norm_inference_rule 6 1 0
func masked_fill_inference_rule 5 1 0
func index_select_inference_rule 5 1 0
func transpose_inference_rule 5 1 0
func gen_layer_norm_constraints 5 1 0
func neq_inference_rule 5 1 0
func gt_inference_rule 5 1 0
func embedding_inference_rule 6 1 0
func getitem_inference_rule 5 1 0
func assert_inference_rule 5 1 0
func bmm_inference_rule 5 1 0
func lt_inference_rule 5 1 0
func torch_linear_inference_rule 5 1 0
func add_linear_constraints 5 0 0
func embedding_inference_rule_functional 5 1 0
func range_check 3 0 0
func conv2d_inference_rule 6 1 0
func gen_broadcasting_constraints 6 0 0
func broadcasting_inference_rule 5 1 0
func adaptive_inference_rule 6 1 0
func linear_inference_rule 6 1 0
func add_layer_norm_constraints 3 0 0
func eq_inference_rule 5 1 0
func reshape_inference_rule 5 1 0
func equality_inference_rule 5 1 0
func full_inference_rule 5 1 0
func tensor_inference_rule 5 1 0
func expand_inference_rule 5 1 0
func size_inference_rule 5 1 0
func generate_flatten_constraints 7 0 0
func arange_inference_rule 5 1 0
func type_inference_rule 5 1 0

torch.fx.experimental.migrate_gradual_types.constraint_transformation (77 missing, 0 any)

Symbol Typable Typed Any
func gen_greatest_upper_bound 3 2 0
func generate_calc_maxpool 3 0 0
func generate_broadcasting 3 0 0
func generate_binconstraint_d 3 0 0
func transform_transpose 3 0 0
func valid_index 3 0 0
func register_transformation_rule 2 0 0
func apply_padding 9 8 0
func generate_d_gub 3 0 0
func calc_last_two_dims 3 1 0
func broadcast_dim 7 0 0
func transform_get_item_tensor 3 0 0
func is_dim_div_by_target 3 2 0
func gen_all_reshape_possibilities 3 0 0
func generate_calc_conv 3 0 0
func transform_constraint 3 2 0
func generate_calc_product 3 0 0
func transform_index_select 3 0 0
func transform_get_item 3 0 0
func generate_disj 3 0 0
func create_equality_constraints_for_broadcasting 9 8 0
func generate_conj 3 0 0
func generate_all_broadcasting_possibilities_no_padding 5 4 0
func gen_broadcasting_constraints 7 6 0
func gen_lists_of_dims 4 3 0
func generate_binconstraint_t 3 0 0
func is_target_div_by_dim 3 2 0
func gen_consistency_constraints 3 2 0
func no_broadcast_dim_with_index 6 5 0
func valid_index_tensor 3 0 0
func generate_all_int_dyn_dim_possibilities 2 1 0
func generate_gub 3 0 0
func generate_reshape 3 0 0

torch.fx.experimental.migrate_gradual_types.transform_to_z3 (33 missing, 0 any)

Symbol Typable Typed Any
func evaluate_conditional_with_constraints 6 0 0
func transform_dimension 4 0 0
func transform_var 4 0 0
func transform_to_z3 4 0 0
func transform_all_constraints_trace_time 5 0 0
func transform_all_constraints 3 0 0
func iterate_till_fixed_point 3 0 0
func transform_algebraic_expression 4 0 0

torch.fx.experimental.migrate_gradual_types.z3_types (5 missing, 0 any)

Symbol Typable Typed Any
attr z3_dyn 1 0 0
attr tensor_type 1 0 0
attr dyn_type 1 0 0
attr dyn 1 0 0
attr dim 1 0 0

torch.fx.experimental.normalize (5 missing, 1 any)

Symbol Typable Typed Any
meth NormalizeOperators.call_function 4 3 0
meth NormalizeArgs.init 3 2 0
meth NormalizeArgs.run_node 2 2 1
meth NormalizeArgs.call_function 6 5 0
meth NormalizeArgs.call_module 4 3 0
attr NormalizeArgs.normalize_to_only_use_kwargs 1 0 0

torch.fx.experimental.optimization (17 missing, 0 any)

Symbol Typable Typed Any
func reset_modules 4 3 0
meth UnionFind.init 2 0 0
meth UnionFind.make_set 2 1 0
meth UnionFind.join 3 2 0
func gen_mkl_autotuner 4 0 0
func fuse 4 2 0
meth MklSubgraph.init 2 1 0
attr MklSubgraph.fx_graph 1 0 0
func modules_to_mkldnn 3 2 0
func matches_module_pattern 4 3 0
func extract_subgraph 5 4 0
func replace_node_module 4 3 0

torch.fx.experimental.partitioner_utils (9 missing, 0 any)

Symbol Typable Typed Any
func get_latency_of_partitioned_graph 4 3 0
meth Partition.str 1 0 0
meth Partition.recalculate_mem_size 1 0 0
meth Partition.add_node 2 0 0
meth Partition.remove_node 2 0 0
attr Partition.partition_id 1 0 0
func get_comm_latency_between 4 3 0

torch.fx.experimental.proxy_tensor (8 missing, 1 any)

Symbol Typable Typed Any
func selective_decompose 6 3 0
meth PythonKeyTracer.call_module 5 5 1
attr PythonKeyTracer.torch_fn_metadata 1 0 0
attr DecompositionInterpreter.new_graph 1 0 0
attr DecompositionInterpreter.tracer 1 0 0
attr DecompositionInterpreter.decomposition_table 1 0 0
attr DecompositionInterpreter.mode 1 0 0

torch.fx.experimental.recording (9 missing, 1 any)

Symbol Typable Typed Any
func shape_env_check_state_equal 5 0 0
func replay_shape_env_events 2 0 0
meth ShapeEnvEvent.run 2 1 1
meth FakeTensorMeta.from_fake 2 1 0

torch.fx.experimental.refinement_types (3 missing, 0 any)

Symbol Typable Typed Any
meth Equality.init 3 2 0
attr Equality.lhs 1 0 0
attr Equality.rhs 1 0 0

torch.fx.experimental.rewriter (5 missing, 0 any)

Symbol Typable Typed Any
meth AST_Rewriter.rewrite 2 1 0
meth AST_Rewriter.visit_Assert 2 0 0
meth AST_Rewriter.visit_AnnAssign 2 0 0

torch.fx.experimental.schema_type_annotation (7 missing, 0 any)

Symbol Typable Typed Any
meth AnnotateTypesWithSchema.init 5 4 0
meth AnnotateTypesWithSchema.call_function 4 3 0
meth AnnotateTypesWithSchema.call_module 4 3 0
meth AnnotateTypesWithSchema.get_attr 4 3 0
attr AnnotateTypesWithSchema.annotate_functionals 1 0 0
attr AnnotateTypesWithSchema.annotate_modules 1 0 0
attr AnnotateTypesWithSchema.annotate_get_attrs 1 0 0

torch.fx.experimental.sym_node (127 missing, 0 any)

Symbol Typable Typed Any
meth SymNode.init 8 1 0
meth SymNode.has_hint 1 0 0
meth SymNode.require_hint 2 0 0
meth SymNode.maybe_as_int 1 0 0
meth SymNode.maybe_as_float 1 0 0
meth SymNode.maybe_as_bool 1 0 0
meth SymNode.is_int 1 0 0
meth SymNode.is_float 1 0 0
meth SymNode.is_bool 1 0 0
meth SymNode.is_nested_int 1 0 0
meth SymNode.wrap_int 2 0 0
meth SymNode.wrap_float 2 0 0
meth SymNode.wrap_bool 2 0 0
meth SymNode.clone 1 0 0
meth SymNode.str 1 0 0
meth SymNode.str 1 0 0
meth SymNode.repr 1 0 0
meth SymNode.round 2 1 0
meth SymNode.add 2 1 0
meth SymNode.sub 2 1 0
meth SymNode.mul 2 1 0
meth SymNode.mod 2 1 0
meth SymNode.float_pow 2 1 0
meth SymNode.pow_by_natural 2 1 0
meth SymNode.and_ 2 1 0
meth SymNode.or_ 2 1 0
meth SymNode.float_truediv 2 1 0
meth SymNode.int_truediv 2 1 0
meth SymNode.int_floordiv 2 1 0
meth SymNode.lshift 2 1 0
meth SymNode.rshift 2 1 0
meth SymNode.eq 2 1 0
meth SymNode.ne 2 1 0
meth SymNode.gt 2 1 0
meth SymNode.lt 2 1 0
meth SymNode.le 2 1 0
meth SymNode.ge 2 1 0
meth SymNode.sym_min 2 1 0
meth SymNode.sym_max 2 1 0
meth SymNode.sym_ite 3 1 0
meth SymNode.is_contiguous 3 1 0
meth SymNode.is_channels_last_contiguous_2d 3 1 0
meth SymNode.is_channels_last_contiguous_3d 3 1 0
meth SymNode.is_channels_last_strides_2d 3 1 0
meth SymNode.is_channels_last_strides_3d 3 1 0
meth SymNode.is_non_overlapping_and_dense_indicator 3 1 0
meth SymNode.sym_or 2 0 0
meth SymNode.sym_and 2 0 0
meth SymNode.bitwise_and 2 0 0
meth SymNode.bitwise_or 2 0 0
meth SymNode.bitwise_xor 2 0 0
meth SymNode.truediv 2 0 0
meth SymNode.floordiv 2 1 0
meth SymNode.pow 2 0 0
meth SymNode.is_non_overlapping_and_dense 3 0 0
meth SymNode.int_ 1 0 0
meth SymNode.sym_sum 2 1 0
meth SymNode.evaluate 2 0 0
meth SymNode.guard_int 3 0 0
meth SymNode.guard_float 3 0 0
meth SymNode.guard_bool 3 0 0
meth SymNode.expect_true 3 0 0
meth SymNode.statically_known_true 3 0 0
meth SymNode.guard_size_oblivious 3 0 0
meth SymNode.guard_or_false 3 0 0
meth SymNode.guard_or_true 3 0 0
meth SymNode.bool_ 1 0 0
meth SymNode.is_symbolic 1 0 0
meth SymNode.nested_int 1 0 0
meth SymNode.is_constant 1 0 0
prop SymNode.expr 1 0 0
prop SymNode.hint 1 0 0
attr SymNode.shape_env 1 0 0
attr SymNode.pytype 1 0 0
attr SymNode.fx_node 1 0 0
func method_to_operator 2 0 0
meth DynamicInt.new 2 0 0
meth DynamicInt.repr 1 0 0
meth DynamicInt.floordiv 2 0 0
meth DynamicInt.rfloordiv 2 0 0

torch.fx.experimental.symbolic_shapes (4 missing, 3 any)

Symbol Typable Typed Any
meth ShapeEnv.init 4 4 1
meth ShapeEnv.produce_guards 3 3 2
attr ShapeEnv.should_record_events 1 0 0
attr ShapeEnv.check_recorded_events 1 0 0
attr ShapeEnv.is_recording 1 0 0
attr ShapeEnv.tracked_fakes 1 0 0

torch.fx.experimental.unification.core (7 missing, 0 any)

Symbol Typable Typed Any
func reify 3 0 0
func unify 4 0 0

torch.fx.experimental.unification.dispatch (1 missing, 0 any)

Symbol Typable Typed Any
attr dispatch 1 0 0

torch.fx.experimental.unification.match (30 missing, 0 any)

Symbol Typable Typed Any
func supercedes 3 0 0
meth VarDispatcher.call 3 0 0
func ordering 2 0 0
func match 3 0 0
meth Dispatcher.init 2 0 0
meth Dispatcher.add 3 0 0
meth Dispatcher.call 3 0 0
meth Dispatcher.resolve 2 0 0
meth Dispatcher.register 2 0 0
attr Dispatcher.name 1 0 0
attr Dispatcher.funcs 1 0 0
attr Dispatcher.ordering 1 0 0
func edge 4 0 0

torch.fx.experimental.unification.more (9 missing, 0 any)

Symbol Typable Typed Any
func reify_object 3 0 0
func unifiable 2 0 0
func unify_object 4 0 0

torch.fx.experimental.unification.multipledispatch.conflict (19 missing, 0 any)

Symbol Typable Typed Any
func ambiguous 3 0 0
func ambiguities 2 0 0
func supercedes 3 0 0
func ordering 2 0 0
func consistent 3 0 0
func super_signature 2 0 0
func edge 4 0 0

torch.fx.experimental.unification.multipledispatch.core (2 missing, 1 any)

Symbol Typable Typed Any
func ismethod 2 0 0
func dispatch 3 3 1

torch.fx.experimental.unification.multipledispatch.dispatcher (71 missing, 0 any)

Symbol Typable Typed Any
func variadic_signature_matches 3 0 0
func halt_ordering 1 0 0
meth MethodDispatcher.get_func_params 2 0 0
meth MethodDispatcher.get 3 0 0
meth MethodDispatcher.call 3 0 0
func ambiguity_warn 3 0 0
func warning_text 3 0 0
meth Dispatcher.init 3 0 0
meth Dispatcher.register 3 0 0
meth Dispatcher.get_func_params 2 0 0
meth Dispatcher.get_func_annotations 2 0 0
meth Dispatcher.add 3 0 0
meth Dispatcher.reorder 2 0 0
meth Dispatcher.call 3 0 0
meth Dispatcher.str 1 0 0
meth Dispatcher.repr 1 0 0
meth Dispatcher.dispatch 2 0 0
meth Dispatcher.dispatch_iter 2 0 0
meth Dispatcher.resolve 2 0 0
meth Dispatcher.getstate 1 0 0
meth Dispatcher.setstate 2 0 0
meth Dispatcher._help 2 0 0
meth Dispatcher.help 3 0 0
meth Dispatcher._source 2 0 0
meth Dispatcher.source 3 0 0
prop Dispatcher.ordering 1 0 0
prop Dispatcher.doc 1 0 0
attr Dispatcher.name 1 0 0
attr Dispatcher.funcs 1 0 0
attr Dispatcher.doc 1 0 0
func restart_ordering 2 0 0
func source 2 0 0
func variadic_signature_matches_iter 3 0 0
func str_signature 2 0 0

torch.fx.experimental.unification.multipledispatch.utils (12 missing, 0 any)

Symbol Typable Typed Any
func reverse_dict 2 0 0
func typename 2 0 0
func raises 3 0 0
func groupby 3 0 0
func expand_tuples 2 0 0

torch.fx.experimental.unification.multipledispatch.variadic (9 missing, 0 any)

Symbol Typable Typed Any
func isvariadic 2 0 0
meth VariadicSignatureType.subclasscheck 2 0 0
meth VariadicSignatureType.eq 2 0 0
meth VariadicSignatureType.hash 1 0 0
meth VariadicSignatureMeta.getitem 2 0 0

torch.fx.experimental.unification.unification_tools (56 missing, 0 any)

Symbol Typable Typed Any
func merge 3 0 0
func itemfilter 4 0 0
func get_in 5 0 0
func valmap 4 0 0
func assoc_in 5 0 0
func assoc 5 0 0
func valfilter 4 0 0
func merge_with 4 0 0
func update_in 6 0 0
func itemmap 4 0 0
func dissoc 4 0 0
func keymap 4 0 0
func keyfilter 4 0 0

torch.fx.experimental.unification.utils (14 missing, 0 any)

Symbol Typable Typed Any
func hashable 2 0 0
func reverse_dict 2 0 0
func freeze 2 0 0
func transitive_get 3 0 0
func xfail 2 0 0
func raises 3 0 0

torch.fx.experimental.unification.variable (13 missing, 0 any)

Symbol Typable Typed Any
func variables 2 0 0
func var 1 0 0
func vars 1 0 0
meth Var.new 2 0 0
meth Var.str 1 0 0
meth Var.repr 1 0 0
meth Var.eq 2 0 0
meth Var.hash 1 0 0
func isvar 2 0 0

torch.fx.experimental.unify_refinements (17 missing, 0 any)

Symbol Typable Typed Any
func substitute_solution_one_type 3 0 0
func unify_eq 2 0 0
func infer_symbolic_types_single_pass 2 0 0
func convert_eq 2 0 0
func check_for_type_equality 3 0 0
func infer_symbolic_types 2 0 0
func substitute_all_types 3 0 0

torch.fx.experimental.validator (18 missing, 4 any)

Symbol Typable Typed Any
meth ValidationException.init 5 0 0
meth ValidationException.str 1 0 0
attr ValidationException.msg 1 0 0
attr ValidationException.details 1 0 0
meth BisectValidationException.init 5 0 0
meth BisectValidationException.str 1 0 0
attr BisectValidationException.msg 1 0 0
attr BisectValidationException.details 1 0 0
meth PopulateValidator.init 3 2 0
meth PopulateValidator.placeholder 4 4 1
meth PopulateValidator.call_function 4 4 1
attr PopulateValidator.validator 1 0 0
meth SympyToZ3.constant 3 3 1
meth SympyToZ3.getattr 2 2 1

torch.fx.graph (29 missing, 5 any)

Symbol Typable Typed Any
meth Graph.init 4 3 0
meth Graph.find_nodes 4 3 0
meth Graph.graph_copy 4 3 0
meth Graph.deepcopy 2 1 0
meth Graph.process_inputs 2 0 0
meth Graph.process_outputs 2 0 0
meth Graph.inserting_before 2 1 0
meth Graph.inserting_after 2 1 0
meth Graph.placeholder 4 4 1
meth Graph.output 3 2 0
meth Graph.print_tabular 1 0 0
meth Graph.lint 1 0 0
meth Graph.set_codegen 2 1 0
meth Graph.on_generate_code 2 1 0
prop Graph.owning_module 2 1 0
meth CodeGen.init 1 0 0
meth CodeGen._get_delimiters 2 1 0
meth CodeGen._format_multiline_container 4 1 0
meth CodeGen._get_desc_trailers 3 0 0
meth CodeGen._call_method_with_signature_check 4 0 0
meth CodeGen.process_inputs 2 2 2
meth CodeGen.process_outputs 2 2 2
meth CodeGen._gen_python_code 10 9 0

torch.fx.graph_module (38 missing, 0 any)

Symbol Typable Typed Any
meth GraphModule.new 3 0 0
meth GraphModule.init 4 3 0
meth GraphModule.to_folder 3 2 0
meth GraphModule.reduce_package 2 1 0
meth GraphModule.reduce 1 0 0
meth GraphModule._deepcopy_init 1 0 0
meth GraphModule.deepcopy 2 0 0
meth GraphModule.copy 1 0 0
meth GraphModule.print_readable 7 2 0
meth GraphModule._replicate_for_data_parallel 1 0 0
meth GraphModule._set_replace_hook 2 0 0
meth GraphModule._register_replace_node_hook 2 0 0
meth GraphModule._unregister_replace_node_hook 2 0 0
meth GraphModule._register_create_node_hook 2 0 0
meth GraphModule._unregister_create_node_hook 2 0 0
meth GraphModule._register_erase_node_hook 2 0 0
meth GraphModule._unregister_erase_node_hook 2 0 0
meth GraphModule._register_deepcopy_hook 2 0 0
meth GraphModule._unregister_deepcopy_hook 2 0 0
attr GraphModule.graph 1 0 0
attr GraphModule.shape_env 1 0 0
attr GraphModule.training 1 0 0

torch.fx.interpreter (17 missing, 10 any)

Symbol Typable Typed Any
meth Interpreter.init 4 3 0
meth Interpreter.run 4 3 1
meth Interpreter.boxed_run 2 0 0
meth Interpreter._set_current_node 2 0 0
meth Interpreter.run_node 2 2 1
meth Interpreter.placeholder 4 4 1
meth Interpreter.get_attr 4 4 1
meth Interpreter.call_function 4 4 1
meth Interpreter.call_method 4 4 1
meth Interpreter.call_module 4 4 1
meth Interpreter.output 4 4 1
meth Interpreter.fetch_attr 2 1 0
attr Interpreter.module 1 0 0
attr Interpreter.submodules 1 0 0
attr Interpreter.name 1 0 0
attr Interpreter.garbage_collect_values 1 0 0
attr Interpreter.extra_traceback 1 0 0
attr Interpreter.graph 1 0 0
meth Transformer.init 2 0 0
meth Transformer.call_module 4 4 1
meth Transformer.call_function 4 4 1
attr Transformer.new_graph 1 0 0
attr Transformer.tracer 1 0 0

torch.fx.node (0 missing, 2 any)

Symbol Typable Typed Any
meth Node.setattr 3 3 1
attr Node._sort_key 1 1 1

torch.fx.operator_schemas (5 missing, 2 any)

Symbol Typable Typed Any
func check_for_mutable_operation 4 3 0
func get_signature_for_torch_op 3 2 0
func create_type_hint 2 0 0
func type_matches 3 2 2

torch.fx.passes.backends.cudagraphs (4 missing, 0 any)

Symbol Typable Typed Any
func partition_cudagraphs 3 0 0
meth CudaGraphsSupport.is_node_supported 3 2 0

torch.fx.passes.dialect.common.cse_pass (4 missing, 0 any)

Symbol Typable Typed Any
meth CSEPass.init 2 0 0
attr CSEPass.banned_ops 1 0 0
func get_CSE_banned_ops 1 0 0

torch.fx.passes.fake_tensor_prop (6 missing, 0 any)

Symbol Typable Typed Any
meth FakeTensorProp.init 3 2 0
meth FakeTensorProp.run_node 2 1 0
meth FakeTensorProp.propagate 2 0 0
meth FakeTensorProp.propagate_dont_convert_inputs 2 0 0

torch.fx.passes.graph_drawer (7 missing, 1 any)

Symbol Typable Typed Any
meth FxGraphDrawer.init 9 8 0
meth FxGraphDrawer.get_dot_graph 2 1 0
meth FxGraphDrawer.get_submod_dot_graph 2 1 0
meth FxGraphDrawer._typename 2 2 1
meth FxGraphDrawer._shorten_file_name 3 2 0
meth FxGraphDrawer._tensor_meta_to_label 2 1 0
attr FxGraphDrawer.dot_graph_shape 1 0 0
attr FxGraphDrawer.normalize_args 1 0 0

torch.fx.passes.graph_manipulation (1 missing, 1 any)

Symbol Typable Typed Any
func get_tensor_meta 2 2 1
func replace_target_nodes_with 6 5 0

torch.fx.passes.graph_transform_observer (18 missing, 0 any)

Symbol Typable Typed Any
meth GraphTransformObserver.init 5 4 0
meth GraphTransformObserver.get_current_pass_count 1 0 0
meth GraphTransformObserver._check_disable_pass 1 0 0
meth GraphTransformObserver.enter 1 0 0
meth GraphTransformObserver.exit 4 0 0
meth GraphTransformObserver.get_node_creation_hook 1 0 0
meth GraphTransformObserver.get_node_erase_hook 1 0 0
meth GraphTransformObserver.get_node_replace_hook 1 0 0
meth GraphTransformObserver.get_deepcopy_hook 1 0 0
attr GraphTransformObserver.gm 1 0 0
attr GraphTransformObserver.passname 1 0 0
attr GraphTransformObserver.subsystem 1 0 0
attr GraphTransformObserver.log_url 1 0 0
attr GraphTransformObserver.active 1 0 0
attr GraphTransformObserver.input_dot_graph 1 0 0

torch.fx.passes.infra.partitioner (13 missing, 0 any)

Symbol Typable Typed Any
meth Partition.init 4 3 0
meth Partition.add_node 3 2 0
meth Partition.remove_node 2 1 0
meth Partition.size 1 0 0
attr Partition.id 1 0 0
attr logger 1 0 0
meth CapabilityBasedPartitioner.remove_bookend_non_compute_ops 2 1 0
attr CapabilityBasedPartitioner.graph_module 1 0 0
attr CapabilityBasedPartitioner.operator_support 1 0 0
attr CapabilityBasedPartitioner.allows_single_node_partition 1 0 0
attr CapabilityBasedPartitioner.non_compute_ops 1 0 0
attr CapabilityBasedPartitioner.allowed_single_node_partition_ops 1 0 0
attr CapabilityBasedPartitioner.dependency_viewer 1 0 0

torch.fx.passes.infra.pass_base (3 missing, 0 any)

Symbol Typable Typed Any
meth PassResult.new 3 0 0

torch.fx.passes.infra.pass_manager (10 missing, 0 any)

Symbol Typable Typed Any
meth PassManager.init 6 2 0
meth PassManager.add_pass 2 1 0
meth PassManager.add_constraint 2 1 0
meth PassManager.validate_constraints 1 0 0
meth PassManager.solve_constraints 1 0 0
attr PassManager.run_checks_after_each_pass 1 0 0
attr PassManager.suppress_check_failures 1 0 0

torch.fx.passes.operator_support (1 missing, 0 any)

Symbol Typable Typed Any
meth OperatorSupport.init 2 1 0

torch.fx.passes.pass_manager (19 missing, 0 any)

Symbol Typable Typed Any
func loop_pass 4 3 0
func these_before_those_pass_constraint 3 2 0
meth PassManager.init 3 0 0
meth PassManager.build_from_passlist 2 0 0
meth PassManager.add_pass 2 1 0
meth PassManager.add_constraint 2 0 0
meth PassManager.remove_pass 2 1 0
meth PassManager.replace_pass 3 0 0
meth PassManager.validate 1 0 0
meth PassManager.call 2 0 0
func this_before_that_pass_constraint 3 2 0
func log_hook 3 2 0

torch.fx.passes.regional_inductor (3 missing, 0 any)

Symbol Typable Typed Any
func regional_inductor 3 0 0

torch.fx.passes.reinplace (3 missing, 0 any)

Symbol Typable Typed Any
func reinplace 3 0 0

torch.fx.passes.runtime_assert (0 missing, 1 any)

Symbol Typable Typed Any
func insert_deferred_runtime_asserts 5 5 1

torch.fx.passes.shape_prop (8 missing, 1 any)

Symbol Typable Typed Any
meth ShapeProp.init 3 0 0
meth ShapeProp.run_node 2 2 1
meth ShapeProp.propagate 2 0 0
attr ShapeProp.real_module 1 0 0
attr ShapeProp.fake_module 1 0 0
attr ShapeProp.fake_mode 1 0 0

torch.fx.passes.split_module (3 missing, 0 any)

Symbol Typable Typed Any
meth Partition.init 2 1 0
attr Partition.submod_name 1 0 0
func split_module 9 8 0

torch.fx.passes.split_utils (7 missing, 0 any)

Symbol Typable Typed Any
func getattr_recursive 3 0 0
func setattr_recursive 4 0 0

torch.fx.passes.splitter_base (31 missing, 0 any)

Symbol Typable Typed Any
meth FxNetAccNodesFinder.init 4 3 0
meth FxNetAccNodesFinder.reduce_acc_nodes_non_tensor_input_helper 2 1 0
meth FxNetAccNodesFinder.reduce_acc_nodes_non_tensor_input 1 0 0
meth FxNetAccNodesFinder.reduce_acc_nodes_non_tensor_output 1 0 0
attr FxNetAccNodesFinder.module 1 0 0
attr FxNetAccNodesFinder.operator_support 1 0 0
attr FxNetAccNodesFinder.allow_non_tensor 1 0 0
attr FxNetAccNodesFinder.tracker 1 0 0
meth NodeEvent.init 4 3 0
meth NodeEvent.to_str 1 0 0
attr NodeEvent.source 1 0 0
attr NodeEvent.desc 1 0 0
attr NodeEvent.dep 1 0 0
meth NodeEventTracker.init 3 0 0
meth NodeEventTracker.add 4 3 0
meth NodeEventTracker.print_node 5 0 0
meth NodeEventTracker.to_dict 1 0 0
meth NodeEventTracker.print_all 2 0 0
meth NodeEventTracker.dump 1 0 0
attr NodeEventTracker.tracker_mode 1 0 0
attr NodeEventTracker.dump_prefix 1 0 0
attr NodeEventTracker.events 1 0 0
attr NodeEventTracker.node_events 1 0 0
attr NodeEventTracker.writer 1 0 0

torch.fx.passes.tools_common (7 missing, 0 any)

Symbol Typable Typed Any
meth FxNetAccFusionsFinder.init 3 2 0
meth FxNetAccFusionsFinder.recursive_add_node 4 3 0
attr FxNetAccFusionsFinder.module 1 0 0
attr FxNetAccFusionsFinder.nodes 1 0 0
attr FxNetAccFusionsFinder.acc_nodes 1 0 0
func get_acc_ops_name 2 0 0

torch.fx.passes.utils.common (2 missing, 0 any)

Symbol Typable Typed Any
meth HolderModule.init 2 0 0

torch.fx.passes.utils.matcher_utils (7 missing, 2 any)

Symbol Typable Typed Any
meth InternalMatch.copy 1 0 0
meth SubgraphMatcher._match_literals 4 4 2
attr SubgraphMatcher.pattern 1 0 0
attr SubgraphMatcher.match_output 1 0 0
attr SubgraphMatcher.match_placeholder 1 0 0
attr SubgraphMatcher.remove_overlapping_matches 1 0 0
attr SubgraphMatcher.ignore_literals 1 0 0
attr SubgraphMatcher.pattern_placeholder_nodes 1 0 0

torch.fx.passes.utils.matcher_with_name_node_map_utils (1 missing, 0 any)

Symbol Typable Typed Any
attr SubgraphMatcherWithNameNodeMap.name_node_map 1 0 0

torch.fx.proxy (54 missing, 3 any)

Symbol Typable Typed Any
meth Proxy.init 3 2 0
meth Proxy.getattr 2 1 0
meth Proxy.deepcopy 2 1 0
meth Proxy.setstate 2 0 0
meth Proxy.call 3 1 0
meth Proxy.abs 1 0 0
meth Proxy.keys 1 0 0
meth Proxy.len 1 0 0
meth Proxy.torch_function 5 0 0
attr Proxy.tracer 1 0 0
attr Proxy.node 1 0 0
meth ParameterProxy.init 5 2 0
meth ParameterProxy.size 1 0 0
meth ParameterProxy.dim 1 0 0
meth ParameterProxy.numel 1 0 0
meth ParameterProxy.nelement 1 0 0
prop ParameterProxy.shape 1 0 0
prop ParameterProxy.ndim 1 0 0
attr ParameterProxy.param 1 0 0
attr ParameterProxy.name 1 0 0
meth Attribute.init 3 2 0
meth Attribute.call 3 0 0
prop Attribute.node 1 0 0
attr Attribute.root 1 0 0
attr Attribute.attr 1 0 0
attr Attribute.tracer 1 0 0
meth Scope.init 3 2 1
attr Scope.module_path 1 0 0
attr Scope.module_type 1 0 0
meth TracerBase.create_proxy 8 7 0
meth TracerBase._find_user_frame 1 0 0
meth TracerBase.create_arg 2 2 1
meth TracerBase.keys 2 2 1
meth MetaProxy.init 4 2 0
meth MetaProxy.torch_function 5 0 0
attr MetaProxy.fake_mode 1 0 0
meth ScopeContextManager.init 3 2 0
meth ScopeContextManager.enter 1 0 0
meth ScopeContextManager.exit 2 0 0
meth GraphAppendingTracer.init 2 1 0

torch.fx.tensor_type (14 missing, 0 any)

Symbol Typable Typed Any
meth TensorType.init 2 0 0
meth TensorType.repr 1 0 0
meth TensorType.eq 2 0 0
meth TensorType.class_getitem 2 0 0
func is_consistent 3 0 0
attr Dyn 1 0 0
func is_more_precise 3 0 0

torch.fx.traceback (21 missing, 0 any)

Symbol Typable Typed Any
func reset_grad_fn_seq_nr 1 0 0
func set_current_meta 3 0 0
func annotate 2 1 0
func preserve_node_meta 2 0 0
func get_current_replay_node 1 0 0
meth NodeSource.init 4 3 0
meth NodeSource.repr 1 0 0
meth NodeSource._get_action_string 1 0 0
meth NodeSource.print_readable 2 0 0
meth NodeSource.eq 2 1 0
meth NodeSource.hash 1 0 0
func set_current_replay_node 2 0 0
func set_grad_fn_seq_nr 2 0 0
func set_stack_trace 2 1 0
func annotate_fn 2 1 0

torch.hub (22 missing, 0 any)

Symbol Typable Typed Any
func load 10 0 0
func list 6 0 0
func help 6 0 0

torch.jit (12 missing, 3 any)

Symbol Typable Typed Any
func export_opnames 2 0 0
func annotate 3 0 0
meth strict_fusion.enter 1 0 0
meth strict_fusion.exit 4 4 3
func onednn_fusion_enabled 1 0 0
func isinstance 3 0 0
func script_if_tracing 2 0 0

torch.jit._async (6 missing, 0 any)

Symbol Typable Typed Any
func wait 2 0 0
func fork 4 0 0

torch.jit._freeze (2 missing, 0 any)

Symbol Typable Typed Any
func freeze 4 2 0

torch.jit._fuser (1 missing, 0 any)

Symbol Typable Typed Any
func set_fusion_strategy 2 1 0

torch.jit._script (12 missing, 2 any)

Symbol Typable Typed Any
func script 6 4 0
meth ScriptModule.getattr 2 0 0
meth ScriptModule.setattr 3 1 0
meth ScriptModule.define 2 0 0
meth ScriptModule._replicate_for_data_parallel 1 0 0
meth ScriptModule.reduce_package 2 1 0
attr ScriptModule.jit_unused_properties 1 1 1
attr CompilationUnit 1 1 1
func interface 2 0 0

torch.jit._serialization (8 missing, 0 any)

Symbol Typable Typed Any
func load 5 0 0
func save 4 1 0

torch.jit._trace (26 missing, 0 any)

Symbol Typable Typed Any
func trace 13 0 0
func trace_module 13 0 0

torch.jit.annotations (37 missing, 0 any)

Symbol Typable Typed Any
func get_param_names 3 0 0
func check_fn 3 1 0
func get_type_line 2 0 0
func split_type_line 2 0 0
func ann_to_type 4 0 0
func get_signature 5 0 0
meth Module.init 3 1 0
meth Module.getattr 2 0 0
attr Module.name 1 0 0
attr Module.members 1 0 0
func try_real_annotations 3 0 0
func try_ann_to_type 4 0 0
attr BroadcastingList3 1 0 0
func parse_type_line 4 0 0
attr BroadcastingList2 1 0 0

torch.jit.frontend (191 missing, 0 any)

Symbol Typable Typed Any
func build_stmts 3 0 0
func build_param_list 5 0 0
attr build_expr 1 0 0
attr build_stmt 1 0 0
func find_before 5 0 0
func build_withitems 3 0 0
func build_param 6 0 0
meth FrontendError.init 3 1 0
attr FrontendError.source_range 1 0 0
attr FrontendError.msg 1 0 0
attr FrontendError.error_report 1 0 0
meth Builder.call 3 0 0
func get_class_properties 3 0 0
func get_jit_def 5 0 0
func build_def 7 0 0
meth UnsupportedNodeError.init 4 1 0
attr build_withitem 1 0 0
func is_reserved_name 2 0 0
func get_class_assigns 3 0 0
meth WithItemBuilder.build_withitem 3 0 0
meth StmtBuilder.build_Expr 3 0 0
meth StmtBuilder.build_Assign 3 0 0
meth StmtBuilder.build_AnnAssign 3 0 0
meth StmtBuilder.build_Delete 3 0 0
meth StmtBuilder.build_Return 3 0 0
meth StmtBuilder.build_Raise 3 0 0
meth StmtBuilder.build_Assert 3 0 0
meth StmtBuilder.build_AugAssign 3 0 0
meth StmtBuilder.build_While 3 0 0
meth StmtBuilder.build_For 3 0 0
meth StmtBuilder.build_If 3 0 0
meth StmtBuilder.build_Print 3 0 0
meth StmtBuilder.build_Pass 3 0 0
meth StmtBuilder.build_Break 3 0 0
meth StmtBuilder.build_Continue 3 0 0
meth StmtBuilder.build_With 3 0 0
func build_ignore_context_manager 3 0 0
func is_torch_jit_ignore_context_manager 2 1 0
func build_class_def 7 0 0
func get_jit_class_def 3 0 0
func get_default_args_for_class 2 0 0
meth ExprBuilder.build_Attribute 3 0 0
meth ExprBuilder.build_Call 3 0 0
meth ExprBuilder.build_Ellipsis 3 0 0
meth ExprBuilder.build_Name 3 0 0
meth ExprBuilder.build_NameConstant 3 0 0
meth ExprBuilder.build_BinOp 3 0 0
meth ExprBuilder.build_UnaryOp 3 0 0
meth ExprBuilder.build_BoolOp 3 0 0
meth ExprBuilder.build_IfExp 3 0 0
meth ExprBuilder.build_Compare 3 0 0
meth ExprBuilder.build_Subscript 3 0 0
meth ExprBuilder.build_List 3 0 0
meth ExprBuilder.build_Tuple 3 0 0
meth ExprBuilder.build_Dict 3 0 0
meth ExprBuilder.build_Num 3 0 0
meth ExprBuilder.build_Constant 3 0 0
meth ExprBuilder.build_Str 3 0 0
meth ExprBuilder.build_JoinedStr 3 0 0
meth ExprBuilder.build_ListComp 3 0 0
meth ExprBuilder.build_GeneratorExp 3 0 0
meth ExprBuilder.build_DictComp 3 0 0
meth ExprBuilder.build_Starred 3 0 0
func get_default_args 2 0 0

torch.jit.generate_bytecode (2 missing, 0 any)

Symbol Typable Typed Any
func format_bytecode 2 0 0

torch.jit.mobile (10 missing, 0 any)

Symbol Typable Typed Any
meth LiteScriptModule.init 2 1 0
meth LiteScriptModule.call 2 0 0
meth LiteScriptModule.find_method 2 0 0
meth LiteScriptModule.forward 2 0 0
meth LiteScriptModule.run_method 3 0 0

torch.jit.quantized (29 missing, 0 any)

Symbol Typable Typed Any
meth QuantizedGRUCell.init 2 0 0
meth QuantizedGRU.init 3 0 0
func quantize_rnn_modules 3 0 0
meth QuantizedRNNBase.init 3 0 0
meth QuantizedLSTMCell.init 2 0 0
meth QuantizedRNNCell.init 2 0 0
meth QuantizedLinearFP16.init 2 0 0
meth QuantizedLinear.init 2 0 0
meth QuantizedRNNCellBase.init 2 0 0
func quantize_linear_modules 3 0 0
func quantize_rnn_cell_modules 2 0 0
meth QuantizedLSTM.init 3 0 0

torch.jit.unsupported_tensor_ops (3 missing, 0 any)

Symbol Typable Typed Any
func execWrapper 4 1 0

torch.library (52 missing, 1 any)

Symbol Typable Typed Any
func define 5 0 0
func fallthrough_kernel 1 0 0
func impl_abstract 5 0 0
func register_fake 6 5 0
func register_torch_dispatch 5 4 1
func register_vmap 4 2 0
meth Library.init 4 0 0
meth Library.repr 1 0 0
meth Library.define 4 0 0
meth Library._register_fake 5 0 0
meth Library._register_torch_dispatch_rule 4 0 0
meth Library._impl_with_aoti_compile 3 0 0
meth Library.impl 6 0 0
meth Library.fallback 4 0 0
meth Library._register_effectful_op 3 2 0
meth Library._destroy 1 0 0
attr Library.ns 1 0 0
attr Library.kind 1 0 0
attr Library.dispatch_key 1 0 0
func register_autocast 5 4 0

torch.linalg (41 missing, 0 any)

Symbol Typable Typed Any
attr matrix_norm 1 0 0
attr svd 1 0 0
attr matrix_power 1 0 0
attr matrix_exp 1 0 0
attr cholesky 1 0 0
attr eigvals 1 0 0
attr ldl_factor 1 0 0
attr eig 1 0 0
attr solve 1 0 0
attr lu_factor_ex 1 0 0
attr matrix_rank 1 0 0
attr vander 1 0 0
attr inv_ex 1 0 0
attr det 1 0 0
attr eigh 1 0 0
attr solve_ex 1 0 0
attr eigvalsh 1 0 0
attr cholesky_ex 1 0 0
attr svdvals 1 0 0
attr vecdot 1 0 0
attr norm 1 0 0
attr inv 1 0 0
attr ldl_solve 1 0 0
attr tensorinv 1 0 0
attr matmul 1 0 0
attr lu 1 0 0
attr lu_factor 1 0 0
attr pinv 1 0 0
attr diagonal 1 0 0
attr solve_triangular 1 0 0
attr householder_product 1 0 0
attr slogdet 1 0 0
attr lstsq 1 0 0
attr ldl_factor_ex 1 0 0
attr multi_dot 1 0 0
attr lu_solve 1 0 0
attr cond 1 0 0
attr tensorsolve 1 0 0
attr qr 1 0 0
attr cross 1 0 0
attr vector_norm 1 0 0

torch.masked.maskedtensor.core (43 missing, 1 any)

Symbol Typable Typed Any
func is_masked_tensor 2 2 1
meth MaskedTensor.new 5 0 0
meth MaskedTensor._preprocess_data 3 0 0
meth MaskedTensor._validate_members 1 0 0
meth MaskedTensor.init 4 0 0
meth MaskedTensor._from_values 3 0 0
meth MaskedTensor._set_data_mask 3 0 0
meth MaskedTensor.repr 1 0 0
meth MaskedTensor.torch_function 5 0 0
meth MaskedTensor.unary 4 0 0
meth MaskedTensor.torch_dispatch 5 0 0
meth MaskedTensor.lt 2 0 0
meth MaskedTensor.to_tensor 2 0 0
meth MaskedTensor.get_data 1 0 0
meth MaskedTensor.get_mask 1 0 0
meth MaskedTensor.is_sparse_coo 1 0 0
meth MaskedTensor.is_sparse_csr 1 0 0
prop MaskedTensor.is_sparse 1 0 0

torch.mps (2 missing, 0 any)

Symbol Typable Typed Any
func compile_shader 2 1 0
func set_per_process_memory_fraction 2 1 0

torch.mtia (9 missing, 4 any)

Symbol Typable Typed Any
func is_initialized 1 0 0
meth device.init 2 1 1
meth device.enter 1 0 0
meth device.exit 4 3 3
attr device.idx 1 0 0
attr device.prev_idx 1 0 0
func set_stream 2 1 0
func init 1 0 0
func is_bf16_supported 2 1 0

torch.mtia.mtia_graph (4 missing, 0 any)

Symbol Typable Typed Any
meth graph.init 4 3 0
attr graph.capture_stream 1 0 0
attr graph.stream_ctx 1 0 0
attr graph.mtia_graph 1 0 0

torch.multiprocessing (4 missing, 0 any)

Symbol Typable Typed Any
func get_all_sharing_strategies 1 0 0
func set_sharing_strategy 2 0 0
func get_sharing_strategy 1 0 0

torch.multiprocessing.pool (5 missing, 0 any)

Symbol Typable Typed Any
func clean_worker 3 0 0
meth Pool._setup_queues 1 0 0
meth Pool._repopulate_pool 1 0 0

torch.multiprocessing.queue (12 missing, 0 any)

Symbol Typable Typed Any
meth ConnectionWrapper.init 2 0 0
meth ConnectionWrapper.send 2 0 0
meth ConnectionWrapper.recv 1 0 0
meth ConnectionWrapper.getattr 2 0 0
attr ConnectionWrapper.conn 1 0 0
meth SimpleQueue._make_methods 1 0 0
meth Queue.init 3 0 0

torch.multiprocessing.reductions (113 missing, 0 any)

Symbol Typable Typed Any
func rebuild_typed_storage 3 0 0
func rebuild_event 3 0 0
func reduce_nested_tensor 2 0 0
func reduce_event 2 0 0
func rebuild_meta_tensor 8 0 0
func storage_from_cache 3 0 0
func rebuild_storage_filename 6 0 0
attr shared_cache 1 0 0
func rebuild_cuda_tensor 16 0 0
meth SharedCache._after_fork 1 0 0
meth SharedCache.get 2 0 0
meth SharedCache.setitem 3 0 0
meth SharedCache.free_dead_references 1 0 0
attr SharedCache.limit 1 0 0
func reduce_typed_storage 2 0 0
func rebuild_typed_storage_child 3 0 0
func rebuild_storage_fd 4 0 0
meth StorageWeakRef.init 2 0 0
meth StorageWeakRef.from_weakref 2 0 0
meth StorageWeakRef.expired 1 0 0
meth StorageWeakRef.del 1 0 0
meth StorageWeakRef.hash 1 0 0
meth StorageWeakRef.eq 2 0 0
attr StorageWeakRef.cdata 1 0 0
func rebuild_nested_tensor 9 0 0
func reduce_typed_storage_child 2 0 0
func rebuild_storage_empty 2 0 0
func reduce_sparse_tensor 2 0 0
func reduce_tensor 2 0 0
func init_reductions 1 0 0
func rebuild_sparse_coo_tensor 7 0 0
func rebuild_tensor 4 0 0
func reduce_storage 2 0 0
func rebuild_sparse_compressed_tensor 9 0 0
func fd_id 2 0 0

torch.multiprocessing.spawn (35 missing, 0 any)

Symbol Typable Typed Any
meth ProcessContext.init 3 0 0
meth ProcessContext.pids 1 0 0
meth ProcessContext._join_procs_with_timeout 2 1 0
meth ProcessContext.join 3 2 0
attr ProcessContext.error_files 1 0 0
attr ProcessContext.processes 1 0 0
attr ProcessContext.sentinels 1 0 0
func spawn 7 0 0
func start_processes 7 0 0
meth SpawnContext.init 3 0 0
meth ProcessException.init 4 3 0
meth ProcessException.reduce 1 0 0
attr ProcessException.msg 1 0 0
attr ProcessException.error_index 1 0 0
attr ProcessException.error_pid 1 0 0
meth ProcessExitedException.init 6 5 0
meth ProcessExitedException.reduce 1 0 0
attr ProcessExitedException.exit_code 1 0 0
attr ProcessExitedException.signal_name 1 0 0

torch.nested (9 missing, 0 any)

Symbol Typable Typed Any
func as_nested_tensor 5 4 0
func nested_tensor 7 1 0
func narrow 6 5 0
attr to_padded_tensor 1 0 0

torch.nn (2 missing, 0 any)

Symbol Typable Typed Any
func factory_kwargs 2 0 0

torch.nn.attention (1 missing, 0 any)

Symbol Typable Typed Any
func sdpa_kernel 3 2 0

torch.nn.attention.bias (10 missing, 0 any)

Symbol Typable Typed Any
func causal_upper_left 2 1 0
func causal_lower_right 2 1 0
meth CausalBias.torch_function 5 0 0
attr CausalBias.variant 1 0 0
attr CausalBias.seq_len_q 1 0 0
attr CausalBias.seq_len_kv 1 0 0

torch.nn.attention.flex_attention (15 missing, 0 any)

Symbol Typable Typed Any
meth BlockMask.from_kv_blocks 9 8 0
meth BlockMask.as_tuple 2 1 0
meth BlockMask.getitem 2 1 0
meth BlockMask._adjust 3 2 0
meth BlockMask.numel 1 0 0
meth BlockMask.to_string 3 0 0
meth BlockMask._flatten 1 0 0
meth BlockMask._unflatten 3 0 0
meth BlockMask._flatten_with_keys 1 0 0
prop BlockMask.shape 1 0 0
func create_block_mask 9 8 0

torch.nn.cpp (19 missing, 0 any)

Symbol Typable Typed Any
meth ModuleWrapper.init 2 1 0
meth ModuleWrapper._apply 3 0 0
prop ModuleWrapper.training 2 0 0
attr ModuleWrapper.cpp_module 1 0 0
meth OrderedDictWrapper.init 3 1 0
meth OrderedDictWrapper.items 1 0 0
meth OrderedDictWrapper.keys 1 0 0
meth OrderedDictWrapper.values 1 0 0
meth OrderedDictWrapper.iter 1 0 0
meth OrderedDictWrapper.contains 2 1 0
meth OrderedDictWrapper.getitem 2 0 0
prop OrderedDictWrapper.cpp_dict 1 0 0
attr OrderedDictWrapper.cpp_module 1 0 0
attr OrderedDictWrapper.attr 1 0 0

torch.nn.functional (6 missing, 10 any)

Symbol Typable Typed Any
func sigmoid 2 2 1
func tanhshrink 2 1 1
func softsign 2 1 1
func tanh 2 1 1
func upsample_bilinear 4 3 1
func upsample 6 5 1
func assert_int_or_pair 4 4 3
func upsample_nearest 4 3 1

torch.nn.grad (48 missing, 0 any)

Symbol Typable Typed Any
func conv2d_input 8 0 0
func conv3d_weight 8 0 0
func conv2d_weight 8 0 0
func conv3d_input 8 0 0
func conv1d_weight 8 0 0
func conv1d_input 8 0 0

torch.nn.init (11 missing, 0 any)

Symbol Typable Typed Any
attr normal 1 0 0
attr eye 1 0 0
attr dirac 1 0 0
attr xavier_normal 1 0 0
attr sparse 1 0 0
attr uniform 1 0 0
attr kaiming_uniform 1 0 0
attr xavier_uniform 1 0 0
attr kaiming_normal 1 0 0
attr constant 1 0 0
attr orthogonal 1 0 0

torch.nn.modules.activation (37 missing, 0 any)

Symbol Typable Typed Any
meth LogSoftmax.setstate 2 0 0
meth PReLU.init 5 3 0
attr PReLU.init 1 0 0
attr PReLU.weight 1 0 0
meth Softmin.setstate 2 0 0
meth MultiheadAttention.init 12 1 0
meth MultiheadAttention.setstate 2 0 0
attr MultiheadAttention.embed_dim 1 0 0
attr MultiheadAttention.kdim 1 0 0
attr MultiheadAttention.vdim 1 0 0
attr MultiheadAttention.num_heads 1 0 0
attr MultiheadAttention.dropout 1 0 0
attr MultiheadAttention.batch_first 1 0 0
attr MultiheadAttention.head_dim 1 0 0
attr MultiheadAttention.out_proj 1 0 0
attr MultiheadAttention.add_zero_attn 1 0 0
attr MultiheadAttention.q_proj_weight 1 0 0
attr MultiheadAttention.k_proj_weight 1 0 0
attr MultiheadAttention.v_proj_weight 1 0 0
attr MultiheadAttention.in_proj_bias 1 0 0
attr MultiheadAttention.in_proj_weight 1 0 0
meth Softmax.setstate 2 0 0

torch.nn.modules.adaptive (8 missing, 0 any)

Symbol Typable Typed Any
meth AdaptiveLogSoftmaxWithLoss.init 8 6 0
meth AdaptiveLogSoftmaxWithLoss._get_full_log_prob 3 0 0
attr AdaptiveLogSoftmaxWithLoss.shortlist_size 1 0 0
attr AdaptiveLogSoftmaxWithLoss.n_clusters 1 0 0
attr AdaptiveLogSoftmaxWithLoss.head_size 1 0 0

torch.nn.modules.batchnorm (14 missing, 0 any)

Symbol Typable Typed Any
meth LazyBatchNorm2d._check_input_dim 2 1 0
meth BatchNorm3d._check_input_dim 2 1 0
meth SyncBatchNorm.init 9 7 0
meth SyncBatchNorm._check_input_dim 2 1 0
meth SyncBatchNorm._check_non_zero_input_channels 2 1 0
meth SyncBatchNorm.convert_sync_batchnorm 3 0 0
attr SyncBatchNorm.process_group 1 0 0
meth LazyBatchNorm1d._check_input_dim 2 1 0
meth BatchNorm2d._check_input_dim 2 1 0
meth LazyBatchNorm3d._check_input_dim 2 1 0
meth BatchNorm1d._check_input_dim 2 1 0

torch.nn.modules.container (13 missing, 10 any)

Symbol Typable Typed Any
meth ModuleList._get_abs_string_index 2 0 0
meth ParameterDict.init 2 2 1
meth ParameterDict.getitem 2 2 1
meth ParameterDict.setitem 3 3 1
meth ParameterDict.setdefault 3 3 1
meth ParameterDict.pop 2 2 1
meth ParameterDict.get 3 3 1
meth ParameterDict.call 2 0 0
meth ParameterList._get_abs_string_index 2 0 0
meth ParameterList.getitem 2 2 1
meth ParameterList.setitem 3 3 1
meth ParameterList.append 2 2 1
meth ParameterList.call 3 0 0
meth Container.init 2 2 1
meth Sequential.add 2 1 0
meth Sequential.iadd 2 1 0
meth Sequential.forward 2 0 0

torch.nn.modules.conv (33 missing, 0 any)

Symbol Typable Typed Any
meth LazyConv2d.init 11 9 0
meth Conv1d.init 12 10 0
meth Conv1d._conv_forward 4 3 0
attr Conv1d.doc 1 0 0
meth ConvTranspose2d.init 13 11 0
attr ConvTranspose2d.doc 1 0 0
meth LazyConvTranspose1d.init 12 10 0
meth LazyConvTranspose2d.init 12 10 0
meth ConvTranspose3d.init 13 11 0
attr ConvTranspose3d.doc 1 0 0
meth Conv2d.init 12 10 0
meth Conv2d._conv_forward 4 3 0
attr Conv2d.doc 1 0 0
meth LazyConv1d.init 11 9 0
meth LazyConvTranspose3d.init 12 10 0
meth LazyConv3d.init 11 9 0
meth Conv3d.init 12 10 0
meth Conv3d._conv_forward 4 3 0
attr Conv3d.doc 1 0 0
meth ConvTranspose1d.init 13 11 0
attr ConvTranspose1d.doc 1 0 0

torch.nn.modules.flatten (2 missing, 0 any)

Symbol Typable Typed Any
meth Unflatten._require_tuple_tuple 2 1 0
meth Unflatten._require_tuple_int 2 1 0

torch.nn.modules.instancenorm (6 missing, 0 any)

Symbol Typable Typed Any
meth InstanceNorm1d._check_input_dim 2 1 0
meth LazyInstanceNorm3d._check_input_dim 2 1 0
meth LazyInstanceNorm2d._check_input_dim 2 1 0
meth LazyInstanceNorm1d._check_input_dim 2 1 0
meth InstanceNorm3d._check_input_dim 2 1 0
meth InstanceNorm2d._check_input_dim 2 1 0

torch.nn.modules.lazy (24 missing, 0 any)

Symbol Typable Typed Any
meth LazyModuleMixin.init 3 0 0
meth LazyModuleMixin._save_to_state_dict 4 0 0
meth LazyModuleMixin._lazy_load_hook 8 0 0
meth LazyModuleMixin.initialize_parameters 3 0 0
meth LazyModuleMixin.has_uninitialized_params 1 0 0
meth LazyModuleMixin._infer_parameters 4 0 0
meth LazyModuleMixin._replicate_for_data_parallel 1 0 0

torch.nn.modules.linear (9 missing, 2 any)

Symbol Typable Typed Any
meth Linear.init 6 4 0
attr Linear.bias 1 0 0
meth LazyLinear.init 5 3 0
meth LazyLinear.initialize_parameters 2 1 0
meth Identity.init 3 3 2
meth Bilinear.init 7 5 0
attr Bilinear.bias 1 0 0

torch.nn.modules.loss (27 missing, 0 any)

Symbol Typable Typed Any
meth CrossEntropyLoss.init 7 5 0
meth NLLLoss2d.init 6 4 0
attr HuberLoss.delta 1 0 0
meth BCEWithLogitsLoss.init 6 4 0
meth KLDivLoss.init 5 3 0
attr KLDivLoss.log_target 1 0 0
meth TripletMarginLoss.init 8 6 0
meth HingeEmbeddingLoss.init 5 3 0
meth MultiMarginLoss.init 7 5 0
meth PoissonNLLLoss.init 7 5 0
meth SmoothL1Loss.init 5 3 0
attr SmoothL1Loss.beta 1 0 0
meth NLLLoss.init 6 4 0
meth MarginRankingLoss.init 5 3 0
meth CosineEmbeddingLoss.init 5 3 0

torch.nn.modules.module (55 missing, 4 any)

Symbol Typable Typed Any
meth Module.init 3 3 2
meth Module.get_extra_state 1 1 1
meth Module.set_extra_state 2 2 1
meth Module._apply 3 0 0
meth Module._get_backward_hooks 1 0 0
meth Module._get_backward_pre_hooks 1 0 0
meth Module._maybe_warn_non_full_backward_hook 4 1 0
meth Module._slow_forward 3 0 0
meth Module._wrapped_call_impl 3 0 0
meth Module._call_impl 3 0 0
meth Module.getstate 1 0 0
meth Module.setstate 2 0 0
meth Module.delattr 2 1 0
meth Module._register_state_dict_hook 2 0 0
meth Module.register_state_dict_post_hook 2 0 0
meth Module.register_state_dict_pre_hook 2 0 0
meth Module._save_to_state_dict 4 1 0
meth Module._register_load_state_dict_pre_hook 3 0 0
meth Module.register_load_state_dict_pre_hook 2 0 0
meth Module.register_load_state_dict_post_hook 2 0 0
meth Module._load_from_state_dict 8 1 0
meth Module.load_state_dict 4 3 0
meth Module._named_members 5 1 0
meth Module.named_modules 4 3 0
meth Module._get_name 1 0 0
meth Module.dir 1 0 0
meth Module._replicate_for_data_parallel 1 0 0
meth Module.compile 3 1 0

torch.nn.modules.normalization (12 missing, 0 any)

Symbol Typable Typed Any
meth GroupNorm.init 7 5 0
attr GroupNorm.weight 1 0 0
attr GroupNorm.bias 1 0 0
meth LayerNorm.init 7 5 0
attr LayerNorm.weight 1 0 0
attr LayerNorm.bias 1 0 0
meth LocalResponseNorm.extra_repr 1 0 0
meth RMSNorm.init 6 4 0
attr RMSNorm.weight 1 0 0

torch.nn.modules.padding (3 missing, 0 any)

Symbol Typable Typed Any
meth CircularPad2d._check_input_dim 2 1 0
meth CircularPad3d._check_input_dim 2 1 0
meth CircularPad1d._check_input_dim 2 1 0

torch.nn.modules.pooling (14 missing, 0 any)

Symbol Typable Typed Any
meth AdaptiveMaxPool2d.forward 2 1 0
meth AdaptiveMaxPool3d.forward 2 1 0
meth FractionalMaxPool3d.init 6 5 0
meth FractionalMaxPool3d.forward 2 1 0
meth AvgPool3d.setstate 2 0 0
attr AvgPool3d.divisor_override 1 0 0
meth AdaptiveMaxPool1d.forward 2 1 0
meth MaxPool3d.forward 2 1 0
meth MaxPool1d.forward 2 1 0
attr AvgPool2d.divisor_override 1 0 0
meth FractionalMaxPool2d.init 6 5 0
meth FractionalMaxPool2d.forward 2 1 0
meth MaxPool2d.forward 2 1 0

torch.nn.modules.rnn (36 missing, 0 any)

Symbol Typable Typed Any
meth RNN.init 13 9 0
attr RNN.nonlinearity 1 0 0
meth RNNCell.init 7 5 0
meth LSTM.init 13 9 0
meth RNNCellBase.init 7 5 0
attr RNNCellBase.bias_ih 1 0 0
attr RNNCellBase.bias_hh 1 0 0
meth GRU.init 12 8 0
meth GRUCell.init 6 4 0
meth RNNBase.init 12 10 0
meth RNNBase.setattr 3 1 0
meth RNNBase._apply 3 0 0
meth RNNBase._weights_have_changed 1 0 0
meth RNNBase.permute_hidden 3 2 0
meth RNNBase.getstate 1 0 0
meth RNNBase.setstate 2 0 0
meth RNNBase._replicate_for_data_parallel 1 0 0
meth LSTMCell.init 6 4 0

torch.nn.modules.sparse (12 missing, 0 any)

Symbol Typable Typed Any
meth Embedding.init 12 10 0
meth Embedding.from_pretrained 8 0 0
meth EmbeddingBag.init 13 11 0

torch.nn.modules.transformer (48 missing, 0 any)

Symbol Typable Typed Any
meth Transformer.init 16 14 0
attr Transformer.d_model 1 0 0
attr Transformer.nhead 1 0 0
attr Transformer.batch_first 1 0 0
attr Transformer.encoder 1 0 0
attr Transformer.decoder 1 0 0
meth TransformerEncoderLayer.init 12 10 0
meth TransformerEncoderLayer.setstate 2 0 0
attr TransformerEncoderLayer.self_attn 1 0 0
attr TransformerEncoderLayer.linear1 1 0 0
attr TransformerEncoderLayer.dropout 1 0 0
attr TransformerEncoderLayer.linear2 1 0 0
attr TransformerEncoderLayer.norm_first 1 0 0
attr TransformerEncoderLayer.norm1 1 0 0
attr TransformerEncoderLayer.norm2 1 0 0
attr TransformerEncoderLayer.dropout1 1 0 0
attr TransformerEncoderLayer.dropout2 1 0 0
attr TransformerEncoderLayer.activation 1 0 0
attr TransformerEncoderLayer.activation_relu_or_gelu 1 0 0
attr TransformerEncoder.layers 1 0 0
attr TransformerEncoder.num_layers 1 0 0
attr TransformerEncoder.norm 1 0 0
attr TransformerEncoder.enable_nested_tensor 1 0 0
attr TransformerEncoder.use_nested_tensor 1 0 0
attr TransformerEncoder.mask_check 1 0 0
meth TransformerDecoderLayer.init 12 10 0
meth TransformerDecoderLayer.setstate 2 0 0
attr TransformerDecoderLayer.self_attn 1 0 0
attr TransformerDecoderLayer.multihead_attn 1 0 0
attr TransformerDecoderLayer.linear1 1 0 0
attr TransformerDecoderLayer.dropout 1 0 0
attr TransformerDecoderLayer.linear2 1 0 0
attr TransformerDecoderLayer.norm_first 1 0 0
attr TransformerDecoderLayer.norm1 1 0 0
attr TransformerDecoderLayer.norm2 1 0 0
attr TransformerDecoderLayer.norm3 1 0 0
attr TransformerDecoderLayer.dropout1 1 0 0
attr TransformerDecoderLayer.dropout2 1 0 0
attr TransformerDecoderLayer.dropout3 1 0 0
attr TransformerDecoderLayer.activation 1 0 0
attr TransformerDecoder.layers 1 0 0
attr TransformerDecoder.num_layers 1 0 0
attr TransformerDecoder.norm 1 0 0

torch.nn.modules.upsampling (2 missing, 0 any)

Symbol Typable Typed Any
meth Upsample.setstate 2 0 0

torch.nn.parallel.comm (27 missing, 0 any)

Symbol Typable Typed Any
func reduce_add 3 0 0
func gather 5 0 0
func scatter 7 0 0
func reduce_add_coalesced 4 0 0
func broadcast 4 0 0
func broadcast_coalesced 4 0 0

torch.nn.parallel.data_parallel (5 missing, 8 any)

Symbol Typable Typed Any
meth DataParallel.forward 3 3 3
meth DataParallel.scatter 4 4 1
meth DataParallel.parallel_apply 4 4 1
meth DataParallel.gather 3 3 2
attr DataParallel.dim 1 0 0
attr DataParallel.module 1 0 0
attr DataParallel.device_ids 1 0 0
attr DataParallel.output_device 1 0 0
attr DataParallel.src_device_obj 1 0 0
func data_parallel 7 7 1

torch.nn.parallel.distributed (160 missing, 4 any)

Symbol Typable Typed Any
meth DistributedDataParallel.init 18 1 0
meth DistributedDataParallel._register_accum_grad_hook 1 0 0
meth DistributedDataParallel._delayed_all_reduce_hook 2 0 0
meth DistributedDataParallel._register_delay_all_reduce_hook 4 0 0
meth DistributedDataParallel._setup_in_backward_optimizers 1 0 0
meth DistributedDataParallel._fire_reducer_autograd_hook 3 0 0
meth DistributedDataParallel._root_copy_hook 3 3 2
meth DistributedDataParallel._module_wait_for_copy_hook 4 3 2
meth DistributedDataParallel._log_and_throw 3 0 0
meth DistributedDataParallel._ddp_init_helper 5 0 0
meth DistributedDataParallel.getstate 1 0 0
meth DistributedDataParallel.setstate 2 0 0
meth DistributedDataParallel._build_params_for_reducer 1 0 0
meth DistributedDataParallel._assign_modules_buffers 1 0 0
meth DistributedDataParallel._build_debug_param_to_name_mapping 2 0 0
meth DistributedDataParallel._get_parameters 3 0 0
meth DistributedDataParallel._check_default_group 1 0 0
meth DistributedDataParallel.no_sync 1 0 0
meth DistributedDataParallel._get_active_ddp_module 1 0 0
meth DistributedDataParallel._inside_ddp_forward 1 0 0
meth DistributedDataParallel._run_ddp_forward 3 0 0
meth DistributedDataParallel._clear_grad_buffer 1 0 0
meth DistributedDataParallel._lazy_init 1 0 0
meth DistributedDataParallel._pre_forward 3 0 0
meth DistributedDataParallel._post_forward 2 0 0
meth DistributedDataParallel.forward 3 0 0
meth DistributedDataParallel.scatter 4 0 0
meth DistributedDataParallel.to_kwargs 4 0 0
meth DistributedDataParallel.gather 3 0 0
meth DistributedDataParallel.train 2 0 0
meth DistributedDataParallel._check_global_requires_backward_grad_sync 2 0 0
meth DistributedDataParallel._check_and_sync_module_buffers 1 0 0
meth DistributedDataParallel._sync_final_model 2 0 0
meth DistributedDataParallel._match_all_reduce_for_bwd_pass 1 0 0
meth DistributedDataParallel._match_unused_params_allreduce 1 0 0
meth DistributedDataParallel.join 4 3 0
meth DistributedDataParallel.join_hook 2 0 0
meth DistributedDataParallel._register_buffer_comm_hook 4 1 0
meth DistributedDataParallel.register_comm_hook 3 2 0
meth DistributedDataParallel._register_builtin_comm_hook 2 0 0
meth DistributedDataParallel._register_fused_optim 5 1 0
meth DistributedDataParallel._distributed_broadcast_coalesced 4 0 0
meth DistributedDataParallel._check_sync_bufs_post_fwd 1 0 0
meth DistributedDataParallel._check_sync_bufs_pre_fwd 1 0 0
meth DistributedDataParallel.will_sync_module_buffers 1 0 0
meth DistributedDataParallel._find_common_rank 3 0 0
meth DistributedDataParallel._sync_buffers 1 0 0
meth DistributedDataParallel._sync_module_buffers 2 0 0
meth DistributedDataParallel._default_broadcast_coalesced 4 0 0
meth DistributedDataParallel._passing_sync_batchnorm_handle 2 0 0
meth DistributedDataParallel._check_comm_hook 2 0 0
meth DistributedDataParallel._get_data_parallel_params 3 0 0
meth DistributedDataParallel._set_params_and_buffers_to_ignore_for_model 3 0 0
meth DistributedDataParallel._get_ddp_logging_data 1 0 0
meth DistributedDataParallel._set_ddp_runtime_logging_sample_rate 2 0 0
meth DistributedDataParallel._set_static_graph 1 0 0
meth DistributedDataParallel._remove_autograd_hooks 1 0 0
meth DistributedDataParallel._check_reducer_finalized 1 0 0
meth DistributedDataParallel._set_sparse_metadata 2 0 0
meth DistributedDataParallel._update_process_group 2 0 0
meth DistributedDataParallel._set_ddp_sink_clone 2 1 0
prop DistributedDataParallel.join_device 1 0 0
prop DistributedDataParallel.join_process_group 1 0 0
prop DistributedDataParallel._distributed_rank 1 0 0
attr DistributedDataParallel.is_multi_device_module 1 0 0
attr DistributedDataParallel.device_type 1 0 0
attr DistributedDataParallel.static_graph 1 0 0
attr DistributedDataParallel.dim 1 0 0
attr DistributedDataParallel.module 1 0 0
attr DistributedDataParallel.device 1 0 0
attr DistributedDataParallel.broadcast_buffers 1 0 0
attr DistributedDataParallel.find_unused_parameters 1 0 0
attr DistributedDataParallel.require_backward_grad_sync 1 0 0
attr DistributedDataParallel.require_forward_param_sync 1 0 0
attr DistributedDataParallel.gradient_as_bucket_view 1 0 0
attr DistributedDataParallel.mixed_precision 1 0 0
attr DistributedDataParallel.broadcast_bucket_size 1 0 0
attr DistributedDataParallel.bucket_bytes_cap 1 0 0
attr DistributedDataParallel.use_side_stream_for_tensor_copies 1 0 0
attr DistributedDataParallel.skip_all_reduce_unused_params 1 0 0
attr DistributedDataParallel.parameters_to_ignore 1 0 0
attr DistributedDataParallel.device_ids 1 0 0
attr DistributedDataParallel.output_device 1 0 0
attr DistributedDataParallel.bucket_bytes_cap_default 1 0 0
attr DistributedDataParallel.process_group 1 0 0
attr DistributedDataParallel.device_mesh 1 0 0

torch.nn.parallel.scatter_gather (0 missing, 2 any)

Symbol Typable Typed Any
func gather 4 4 2

torch.nn.utils.init (4 missing, 0 any)

Symbol Typable Typed Any
func skip_init 4 0 0

torch.nn.utils.parametrizations (1 missing, 0 any)

Symbol Typable Typed Any
func weight_norm 4 3 0

torch.nn.utils.parametrize (3 missing, 0 any)

Symbol Typable Typed Any
func cached 1 0 0
attr ParametrizationList.is_tensor 1 0 0
attr ParametrizationList.ntensors 1 0 0

torch.nn.utils.prune (125 missing, 0 any)

Symbol Typable Typed Any
meth L1Unstructured.init 2 1 0
meth L1Unstructured.compute_mask 3 0 0
meth L1Unstructured.apply 5 0 0
attr L1Unstructured.amount 1 0 0
meth BasePruningMethod.call 3 0 0
meth BasePruningMethod.compute_mask 3 0 0
meth BasePruningMethod.apply_mask 2 0 0
meth BasePruningMethod.apply 6 0 0
meth BasePruningMethod.prune 4 0 0
meth BasePruningMethod.remove 2 1 0
func identity 3 0 0
func remove 3 0 0
func global_unstructured 5 1 0
meth Identity.compute_mask 3 0 0
meth Identity.apply 3 0 0
func is_pruned 2 1 0
meth LnStructured.init 4 1 0
meth LnStructured.compute_mask 3 0 0
meth LnStructured.apply 7 0 0
attr LnStructured.amount 1 0 0
attr LnStructured.n 1 0 0
attr LnStructured.dim 1 0 0
func random_unstructured 4 0 0
meth CustomFromMask.init 2 1 0
meth CustomFromMask.compute_mask 3 0 0
meth CustomFromMask.apply 4 0 0
attr CustomFromMask.mask 1 0 0
func custom_from_mask 4 0 0
func ln_structured 7 0 0
meth RandomStructured.init 3 1 0
meth RandomStructured.compute_mask 3 0 0
meth RandomStructured.apply 5 0 0
attr RandomStructured.amount 1 0 0
attr RandomStructured.dim 1 0 0
meth PruningContainer.init 2 1 0
meth PruningContainer.add_pruning_method 2 1 0
meth PruningContainer.iter 1 0 0
meth PruningContainer.getitem 2 0 0
meth PruningContainer.compute_mask 3 0 0
func random_structured 5 0 0
func l1_unstructured 5 0 0
meth RandomUnstructured.init 2 1 0
meth RandomUnstructured.compute_mask 3 0 0
meth RandomUnstructured.apply 4 0 0
attr RandomUnstructured.amount 1 0 0

torch.nn.utils.rnn (0 missing, 4 any)

Symbol Typable Typed Any
meth PackedSequence.cuda 3 3 2
meth PackedSequence.cpu 3 3 2

torch.nn.utils.spectral_norm (19 missing, 1 any)

Symbol Typable Typed Any
meth SpectralNormStateDictHook.init 2 1 0
meth SpectralNormStateDictHook.call 5 1 0
attr SpectralNormStateDictHook.fn 1 0 0
meth SpectralNorm.call 3 3 1
meth SpectralNorm._solve_v_and_rescale 4 0 0
meth SpectralNormLoadStateDictPreHook.init 2 1 0
meth SpectralNormLoadStateDictPreHook.call 8 1 0
attr SpectralNormLoadStateDictPreHook.fn 1 0 0

torch.nn.utils.stateless (1 missing, 0 any)

Symbol Typable Typed Any
func functional_call 7 6 0

torch.nn.utils.weight_norm (1 missing, 2 any)

Symbol Typable Typed Any
meth WeightNorm.compute_weight 2 2 1
meth WeightNorm.apply 4 3 0
meth WeightNorm.call 3 3 1

torch.onnx._internal.exporter._onnx_program (7 missing, 0 any)

Symbol Typable Typed Any
meth ONNXProgram.call 3 1 0
meth ONNXProgram.call_reference 3 1 0
meth ONNXProgram.compute_values 4 2 0
attr ONNXProgram.exported_program 1 0 0

torch.optim._adafactor (11 missing, 0 any)

Symbol Typable Typed Any
meth Adafactor.setstate 2 0 0
meth Adafactor._init_group 8 1 0
meth Adafactor.step 2 0 0

torch.optim._muon (2 missing, 0 any)

Symbol Typable Typed Any
meth Muon.step 2 0 0

torch.optim.adadelta (5 missing, 0 any)

Symbol Typable Typed Any
meth Adadelta.setstate 2 0 0
meth Adadelta._init_group 7 6 0
meth Adadelta.step 2 0 0

torch.optim.adagrad (10 missing, 0 any)

Symbol Typable Typed Any
meth Adagrad.setstate 2 0 0
meth Adagrad._init_group 6 0 0
meth Adagrad.step 2 0 0

torch.optim.adam (12 missing, 0 any)

Symbol Typable Typed Any
meth Adam.setstate 2 0 0
meth Adam._init_group 8 0 0
meth Adam.step 2 0 0

torch.optim.adamax (11 missing, 0 any)

Symbol Typable Typed Any
meth Adamax.setstate 2 0 0
meth Adamax._init_group 7 0 0
meth Adamax.step 2 0 0

torch.optim.adamw (2 missing, 0 any)

Symbol Typable Typed Any
meth AdamW.setstate 2 0 0

torch.optim.asgd (12 missing, 0 any)

Symbol Typable Typed Any
meth ASGD.setstate 2 0 0
meth ASGD._init_group 8 0 0
meth ASGD.step 2 0 0

torch.optim.lbfgs (13 missing, 0 any)

Symbol Typable Typed Any
meth LBFGS._numel 1 0 0
meth LBFGS._gather_flat_grad 1 0 0
meth LBFGS._add_grad 3 1 0
meth LBFGS._clone_param 1 0 0
meth LBFGS._set_param 2 1 0
meth LBFGS._directional_evaluate 5 0 0
meth LBFGS.step 2 0 0

torch.optim.lr_scheduler (76 missing, 0 any)

Symbol Typable Typed Any
meth ConstantLR._get_closed_form_lr 1 0 0
attr ConstantLR.factor 1 0 0
attr ConstantLR.total_iters 1 0 0
attr ChainedScheduler.optimizer 1 0 0
meth CosineAnnealingWarmRestarts.step 2 1 0
attr CosineAnnealingWarmRestarts.T_0 1 0 0
attr CosineAnnealingWarmRestarts.T_i 1 0 0
attr CosineAnnealingWarmRestarts.T_mult 1 0 0
attr CosineAnnealingWarmRestarts.eta_min 1 0 0
attr CosineAnnealingWarmRestarts.T_cur 1 0 0
attr LambdaLR.optimizer 1 0 0
meth MultiStepLR._get_closed_form_lr 1 0 0
attr MultiStepLR.milestones 1 0 0
attr MultiStepLR.gamma 1 0 0
meth ExponentialLR._get_closed_form_lr 1 0 0
attr ExponentialLR.gamma 1 0 0
attr MultiplicativeLR.optimizer 1 0 0
attr LRScheduler.optimizer 1 0 0
attr LRScheduler.last_epoch 1 0 0
meth ReduceLROnPlateau.step 3 2 0
meth ReduceLROnPlateau._reduce_lr 2 1 0
meth ReduceLROnPlateau._is_better 3 0 0
meth ReduceLROnPlateau._init_is_better 4 1 0
prop ReduceLROnPlateau.in_cooldown 1 0 0
attr ReduceLROnPlateau.factor 1 0 0
attr ReduceLROnPlateau.optimizer 1 0 0
attr ReduceLROnPlateau.patience 1 0 0
attr ReduceLROnPlateau.cooldown 1 0 0
attr ReduceLROnPlateau.eps 1 0 0
attr ReduceLROnPlateau.last_epoch 1 0 0
attr ReduceLROnPlateau.default_min_lr 1 0 0
attr ReduceLROnPlateau.min_lrs 1 0 0
meth LinearLR._get_closed_form_lr 1 0 0
attr LinearLR.start_factor 1 0 0
attr LinearLR.end_factor 1 0 0
attr LinearLR.total_iters 1 0 0
attr StepLR.step_size 1 0 0
attr StepLR.gamma 1 0 0
meth CyclicLR.scale_fn 2 1 0
attr CyclicLR.optimizer 1 0 0
attr CyclicLR.max_lrs 1 0 0
attr CyclicLR.total_size 1 0 0
attr CyclicLR.step_ratio 1 0 0
attr CyclicLR.mode 1 0 0
attr CyclicLR.gamma 1 0 0
attr CyclicLR.scale_mode 1 0 0
attr CyclicLR.cycle_momentum 1 0 0
attr CyclicLR.use_beta1 1 0 0
attr CyclicLR.base_momentums 1 0 0
attr CyclicLR.max_momentums 1 0 0
meth SequentialLR.recursive_undo 2 1 0
attr SequentialLR.last_epoch 1 0 0
attr SequentialLR.optimizer 1 0 0
attr CosineAnnealingLR.T_max 1 0 0
attr CosineAnnealingLR.eta_min 1 0 0
attr PolynomialLR.total_iters 1 0 0
attr PolynomialLR.power 1 0 0
meth OneCycleLR._anneal_func 3 0 0
meth OneCycleLR._annealing_cos 4 0 0
meth OneCycleLR._annealing_linear 4 0 0
attr OneCycleLR.optimizer 1 0 0
attr OneCycleLR.cycle_momentum 1 0 0
attr OneCycleLR.total_steps 1 0 0
attr OneCycleLR.use_beta1 1 0 0

torch.optim.nadam (12 missing, 0 any)

Symbol Typable Typed Any
meth NAdam.setstate 2 0 0
meth NAdam._init_group 8 0 0
meth NAdam.step 2 0 0

torch.optim.optimizer (1 missing, 0 any)

Symbol Typable Typed Any
attr Optimizer.defaults 1 0 0

torch.optim.radam (11 missing, 0 any)

Symbol Typable Typed Any
meth RAdam.setstate 2 0 0
meth RAdam._init_group 7 0 0
meth RAdam.step 2 0 0

torch.optim.rmsprop (12 missing, 0 any)

Symbol Typable Typed Any
meth RMSprop.setstate 2 0 0
meth RMSprop._init_group 8 0 0
meth RMSprop.step 2 0 0

torch.optim.rprop (11 missing, 0 any)

Symbol Typable Typed Any
meth Rprop.setstate 2 0 0
meth Rprop._init_group 7 0 0
meth Rprop.step 2 0 0

torch.optim.sgd (9 missing, 0 any)

Symbol Typable Typed Any
meth SGD.setstate 2 0 0
meth SGD._init_group 5 0 0
meth SGD.step 2 0 0

torch.optim.sparse_adam (2 missing, 0 any)

Symbol Typable Typed Any
meth SparseAdam.step 2 0 0

torch.optim.swa_utils (26 missing, 0 any)

Symbol Typable Typed Any
func get_ema_avg_fn 2 0 0
meth AveragedModel.init 6 5 0
meth AveragedModel.forward 3 0 0
attr AveragedModel.module 1 0 0
attr AveragedModel.avg_fn 1 0 0
attr AveragedModel.multi_avg_fn 1 0 0
attr AveragedModel.use_buffers 1 0 0
func get_ema_multi_avg_fn 2 0 0
meth SWALR.init 6 4 0
meth SWALR._linear_anneal 2 0 0
meth SWALR._cosine_anneal 2 0 0
meth SWALR._get_initial_lr 4 0 0
meth SWALR.get_lr 1 0 0
attr SWALR.anneal_epochs 1 0 0
func get_swa_avg_fn 1 0 0
func get_swa_multi_avg_fn 1 0 0

torch.overrides (9 missing, 1 any)

Symbol Typable Typed Any
func wrap_torch_function 2 1 0
func enable_reentrant_dispatch 1 0 0
func is_tensor_like 2 0 0
func resolve_name 2 0 0
attr has_torch_function 1 0 0
func handle_torch_function 5 3 1

torch.package.analyze.is_from_package (0 missing, 1 any)

Symbol Typable Typed Any
func is_from_package 2 2 1

torch.package.file_structure_representation (6 missing, 0 any)

Symbol Typable Typed Any
meth Directory.init 3 2 0
meth Directory._add_file 2 1 0
meth Directory.str 1 0 0
meth Directory._stringify_tree 4 3 0
attr Directory.name 1 0 0
attr Directory.is_dir 1 0 0

torch.package.glob_group (8 missing, 0 any)

Symbol Typable Typed Any
meth GlobGroup.init 4 3 0
meth GlobGroup.str 1 0 0
meth GlobGroup.repr 1 0 0
meth GlobGroup._glob_list 3 2 0
meth GlobGroup._glob_to_re 3 2 0
attr GlobGroup.include 1 0 0
attr GlobGroup.exclude 1 0 0
attr GlobGroup.separator 1 0 0

torch.package.importer (4 missing, 4 any)

Symbol Typable Typed Any
meth Importer.get_name 3 3 1
meth Importer.whichmodule 3 3 1
meth OrderedImporter.init 2 0 0
meth OrderedImporter._is_torchpackage_dummy 2 0 0
meth OrderedImporter.get_name 3 3 1
meth OrderedImporter.whichmodule 3 3 1

torch.package.package_exporter (48 missing, 1 any)

Symbol Typable Typed Any
meth PackagingErrorReason.repr 1 0 0
meth PackageExporter.save_source_file 4 2 0
meth PackageExporter.save_source_string 5 4 0
meth PackageExporter._write_source_string 4 3 0
meth PackageExporter._import_module 2 1 0
meth PackageExporter.add_dependency 3 1 0
meth PackageExporter.save_module 3 1 0
meth PackageExporter._intern_module 3 2 0
meth PackageExporter.save_pickle 6 5 1
meth PackageExporter.save_text 4 3 0
meth PackageExporter.save_binary 4 1 0
meth PackageExporter.intern 4 3 0
meth PackageExporter.mock 4 3 0
meth PackageExporter.extern 4 3 0
meth PackageExporter.deny 3 2 0
meth PackageExporter._persistent_id 2 0 0
meth PackageExporter.enter 1 0 0
meth PackageExporter.exit 4 0 0
meth PackageExporter._write 3 0 0
meth PackageExporter._validate_dependency_graph 1 0 0
meth PackageExporter._write_mock_file 1 0 0
meth PackageExporter._execute_dependency_graph 1 0 0
meth PackageExporter._write_python_version 1 0 0
meth PackageExporter.close 1 0 0
meth PackageExporter._finalize_zip 1 0 0
meth PackageExporter._filename 3 0 0
meth PackageExporter._can_implicitly_extern 2 1 0
attr PackageExporter.debug 1 0 0
attr PackageExporter.zip_file 1 0 0
attr PackageExporter.dependency_graph 1 0 0
attr PackageExporter.script_module_serializer 1 0 0
attr PackageExporter.storage_context 1 0 0
meth PackagingError.init 3 1 0
attr PackagingError.dependency_graph 1 0 0

torch.package.package_importer (45 missing, 3 any)

Symbol Typable Typed Any
meth PackageImporter.init 3 2 0
meth PackageImporter.import_module 3 1 0
meth PackageImporter.load_pickle 4 3 1
meth PackageImporter.id 1 0 0
meth PackageImporter.python_version 1 0 0
meth PackageImporter._read_extern 1 0 0
meth PackageImporter._make_module 5 4 0
meth PackageImporter._load_module 3 2 0
meth PackageImporter._compile_source 3 2 0
meth PackageImporter.get_source 2 1 0
meth PackageImporter.get_resource_reader 2 0 0
meth PackageImporter._install_on_parent 4 3 0
meth PackageImporter._do_find_and_load 2 0 0
meth PackageImporter._find_and_load 2 0 0
meth PackageImporter._gcd_import 4 0 0
meth PackageImporter._handle_fromlist 4 0 0
meth PackageImporter.import 6 0 0
meth PackageImporter._get_package 2 0 0
meth PackageImporter._zipfile_path 3 0 0
meth PackageImporter._add_file 2 1 0
meth PackageImporter._add_extern 2 1 0
attr PackageImporter.zip_reader 1 1 1
attr PackageImporter.root 1 0 0
attr PackageImporter.extern_modules 1 0 0
attr PackageImporter.patched_builtins 1 0 0
attr PackageImporter.storage_context 1 1 1
attr PackageImporter.last_map_location 1 0 0
attr PackageImporter.Unpickler 1 0 0
attr PackageImporter.filename 1 0 0

torch.profiler.itt (10 missing, 0 any)

Symbol Typable Typed Any
func range_pop 1 0 0
func range 4 0 0
func range_push 2 0 0
func is_available 1 0 0
func mark 2 0 0

torch.profiler.profiler (22 missing, 0 any)

Symbol Typable Typed Any
func tensorboard_trace_handler 4 3 0
func supported_activities 1 0 0
meth profile.enter 1 0 0
meth profile.exit 4 0 0
meth profile.set_custom_trace_id_callback 2 1 0
meth profile.get_trace_id 1 0 0
meth profile._transit_action 3 1 0
attr profile.on_trace_ready 1 0 0
attr profile.step_num 1 0 0
attr profile.current_action 1 0 0
attr profile.schedule 1 0 0
attr profile.record_steps 1 0 0
meth ExecutionTraceObserver.set_extra_resource_collection 2 1 0
meth ExecutionTraceObserver.get_resources_dir 2 1 0
meth ExecutionTraceObserver.get_resources_dir_for_et_path 3 2 0
meth ExecutionTraceObserver.is_running 1 0 0
prop ExecutionTraceObserver.is_registered 1 0 0
attr ExecutionTraceObserver.extra_resources_collection 1 0 0

torch.quantization (46 missing, 0 any)

Symbol Typable Typed Any
attr prepare 1 0 0
attr get_quantized_operator 1 0 0
attr prepare_ondevice_dynamic_jit 1 0 0
attr QuantStub 1 0 0
attr propagate_qconfig 1 0 0
attr HistogramObserver 1 0 0
attr default_dynamic_qconfig 1 0 0
attr get_fuser_method 1 0 0
attr default_per_channel_weight_observer 1 0 0
attr get_default_compare_output_module_list 1 0 0
attr _convert_ondevice_dynamic_jit 1 0 0
attr get_default_dynamic_quant_module_mappings 1 0 0
attr default_fixed_qparams_range_0to1_fake_quant 1 0 0
attr default_per_channel_weight_fake_quant 1 0 0
attr float_qparams_weight_only_qconfig 1 0 0
attr default_fixed_qparams_range_neg1to1_fake_quant 1 0 0
attr get_static_quant_module_class 1 0 0
attr default_weight_fake_quant 1 0 0
attr convert 1 0 0
attr prepare_qat 1 0 0
attr quantize_dynamic 1 0 0
func default_eval_fn 3 0 0
attr default_histogram_fake_quant 1 0 0
attr get_default_qconfig_propagation_list 1 0 0
attr WeightObserver 1 0 0
attr _quantize_ondevice_dynamic_jit 1 0 0
attr QuantWrapper 1 0 0
attr default_weight_observer 1 0 0
attr quantize_dynamic_jit 1 0 0
attr add_quant_dequant 1 0 0
attr default_qat_qconfig 1 0 0
attr quantize_qat 1 0 0
attr default_fake_quant 1 0 0
attr QuantType 1 0 0
attr get_default_qat_module_mappings 1 0 0
attr DeQuantStub 1 0 0
attr default_qconfig 1 0 0
attr default_observer 1 0 0
attr swap_module 1 0 0
attr get_default_static_quant_module_mappings 1 0 0
attr float16_dynamic_qconfig 1 0 0
attr QConfig 1 0 0
attr default_placeholder_observer 1 0 0
attr ObserverBase 1 0 0

torch.quasirandom (16 missing, 0 any)

Symbol Typable Typed Any
meth SobolEngine.init 4 0 0
meth SobolEngine.reset 1 0 0
meth SobolEngine.fast_forward 2 0 0
meth SobolEngine._scramble 1 0 0
meth SobolEngine.repr 1 0 0
attr SobolEngine.seed 1 0 0
attr SobolEngine.scramble 1 0 0
attr SobolEngine.dimension 1 0 0
attr SobolEngine.sobolstate 1 0 0
attr SobolEngine.quasi 1 0 0
attr SobolEngine.num_generated 1 0 0
attr SobolEngine.shift 1 0 0

torch.random (6 missing, 0 any)

Symbol Typable Typed Any
func manual_seed 2 1 0
func fork_rng 6 1 0

torch.serialization (33 missing, 7 any)

Symbol Typable Typed Any
func load 7 7 3
func save 7 7 1
func normalize_storage_type 2 0 0
func validate_hpu_device 2 0 0
func storage_to_tensor_type 2 0 0
func check_module_version_greater_or_equal 4 0 0
meth skip_data.init 2 1 0
meth skip_data.enter 1 0 0
meth skip_data.exit 4 0 0
attr skip_data.materialize_fake_tensors 1 0 0
func validate_cuda_device 2 0 0
meth set_default_mmap_options.exit 4 4 3
attr set_default_mmap_options.prev 1 0 0
meth StorageType.init 2 0 0
meth StorageType.str 1 0 0
prop StorageType.dtype 1 0 0
func set_crc32_options 2 1 0
func default_restore_location 3 0 0
func set_default_load_endianness 2 0 0
func mkdtemp 1 0 0
func location_tag 2 1 0
func register_package 4 3 0

torch.signal.windows.windows (2 missing, 0 any)

Symbol Typable Typed Any
func general_hamming 8 7 0
func general_cosine 8 7 0

torch.sparse (20 missing, 0 any)

Symbol Typable Typed Any
attr mm 1 0 0
func as_sparse_gradcheck 2 0 0
meth check_sparse_tensor_invariants.is_enabled 1 0 0
meth check_sparse_tensor_invariants.enable 1 0 0
meth check_sparse_tensor_invariants.disable 1 0 0
meth check_sparse_tensor_invariants.init 2 0 0
meth check_sparse_tensor_invariants.enter 1 0 0
meth check_sparse_tensor_invariants.exit 4 0 0
meth check_sparse_tensor_invariants.call 2 0 0
attr check_sparse_tensor_invariants.state 1 0 0
attr log_softmax 1 0 0
attr solve 1 0 0
attr softmax 1 0 0
attr addmm 1 0 0

torch.sparse.semi_structured (19 missing, 1 any)

Symbol Typable Typed Any
meth SparseSemiStructuredTensorCUSPARSELT.prune_dense_static_sort 3 2 0
meth SparseSemiStructuredTensorCUSPARSELT._mm 4 3 0
attr SparseSemiStructuredTensorCUSPARSELT._DTYPE_SHAPE_CONSTRAINTS 1 0 0
meth SparseSemiStructuredTensorCUTLASS.to_dense 1 0 0
meth SparseSemiStructuredTensorCUTLASS.prune_dense_static_sort 3 2 0
meth SparseSemiStructuredTensorCUTLASS._mm 4 3 0
attr SparseSemiStructuredTensorCUTLASS._DTYPE_SHAPE_CONSTRAINTS 1 0 0
meth SparseSemiStructuredTensor.new 11 9 0
meth SparseSemiStructuredTensor.tensor_unflatten 5 2 0
meth SparseSemiStructuredTensor.torch_dispatch 5 1 1
meth SparseSemiStructuredTensor._load_dispatch_table 2 1 0
meth SparseSemiStructuredTensor.to_dense 1 0 0
meth SparseSemiStructuredTensor._mm 4 3 0

torch.special (56 missing, 0 any)

Symbol Typable Typed Any
attr gammaincc 1 0 0
attr psi 1 0 0
attr shifted_chebyshev_polynomial_t 1 0 0
attr modified_bessel_k1 1 0 0
attr logsumexp 1 0 0
attr erfinv 1 0 0
attr log1p 1 0 0
attr entr 1 0 0
attr legendre_polynomial_p 1 0 0
attr shifted_chebyshev_polynomial_w 1 0 0
attr expm1 1 0 0
attr erfc 1 0 0
attr scaled_modified_bessel_k1 1 0 0
attr log_ndtr 1 0 0
attr bessel_j1 1 0 0
attr erfcx 1 0 0
attr zeta 1 0 0
attr hermite_polynomial_he 1 0 0
attr modified_bessel_k0 1 0 0
attr expit 1 0 0
attr bessel_j0 1 0 0
attr spherical_bessel_j0 1 0 0
attr logit 1 0 0
attr log_softmax 1 0 0
attr chebyshev_polynomial_u 1 0 0
attr scaled_modified_bessel_k0 1 0 0
attr modified_bessel_i0 1 0 0
attr erf 1 0 0
attr round 1 0 0
attr i1e 1 0 0
attr laguerre_polynomial_l 1 0 0
attr multigammaln 1 0 0
attr chebyshev_polynomial_v 1 0 0
attr xlog1py 1 0 0
attr bessel_y1 1 0 0
attr exp2 1 0 0
attr hermite_polynomial_h 1 0 0
attr softmax 1 0 0
attr i0e 1 0 0
attr shifted_chebyshev_polynomial_u 1 0 0
attr gammainc 1 0 0
attr airy_ai 1 0 0
attr chebyshev_polynomial_w 1 0 0
attr gammaln 1 0 0
attr ndtr 1 0 0
attr i0 1 0 0
attr bessel_y0 1 0 0
attr chebyshev_polynomial_t 1 0 0
attr polygamma 1 0 0
attr ndtri 1 0 0
attr shifted_chebyshev_polynomial_v 1 0 0
attr digamma 1 0 0
attr modified_bessel_i1 1 0 0
attr i1 1 0 0
attr sinc 1 0 0
attr xlogy 1 0 0

torch.storage (160 missing, 0 any)

Symbol Typable Typed Any
meth TypedStorage.fill_ 2 0 0
meth TypedStorage.new 6 0 0
meth TypedStorage.init 6 0 0
meth TypedStorage.untyped 1 0 0
meth TypedStorage.new_wrapped_storage 2 1 0
meth TypedStorage.__len__ 1 0 0
meth TypedStorage._maybe_wrap_index 3 0 0
meth TypedStorage.__setitem__ 3 0 0
meth TypedStorage._setitem 3 0 0
meth TypedStorage.__getitem__ 2 0 0
meth TypedStorage._getitem 2 0 0
meth TypedStorage.copy 3 2 0
meth TypedStorage.nbytes 1 0 0
meth TypedStorage.nbytes 1 0 0
meth TypedStorage.cuda 3 1 0
meth TypedStorage.hpu 3 1 0
meth TypedStorage.element_size 1 0 0
meth TypedStorage._element_size 1 0 0
meth TypedStorage.__str__ 1 0 0
meth TypedStorage.__repr__ 1 0 0
meth TypedStorage.__iter__ 1 0 0
meth TypedStorage.__copy__ 1 0 0
meth TypedStorage.__deepcopy__ 2 0 0
meth TypedStorage._deepcopy 2 0 0
meth TypedStorage.__sizeof__ 1 0 0
meth TypedStorage.clone 1 0 0
meth TypedStorage.tolist 1 0 0
meth TypedStorage.cpu 1 0 0
meth TypedStorage.is_pinned 2 1 0
meth TypedStorage.pin_memory 2 1 0
meth TypedStorage.share_memory 1 0 0
meth TypedStorage.share_memory 1 0 0
meth TypedStorage.new_shared 3 0 0
meth TypedStorage.size 1 0 0
meth TypedStorage._size 1 0 0
meth TypedStorage.pickle_storage_type 1 0 0
meth TypedStorage._pickle_storage_type 1 0 0
meth TypedStorage.__reduce__ 1 0 0
meth TypedStorage.data_ptr 1 0 0
meth TypedStorage._data_ptr 1 0 0
meth TypedStorage.resizable 1 0 0
meth TypedStorage.resize 2 0 0
meth TypedStorage.resize 2 0 0
meth TypedStorage.free_weak_ref 3 0 0
meth TypedStorage._weak_ref 3 0 0
meth TypedStorage.from_buffer 3 0 0
meth TypedStorage._from_buffer 5 0 0
meth TypedStorage._to 2 0 0
meth TypedStorage.double 1 0 0
meth TypedStorage.float 1 0 0
meth TypedStorage.half 1 0 0
meth TypedStorage.long 1 0 0
meth TypedStorage.int 1 0 0
meth TypedStorage.short 1 0 0
meth TypedStorage.char 1 0 0
meth TypedStorage.byte 1 0 0
meth TypedStorage.bool 1 0 0
meth TypedStorage.bfloat16 1 0 0
meth TypedStorage.complex_double 1 0 0
meth TypedStorage.complex_float 1 0 0
meth TypedStorage.float8_e5m2 1 0 0
meth TypedStorage.float8_e4m3fn 1 0 0
meth TypedStorage.float8_e5m2fnuz 1 0 0
meth TypedStorage.float8_e4m3fnuz 1 0 0
meth TypedStorage.from_file 4 0 0
meth TypedStorage._expired 3 0 0
meth TypedStorage._write_file 3 0 0
meth TypedStorage._set_from_file 3 0 0
meth TypedStorage._set_cdata 3 0 0
meth TypedStorage._share_cuda 3 0 0
meth TypedStorage.is_shared 1 0 0
meth TypedStorage.is_shared 1 0 0
meth TypedStorage._new_shared_cuda 3 0 0
meth TypedStorage._share_filename_cpu 3 0 0
meth TypedStorage.shared_decref 1 0 0
meth TypedStorage._release_ipc_counter 4 0 0
meth TypedStorage._shared_incref 3 0 0
meth TypedStorage._share_fd_cpu 3 0 0
meth TypedStorage.get_legacy_storage_class 1 0 0
prop TypedStorage._dtype 1 0 0
prop TypedStorage.is_cuda 1 0 0
prop TypedStorage.is_hpu 1 0 0
prop TypedStorage._cdata 1 0 0
prop TypedStorage.device 1 0 0
meth UntypedStorage.__getitem__ 3 0 0
meth UntypedStorage.share_memory 3 0 0
meth UntypedStorage.share_fd_cpu 3 0 0
meth UntypedStorage.share_filename_cpu 3 0 0
prop UntypedStorage.is_cuda 1 0 0
prop UntypedStorage.is_hpu 1 0 0

torch.testing._comparison (1 missing, 2 any)

Symbol Typable Typed Any
func assert_close 12 11 2

torch.torch_version (0 missing, 3 any)

Symbol Typable Typed Any
meth TorchVersion._convert_to_version 2 2 2
meth TorchVersion._cmp_wrapper 3 3 1

torch.types (0 missing, 2 any)

Symbol Typable Typed Any
meth Storage._write_file 5 5 1
meth Storage._new_with_file 3 3 1

torch.utils (7 missing, 0 any)

Symbol Typable Typed Any
attr cmake_prefix_path 1 0 0
func swap_tensors 3 0 0
func set_module 3 0 0

torch.utils.backcompat (9 missing, 0 any)

Symbol Typable Typed Any
meth Warning.init 3 1 0
meth Warning.set_enabled 2 1 0
meth Warning.get_enabled 1 0 0
attr Warning.setter 1 0 0
attr Warning.getter 1 0 0
attr Warning.enabled 1 0 0
attr keepdim_warning 1 0 0
attr broadcast_warning 1 0 0

torch.utils.benchmark.op_fuzzers.binary (3 missing, 0 any)

Symbol Typable Typed Any
meth BinaryOpFuzzer.init 4 1 0

torch.utils.benchmark.op_fuzzers.sparse_binary (3 missing, 0 any)

Symbol Typable Typed Any
meth BinaryOpSparseFuzzer.init 4 1 0

torch.utils.benchmark.op_fuzzers.spectral (1 missing, 0 any)

Symbol Typable Typed Any
meth SpectralOpFuzzer.init 5 4 0

torch.utils.benchmark.op_fuzzers.unary (3 missing, 0 any)

Symbol Typable Typed Any
meth UnaryOpFuzzer.init 4 1 0

torch.utils.benchmark.utils.common (0 missing, 1 any)

Symbol Typable Typed Any
meth Measurement.getattr 2 2 1

torch.utils.benchmark.utils.compare (5 missing, 0 any)

Symbol Typable Typed Any
meth Compare.extend_results 2 1 0
meth Compare.colorize 2 1 0
meth Compare._render 1 0 0
meth Compare._group_by_label 2 1 0
meth Compare._layout 2 1 0

torch.utils.benchmark.utils.compile (2 missing, 0 any)

Symbol Typable Typed Any
func bench_all 6 5 0
func benchmark_compile 8 7 0

torch.utils.benchmark.utils.cpp_jit (3 missing, 0 any)

Symbol Typable Typed Any
attr LOCK 1 0 0
attr CONDA_PREFIX 1 0 0
attr SOURCE_ROOT 1 0 0

torch.utils.benchmark.utils.fuzzer (37 missing, 0 any)

Symbol Typable Typed Any
meth Fuzzer._unpack 3 0 0
meth Fuzzer.take 2 0 0
meth Fuzzer._generate 2 0 0
meth Fuzzer._resolve_aliases 2 0 0
prop Fuzzer.rejection_rate 1 0 0
meth FuzzedTensor.init 13 11 0
meth FuzzedTensor.default_tensor_constructor 4 0 0
meth FuzzedTensor._make_tensor 3 0 0
meth FuzzedTensor._get_size_and_steps 2 0 0
meth FuzzedTensor.satisfies_constraints 2 1 0
prop FuzzedTensor.name 1 0 0
meth FuzzedParameter.sample 2 0 0
meth FuzzedParameter._check_distribution 2 0 0
meth FuzzedParameter._loguniform 2 0 0
meth FuzzedParameter._uniform 2 0 0
meth FuzzedParameter._custom_distribution 2 0 0
prop FuzzedParameter.name 1 0 0
attr FuzzedParameter.strict 1 0 0
meth ParameterAlias.init 2 1 0
attr ParameterAlias.alias_to 1 0 0

torch.utils.benchmark.utils.sparse_fuzzer (11 missing, 0 any)

Symbol Typable Typed Any
meth FuzzedSparseTensor.init 12 10 0
meth FuzzedSparseTensor.sparse_tensor_constructor 6 0 0
meth FuzzedSparseTensor._make_tensor 3 0 0

torch.utils.benchmark.utils.valgrind_wrapper.timer_interface (0 missing, 3 any)

Symbol Typable Typed Any
meth CopyIfCallgrind.init 3 3 1
prop CopyIfCallgrind.value 1 1 1
meth FunctionCounts.getitem 2 2 1

torch.utils.bundled_inputs (7 missing, 0 any)

Symbol Typable Typed Any
func bundle_randn 3 0 0
func augment_model_with_bundled_inputs 6 5 0
func augment_many_model_functions_with_bundled_inputs 6 5 0
func bundle_large_tensor 2 0 0

torch.utils.checkpoint (31 missing, 0 any)

Symbol Typable Typed Any
func noop_context_fn 1 0 0
func create_selective_checkpoint_contexts 3 0 0
func set_device_states 4 1 0
func get_device_states 2 1 0
meth SelectiveCheckpointContext.init 2 1 0
attr SelectiveCheckpointContext.is_recompute 1 0 0
meth CheckpointFunction.forward 5 0 0
meth CheckpointFunction.backward 3 0 0
func checkpoint_sequential 6 0 0
func checkpoint 9 5 0
func set_checkpoint_early_stop 2 1 0
func set_checkpoint_debug_enabled 2 1 0
attr SAC_IGNORED_OPS 1 0 0

torch.utils.collect_env (64 missing, 0 any)

Symbol Typable Typed Any
func get_cudnn_version 2 0 0
func get_python_platform 1 0 0
func get_platform 1 0 0
func get_os 2 0 0
func get_cpu_info 2 0 0
func get_gpu_info 2 0 0
func get_nvidia_driver_version 2 0 0
func check_release_file 2 0 0
func get_env_info 1 0 0
func get_lsb_version 2 0 0
func run 2 0 0
func get_windows_version 2 0 0
func get_intel_gpu_onboard 2 0 0
func get_running_cuda_version 2 0 0
func get_cmake_version 2 0 0
func get_intel_gpu_driver_version 2 0 0
func get_cuda_module_loading_config 1 0 0
func get_clang_version 2 0 0
func pretty_str 2 0 0
func get_mac_version 2 0 0
func get_linux_pkg_version 3 0 0
attr env_info_fmt 1 0 0
func is_xnnpack_available 1 0 0
func run_and_read_all 3 0 0
func get_intel_gpu_detected 2 0 0
func run_and_return_first_line 3 0 0
func get_conda_packages 3 0 0
func get_pip_packages 3 0 0
func get_pretty_env_info 1 0 0
func run_and_parse_first_match 4 0 0
func get_libc_version 1 0 0
func get_gcc_version 2 0 0
func get_nvidia_smi 1 0 0

torch.utils.cpp_backtrace (2 missing, 0 any)

Symbol Typable Typed Any
func get_cpp_backtrace 3 1 0

torch.utils.cpp_extension (63 missing, 0 any)

Symbol Typable Typed Any
func load_inline 20 0 0
func include_paths 3 2 0
func check_compiler_is_gcc 2 1 0
func load 15 3 0
func get_compiler_abi_compatibility_and_version 2 1 0
func CppExtension 5 0 0
func get_cxx_compiler 1 0 0
func SyclExtension 5 0 0
func CUDAExtension 5 0 0
meth BuildExtension.with_options 2 0 0
meth BuildExtension.init 3 1 0
meth BuildExtension.get_ext_filename 2 0 0
meth BuildExtension._add_compile_flag 3 1 0
meth BuildExtension._hipify_compile_flags 2 1 0
meth BuildExtension._define_torch_extension_name 2 1 0
attr BuildExtension.no_python_abi_suffix 1 0 0
attr BuildExtension.use_ninja 1 0 0

torch.utils.data._utils.collate (2 missing, 0 any)

Symbol Typable Typed Any
func default_convert 2 0 0

torch.utils.data.backward_compatibility (1 missing, 0 any)

Symbol Typable Typed Any
func worker_init_fn 2 1 0

torch.utils.data.dataloader (21 missing, 0 any)

Symbol Typable Typed Any
meth DataLoader.init 18 16 0
meth DataLoader.setattr 3 1 0
prop DataLoader.multiprocessing_context 2 0 0
prop DataLoader._auto_collation 1 0 0
prop DataLoader._index_sampler 1 0 0
attr DataLoader.worker_init_fn 1 0 0
attr DataLoader.multiprocessing_context 1 0 0
attr DataLoader.in_order 1 0 0
attr DataLoader.batch_sampler 1 0 0
attr DataLoader.generator 1 0 0
attr DataLoader.collate_fn 1 0 0
attr DataLoader.persistent_workers 1 0 0
meth _DatasetKind.create_fetcher 6 0 0

torch.utils.data.datapipes._decorator (13 missing, 6 any)

Symbol Typable Typed Any
func argument_validation 2 0 0
meth functional_datapipe.init 3 2 0
meth functional_datapipe.call 2 0 0
attr functional_datapipe.enable_df_api_tracing 1 0 0
meth guaranteed_datapipes_determinism.exit 4 4 3
func runtime_validation 2 0 0
meth non_deterministic.call 3 0 0
meth non_deterministic.deterministic_wrapper_fn 3 1 0
meth runtime_validation_disabled.exit 4 4 3

torch.utils.data.datapipes.dataframe.dataframe_wrapper (38 missing, 0 any)

Symbol Typable Typed Any
func create_dataframe 3 0 0
func is_column 2 0 0
func is_dataframe 2 0 0
func iterate 2 0 0
func get_df_wrapper 1 0 0
meth PandasWrapper.create_dataframe 3 0 0
meth PandasWrapper.is_dataframe 2 0 0
meth PandasWrapper.is_column 2 0 0
meth PandasWrapper.iterate 2 0 0
meth PandasWrapper.concat 2 0 0
meth PandasWrapper.get_item 3 0 0
meth PandasWrapper.get_len 2 0 0
meth PandasWrapper.get_columns 2 0 0
func set_df_wrapper 2 1 0
func concat 2 0 0
func get_item 3 0 0
func get_len 2 0 0
func get_columns 2 0 0

torch.utils.data.datapipes.dataframe.dataframes (129 missing, 0 any)

Symbol Typable Typed Any
meth Capture.init 2 1 0
meth Capture._ops_str 1 0 0
meth Capture.getstate 1 0 0
meth Capture.setstate 2 0 0
meth Capture.getattr 2 0 0
meth Capture.getitem 2 0 0
meth Capture.setitem 3 1 0
meth Capture.add 2 0 0
meth Capture.sub 2 0 0
meth Capture.mul 2 0 0
meth Capture._is_context_empty 1 0 0
meth Capture.apply_ops_2 2 1 0
meth Capture.call 3 0 0
prop Capture.columns 1 0 0
attr Capture.ctx 1 0 0
meth CaptureSub.init 4 1 0
meth CaptureSub.execute 1 0 0
attr CaptureSub.ctx 1 0 0
attr CaptureSub.left 1 0 0
attr CaptureSub.right 1 0 0
meth CaptureAdd.init 4 1 0
meth CaptureAdd.execute 1 0 0
attr CaptureAdd.ctx 1 0 0
attr CaptureAdd.left 1 0 0
attr CaptureAdd.right 1 0 0
meth CaptureGetAttr.init 4 1 0
meth CaptureGetAttr.execute 1 0 0
attr CaptureGetAttr.ctx 1 0 0
attr CaptureGetAttr.src 1 0 0
attr CaptureGetAttr.name 1 0 0
meth CaptureCall.init 4 1 0
meth CaptureCall.execute 1 0 0
attr CaptureCall.kwargs 1 0 0
attr CaptureCall.callable 1 0 0
attr CaptureCall.ctx 1 0 0
meth CaptureLikeMock.init 2 1 0
meth CaptureLikeMock.enter 1 0 0
meth CaptureLikeMock.exit 2 0 0
attr CaptureLikeMock.get_target 1 0 0
attr CaptureLikeMock.attribute 1 0 0
attr CaptureLikeMock.name 1 0 0
meth CaptureDataFrameWithDataPipeOps.as_datapipe 1 0 0
meth CaptureDataFrameWithDataPipeOps.raw_iterator 1 0 0
meth CaptureDataFrameWithDataPipeOps.iter 1 0 0
meth CaptureDataFrameWithDataPipeOps.batch 4 1 0
meth CaptureDataFrameWithDataPipeOps.groupby 6 0 0
meth CaptureDataFrameWithDataPipeOps.shuffle 3 0 0
meth CaptureDataFrameWithDataPipeOps.filter 3 0 0
meth CaptureDataFrameWithDataPipeOps.collate 3 1 0
meth CaptureDataFrameWithDataPipeOps.getattr 2 0 0
meth CaptureInitial.init 2 1 0
attr CaptureInitial.name 1 0 0
meth CaptureMul.init 4 1 0
meth CaptureMul.execute 1 0 0
attr CaptureMul.ctx 1 0 0
attr CaptureMul.left 1 0 0
attr CaptureMul.right 1 0 0
meth DataFrameTracedOps.init 3 1 0
meth DataFrameTracedOps.iter 1 0 0
attr DataFrameTracedOps.source_datapipe 1 0 0
attr DataFrameTracedOps.output_var 1 0 0
meth CaptureVariable.init 3 1 0
meth CaptureVariable.execute 1 0 0
meth CaptureVariable.apply_ops 2 0 0
attr CaptureVariable.ctx 1 0 0
attr CaptureVariable.value 1 0 0
attr CaptureVariable.name 1 0 0
meth CaptureSetItem.init 5 1 0
attr CaptureSetItem.ctx 1 0 0
attr CaptureSetItem.left 1 0 0
attr CaptureSetItem.key 1 0 0
attr CaptureSetItem.value 1 0 0
func get_val 2 0 0
meth CaptureF.init 3 1 0
attr CaptureF.kwargs 1 0 0
attr CaptureF.ctx 1 0 0
meth CaptureA.execute 1 0 0
meth DataFrameTracer.set_shuffle_settings 3 1 0
meth DataFrameTracer.init 3 1 0
meth CaptureGetItem.init 4 1 0
meth CaptureGetItem.execute 1 0 0
attr CaptureGetItem.ctx 1 0 0
attr CaptureGetItem.left 1 0 0
attr CaptureGetItem.key 1 0 0

torch.utils.data.datapipes.dataframe.datapipes (28 missing, 0 any)

Symbol Typable Typed Any
meth DataFramesAsTuplesPipe.init 2 1 0
meth DataFramesAsTuplesPipe.iter 1 0 0
attr DataFramesAsTuplesPipe.source_datapipe 1 0 0
meth ShuffleDataFramesPipe.init 2 1 0
meth ShuffleDataFramesPipe.iter 1 0 0
attr ShuffleDataFramesPipe.source_datapipe 1 0 0
meth ConcatDataFramesPipe.init 3 1 0
meth ConcatDataFramesPipe.iter 1 0 0
attr ConcatDataFramesPipe.source_datapipe 1 0 0
attr ConcatDataFramesPipe.n_batch 1 0 0
meth PerRowDataFramesPipe.init 2 1 0
meth PerRowDataFramesPipe.iter 1 0 0
attr PerRowDataFramesPipe.source_datapipe 1 0 0
meth FilterDataFramesPipe.init 3 1 0
meth FilterDataFramesPipe.iter 1 0 0
attr FilterDataFramesPipe.source_datapipe 1 0 0
attr FilterDataFramesPipe.filter_fn 1 0 0
meth ExampleAggregateAsDataFrames.init 4 1 0
meth ExampleAggregateAsDataFrames._as_list 2 0 0
meth ExampleAggregateAsDataFrames.iter 1 0 0
attr ExampleAggregateAsDataFrames.source_datapipe 1 0 0
attr ExampleAggregateAsDataFrames.columns 1 0 0
attr ExampleAggregateAsDataFrames.dataframe_size 1 0 0

torch.utils.data.datapipes.datapipe (15 missing, 19 any)

Symbol Typable Typed Any
meth IterDataPipe.getattr 2 1 1
meth IterDataPipe.register_function 3 3 2
meth IterDataPipe.register_datapipe_as_function 4 3 2
meth IterDataPipe.getstate 1 0 0
meth IterDataPipe.reduce_ex 3 2 2
meth IterDataPipe.set_getstate_hook 2 2 1
meth IterDataPipe.set_reduce_ex_hook 2 2 1
meth IterDataPipe.filter 3 2 0
meth IterDataPipe.map 4 2 0
meth IterDataPipe.mux 2 1 0
meth IterDataPipe.sharding_filter 2 1 0
meth DFIterDataPipe._is_dfpipe 1 0 0
meth DFIterDataPipe.iter 1 0 0
meth MapDataPipe.getattr 2 1 1
meth MapDataPipe.register_function 3 3 2
meth MapDataPipe.register_datapipe_as_function 3 2 2
meth MapDataPipe.getstate 1 0 0
meth MapDataPipe.reduce_ex 3 2 2
meth MapDataPipe.set_getstate_hook 2 2 1
meth MapDataPipe.set_reduce_ex_hook 2 2 1
attr UNTRACABLE_DATAFRAME_PIPES 1 1 1

torch.utils.data.datapipes.iter.callable (6 missing, 0 any)

Symbol Typable Typed Any
meth MapperIterDataPipe.init 5 3 0
meth MapperIterDataPipe._apply_fn 2 0 0
attr MapperIterDataPipe.input_col 1 0 0
attr MapperIterDataPipe.output_col 1 0 0

torch.utils.data.datapipes.iter.combinatorics (8 missing, 0 any)

Symbol Typable Typed Any
attr SamplerIterDataPipe.sampler_args 1 0 0
attr SamplerIterDataPipe.sampler_kwargs 1 0 0
meth ShufflerIterDataPipe.set_shuffle 2 0 0
meth ShufflerIterDataPipe.set_seed 2 1 0
meth ShufflerIterDataPipe.getstate 1 0 0
meth ShufflerIterDataPipe.setstate 2 0 0

torch.utils.data.datapipes.iter.combining (8 missing, 0 any)

Symbol Typable Typed Any
meth MultiplexerIterDataPipe.init 2 1 0
meth MultiplexerIterDataPipe.iter 1 0 0
meth MultiplexerIterDataPipe.getstate 1 0 0
meth MultiplexerIterDataPipe.setstate 2 0 0
attr MultiplexerIterDataPipe.datapipes 1 0 0
meth ForkerIterDataPipe.new 5 4 0
meth DemultiplexerIterDataPipe.new 6 5 0

torch.utils.data.datapipes.iter.grouping (21 missing, 0 any)

Symbol Typable Typed Any
meth GrouperIterDataPipe._remove_biggest_key 1 0 0
meth GrouperIterDataPipe.iter 1 0 0
meth GrouperIterDataPipe.getstate 1 0 0
meth GrouperIterDataPipe.setstate 2 0 0
attr GrouperIterDataPipe.datapipe 1 0 0
attr GrouperIterDataPipe.group_key_fn 1 0 0
attr GrouperIterDataPipe.keep_key 1 0 0
attr GrouperIterDataPipe.max_buffer_size 1 0 0
attr GrouperIterDataPipe.curr_buffer_size 1 0 0
attr GrouperIterDataPipe.group_size 1 0 0
attr GrouperIterDataPipe.guaranteed_group_size 1 0 0
attr GrouperIterDataPipe.drop_remaining 1 0 0
attr GrouperIterDataPipe.wrapper_class 1 0 0
meth UnBatcherIterDataPipe.iter 1 0 0
meth UnBatcherIterDataPipe._dive 3 0 0
attr UnBatcherIterDataPipe.datapipe 1 0 0
attr UnBatcherIterDataPipe.unbatch_level 1 0 0
attr BatcherIterDataPipe.wrapper_class 1 0 0

torch.utils.data.datapipes.iter.routeddecoder (1 missing, 0 any)

Symbol Typable Typed Any
attr RoutedDecoderIterDataPipe.decoder 1 0 0

torch.utils.data.datapipes.iter.selecting (3 missing, 0 any)

Symbol Typable Typed Any
meth FilterIterDataPipe.init 4 3 0
meth FilterIterDataPipe._apply_filter_fn 2 1 0
attr FilterIterDataPipe.input_col 1 0 0

torch.utils.data.datapipes.iter.sharding (10 missing, 0 any)

Symbol Typable Typed Any
meth ShardingFilterIterDataPipe.init 3 2 0
meth ShardingFilterIterDataPipe.apply_sharding 4 0 0
meth ShardingFilterIterDataPipe.iter 1 0 0
attr ShardingFilterIterDataPipe.source_datapipe 1 0 0
attr ShardingFilterIterDataPipe.sharding_group_filter 1 0 0
attr ShardingFilterIterDataPipe.num_of_instances 1 0 0
attr ShardingFilterIterDataPipe.instance_id 1 0 0

torch.utils.data.datapipes.iter.streamreader (2 missing, 0 any)

Symbol Typable Typed Any
attr StreamReaderIterDataPipe.datapipe 1 0 0
attr StreamReaderIterDataPipe.chunk 1 0 0

torch.utils.data.datapipes.iter.utils (2 missing, 0 any)

Symbol Typable Typed Any
attr IterableWrapperIterDataPipe.iterable 1 0 0
attr IterableWrapperIterDataPipe.deepcopy 1 0 0

torch.utils.data.datapipes.map.callable (3 missing, 0 any)

Symbol Typable Typed Any
meth MapperMapDataPipe.getitem 2 1 0
func default_fn 2 0 0

torch.utils.data.datapipes.map.combinatorics (7 missing, 0 any)

Symbol Typable Typed Any
meth ShufflerIterDataPipe.set_shuffle 2 0 0
meth ShufflerIterDataPipe.set_seed 2 1 0
meth ShufflerIterDataPipe.getstate 1 0 0
meth ShufflerIterDataPipe.setstate 2 0 0
attr ShufflerIterDataPipe.indices 1 0 0

torch.utils.data.datapipes.map.combining (2 missing, 0 any)

Symbol Typable Typed Any
meth ConcaterMapDataPipe.getitem 2 1 0
meth ZipperMapDataPipe.getitem 2 1 0

torch.utils.data.datapipes.map.grouping (2 missing, 0 any)

Symbol Typable Typed Any
meth BatcherMapDataPipe.getitem 2 1 0
attr BatcherMapDataPipe.wrapper_class 1 0 0

torch.utils.data.datapipes.utils.common (22 missing, 0 any)

Symbol Typable Typed Any
func get_file_binaries_from_pathnames 4 3 0
meth StreamWrapper.init 4 1 0
meth StreamWrapper.close_streams 3 1 0
meth StreamWrapper.getattr 2 0 0
meth StreamWrapper.close 3 1 0
meth StreamWrapper.dir 1 0 0
meth StreamWrapper.iter 1 0 0
meth StreamWrapper.next 1 0 0
meth StreamWrapper.getstate 1 0 0
meth StreamWrapper.setstate 2 0 0
attr StreamWrapper.file_obj 1 0 0
attr StreamWrapper.child_counter 1 0 0
attr StreamWrapper.parent_stream 1 0 0
attr StreamWrapper.close_on_last_child 1 0 0
attr StreamWrapper.name 1 0 0
attr StreamWrapper.closed 1 0 0

torch.utils.data.datapipes.utils.decoder (42 missing, 0 any)

Symbol Typable Typed Any
func basichandlers 3 1 0
meth ImageHandler.init 2 1 0
meth ImageHandler.call 3 0 0
attr ImageHandler.imagespec 1 0 0
func mathandler 2 0 0
func audiohandler 3 0 0
func extension_extract_fn 2 0 0
func handle_extension 3 0 0
func imagehandler 2 0 0
meth MatHandler.init 2 1 0
meth MatHandler.call 3 0 0
attr MatHandler.sio 1 0 0
attr MatHandler.loadmat_kwargs 1 0 0
meth Decoder.init 3 1 0
meth Decoder.add_handler 2 1 0
meth Decoder._is_stream_handle 2 0 0
meth Decoder.decode1 3 0 0
meth Decoder.decode 2 0 0
meth Decoder.call 2 0 0
attr Decoder.handlers 1 0 0
attr Decoder.key_fn 1 0 0
func videohandler 3 0 0

torch.utils.data.dataset (16 missing, 0 any)

Symbol Typable Typed Any
meth Dataset.getitem 2 1 0
meth ChainDataset.iter 1 0 0
attr ChainDataset.datasets 1 0 0
meth Subset.getitem 2 0 0
meth ConcatDataset.cumsum 2 0 0
meth ConcatDataset.getitem 2 0 0
prop ConcatDataset.cummulative_sizes 1 0 0
meth StackDataset.getitem 2 0 0
meth StackDataset.getitems 2 1 0
meth TensorDataset.getitem 2 0 0
meth IterableDataset.add 2 1 0

torch.utils.data.distributed (9 missing, 0 any)

Symbol Typable Typed Any
attr DistributedSampler.dataset 1 0 0
attr DistributedSampler.num_replicas 1 0 0
attr DistributedSampler.rank 1 0 0
attr DistributedSampler.epoch 1 0 0
attr DistributedSampler.drop_last 1 0 0
attr DistributedSampler.total_size 1 0 0
attr DistributedSampler.shuffle 1 0 0
attr DistributedSampler.seed 1 0 0
attr DistributedSampler.num_samples 1 0 0

torch.utils.data.graph_settings (1 missing, 1 any)

Symbol Typable Typed Any
func apply_shuffle_seed 3 3 1
func apply_sharding 5 4 0

torch.utils.data.sampler (9 missing, 0 any)

Symbol Typable Typed Any
attr BatchSampler.sampler 1 0 0
attr BatchSampler.batch_size 1 0 0
attr BatchSampler.drop_last 1 0 0
meth SubsetRandomSampler.init 3 2 0
attr SubsetRandomSampler.generator 1 0 0
meth RandomSampler.init 5 4 0
attr RandomSampler.generator 1 0 0
meth WeightedRandomSampler.init 5 4 0
attr WeightedRandomSampler.generator 1 0 0

torch.utils.dlpack (0 missing, 1 any)

Symbol Typable Typed Any
func from_dlpack 4 4 1

torch.utils.file_baton (7 missing, 0 any)

Symbol Typable Typed Any
meth FileBaton.init 4 1 0
attr FileBaton.lock_file_path 1 0 0
attr FileBaton.wait_seconds 1 0 0
attr FileBaton.fd 1 0 0
attr FileBaton.warn_after_seconds 1 0 0

torch.utils.flop_counter (16 missing, 0 any)

Symbol Typable Typed Any
func register_flop_formula 3 1 0
meth FlopCounterMode.get_table 2 0 0
meth FlopCounterMode.enter 1 0 0
meth FlopCounterMode.exit 2 0 0
meth FlopCounterMode._count_flops 5 0 0
attr FlopCounterMode.depth 1 0 0
attr FlopCounterMode.display 1 0 0
attr FlopCounterMode.flop_registry 1 0 0
attr FlopCounterMode.mod_tracker 1 0 0

torch.utils.hipify.cuda_to_hip_mappings (8 missing, 0 any)

Symbol Typable Typed Any
attr CUDA_TYPE_NAME_MAP 1 0 0
attr CUDA_IDENTIFIER_MAP 1 0 0
attr MATH_TRANSPILATIONS 1 0 0
attr C10_MAPPINGS 1 0 0
attr CUDA_SPECIAL_MAP 1 0 0
attr CAFFE2_SPECIFIC_MAPPINGS 1 0 0
attr CUDA_INCLUDE_MAP 1 0 0
attr PYTORCH_SPECIFIC_MAPPINGS 1 0 0

torch.utils.hipify.hipify_python (81 missing, 0 any)

Symbol Typable Typed Any
func extract_arguments 3 0 0
meth InputError.init 2 1 0
attr InputError.message 1 0 0
func replace_extern_shared 2 0 0
func openf 3 0 0
meth HipifyResult.init 3 1 0
attr HipifyResult.current_state 1 0 0
attr HipifyResult.hipified_path 1 0 0
attr HipifyResult.status 1 0 0
func find_bracket_group 3 0 0
meth GeneratedFileCleaner.init 2 1 0
meth GeneratedFileCleaner.enter 1 0 0
meth GeneratedFileCleaner.open 4 0 0
meth GeneratedFileCleaner.makedirs 3 1 0
meth GeneratedFileCleaner.exit 4 0 0
attr GeneratedFileCleaner.keep_intermediates 1 0 0
attr GeneratedFileCleaner.files_to_clean 1 0 0
attr GeneratedFileCleaner.dirs_to_clean 1 0 0
func file_specific_replacement 5 1 0
meth Trie.add 2 1 0
meth Trie.dump 1 0 0
meth Trie.quote 2 0 0
meth Trie.search 2 0 0
meth Trie._pattern 3 0 0
meth Trie.pattern 1 0 0
meth Trie.export_to_regex 1 0 0
attr Trie.root 1 0 0
func file_add_header 3 1 0
func find_parentheses_group 3 0 0
func is_out_of_place 2 1 0
func fix_static_global_kernels 2 0 0
func hip_header_magic 2 0 0
func is_pytorch_file 2 1 0
func find_closure_group 4 0 0
func get_hip_file_path 3 0 0
func replace_math_functions 2 0 0
func is_special_file 2 1 0
func processKernelLaunches 3 0 0
func compute_stats 2 1 0
func is_cusparse_file 2 0 0
func is_caffe2_gpu_file 2 0 0
func add_dim3 3 0 0

torch.utils.hooks (32 missing, 5 any)

Symbol Typable Typed Any
func warn_if_has_hooks 2 1 0
meth BackwardHook.init 4 1 0
meth BackwardHook._pack_with_none 4 0 0
meth BackwardHook._unpack_none 3 0 0
meth BackwardHook._set_user_hook 2 1 0
meth BackwardHook._apply_on_tensors 3 0 0
meth BackwardHook.setup_input_hook 2 0 0
meth BackwardHook.setup_output_hook 2 0 0
attr BackwardHook.user_hooks 1 0 0
attr BackwardHook.user_pre_hooks 1 0 0
attr BackwardHook.module 1 0 0
attr BackwardHook.grad_outputs 1 0 0
attr BackwardHook.n_outputs 1 0 0
attr BackwardHook.output_tensors_index 1 0 0
attr BackwardHook.n_inputs 1 0 0
attr BackwardHook.input_tensors_index 1 0 0
meth RemovableHandle.init 3 3 2
meth RemovableHandle.getstate 1 0 0
meth RemovableHandle.setstate 2 1 0
meth RemovableHandle.exit 4 4 3
attr RemovableHandle.hooks_dict_ref 1 0 0
func unserializable_hook 2 0 0

torch.utils.jit.log_extract (23 missing, 0 any)

Symbol Typable Typed Any
func time_cpu 4 0 0
func run_baseline_no_fusion 3 1 0
func run_nvfuser 3 1 0
func make_tensor_from_type 2 1 0
func no_fuser 3 0 0
func time_cuda 4 0 0
func run_test 5 1 0
func run_nnc 4 1 0

torch.utils.mkldnn (37 missing, 0 any)

Symbol Typable Typed Any
meth MkldnnPrelu.init 3 1 0
meth MkldnnPrelu.getstate 1 0 0
meth MkldnnPrelu.setstate 2 0 0
meth MkldnnPrelu.forward 2 0 0
meth MkldnnConv2d.init 3 1 0
meth MkldnnConv2d.setstate 2 0 0
meth MkldnnLinear.init 3 1 0
meth MkldnnLinear.getstate 1 0 0
meth MkldnnLinear.setstate 2 0 0
meth MkldnnLinear.forward 2 0 0
meth MkldnnBatchNorm.init 2 1 0
meth MkldnnBatchNorm.getstate 1 0 0
meth MkldnnBatchNorm.setstate 2 0 0
meth MkldnnBatchNorm.forward 2 0 0
attr MkldnnBatchNorm.eps 1 0 0
attr MkldnnBatchNorm.exponential_average_factor 1 0 0
func to_mkldnn 3 0 0
meth MkldnnConv3d.init 3 1 0
meth MkldnnConv3d.setstate 2 0 0
meth MkldnnConv1d.init 3 1 0
meth MkldnnConv1d.setstate 2 0 0

torch.utils.mobile_optimizer (1 missing, 0 any)

Symbol Typable Typed Any
func generate_mobile_module_lints 2 1 0

torch.utils.model_dump (15 missing, 0 any)

Symbol Typable Typed Any
func burn_in_info 3 0 0
func get_storage_info 2 0 0
func get_inline_skeleton 1 0 0
func get_info_and_burn_skeleton 3 0 0
func hierarchical_pickle 2 0 0
func get_model_info 4 0 0

torch.utils.module_tracker (17 missing, 0 any)

Symbol Typable Typed Any
meth ModuleTracker._get_mod_name 2 0 0
meth ModuleTracker._get_append_fn 3 0 0
meth ModuleTracker._get_pop_fn 3 0 0
meth ModuleTracker._fw_pre_hook 3 1 0
meth ModuleTracker._fw_post_hook 4 1 0
meth ModuleTracker.enter 1 0 0
meth ModuleTracker.exit 2 0 0
prop ModuleTracker.is_bw 1 0 0

torch.utils.show_pickle (39 missing, 0 any)

Symbol Typable Typed Any
meth FakeClass.init 3 1 0
meth FakeClass.call 2 0 0
meth FakeClass.fake_new 2 0 0
attr FakeClass.module 1 0 0
attr FakeClass.name 1 0 0
meth DumpUnpickler.init 4 1 0
meth DumpUnpickler.find_class 3 0 0
meth DumpUnpickler.persistent_load 2 0 0
meth DumpUnpickler.dump 3 0 0
attr DumpUnpickler.catch_invalid_utf8 1 0 0
attr DumpUnpickler.dispatch 1 0 0
func main 3 1 0
meth FakeObject.init 4 1 0
meth FakeObject.setstate 2 0 0
meth FakeObject.pp_format 8 1 0
attr FakeObject.module 1 0 0
attr FakeObject.name 1 0 0
attr FakeObject.args 1 0 0
attr FakeObject.state 1 0 0

torch.utils.tensorboard.summary (97 missing, 0 any)

Symbol Typable Typed Any
func pr_curve 6 0 0
func hparams 4 0 0
func scalar 6 0 0
func make_image 5 0 0
func pr_curve_raw 10 0 0
func audio 4 0 0
func custom_scalars 2 0 0
func histogram_raw 9 0 0
func image 5 0 0
func mesh 8 0 0
func make_video 3 0 0
func draw_boxes 4 0 0
func image_boxes 7 0 0
func text 3 0 0
func make_histogram 4 0 0
func video 4 0 0
func compute_curve 5 0 0
func tensor_proto 3 0 0
func histogram 5 0 0

torch.utils.tensorboard.writer (149 missing, 0 any)

Symbol Typable Typed Any
meth FileWriter.init 5 1 0
meth FileWriter.get_logdir 1 0 0
meth FileWriter.add_event 4 1 0
meth FileWriter.add_summary 4 1 0
meth FileWriter.add_graph 3 1 0
meth FileWriter.add_onnx_graph 3 1 0
attr FileWriter.event_writer 1 0 0
meth SummaryWriter.init 7 1 0
meth SummaryWriter._get_file_writer 1 0 0
meth SummaryWriter.get_logdir 1 0 0
meth SummaryWriter.add_hparams 6 1 0
meth SummaryWriter.add_scalar 7 1 0
meth SummaryWriter.add_scalars 5 1 0
meth SummaryWriter.add_tensor 5 1 0
meth SummaryWriter.add_histogram 7 1 0
meth SummaryWriter.add_histogram_raw 11 1 0
meth SummaryWriter.add_image 6 1 0
meth SummaryWriter.add_images 6 1 0
meth SummaryWriter.add_image_with_boxes 9 1 0
meth SummaryWriter.add_video 6 1 0
meth SummaryWriter.add_audio 6 1 0
meth SummaryWriter.add_text 5 1 0
meth SummaryWriter.add_onnx_graph 2 1 0
meth SummaryWriter.add_graph 5 1 0
meth SummaryWriter._encode 2 0 0
meth SummaryWriter.add_embedding 7 1 0
meth SummaryWriter.add_pr_curve 8 1 0
meth SummaryWriter.add_pr_curve_raw 12 1 0
meth SummaryWriter.add_custom_scalars_multilinechart 4 1 0
meth SummaryWriter.add_custom_scalars_marginchart 4 1 0
meth SummaryWriter.add_custom_scalars 2 1 0
meth SummaryWriter.add_mesh 8 1 0
meth SummaryWriter.enter 1 0 0
meth SummaryWriter.exit 4 0 0
attr SummaryWriter.log_dir 1 0 0
attr SummaryWriter.purge_step 1 0 0
attr SummaryWriter.max_queue 1 0 0
attr SummaryWriter.flush_secs 1 0 0
attr SummaryWriter.filename_suffix 1 0 0
attr SummaryWriter.file_writer 1 0 0
attr SummaryWriter.all_writers 1 0 0
attr SummaryWriter.default_bins 1 0 0

torch.utils.throughput_benchmark (21 missing, 0 any)

Symbol Typable Typed Any
meth ExecutionStats.init 3 1 0
prop ExecutionStats.latency_avg_ms 1 0 0
prop ExecutionStats.num_iters 1 0 0
prop ExecutionStats.iters_per_second 1 0 0
prop ExecutionStats.total_time_seconds 1 0 0
attr ExecutionStats.benchmark_config 1 0 0
func format_time 4 1 0
meth ThroughputBenchmark.init 2 1 0
meth ThroughputBenchmark.run_once 3 0 0
meth ThroughputBenchmark.add_input 3 1 0
meth ThroughputBenchmark.benchmark 5 0 0

torch.utils.weak (46 missing, 0 any)

Symbol Typable Typed Any
meth WeakIdKeyDictionary.init 3 1 0
meth WeakIdKeyDictionary.delitem 2 1 0
meth WeakIdKeyDictionary.getitem 2 0 0
meth WeakIdKeyDictionary.setitem 3 1 0
meth WeakIdKeyDictionary.copy 1 0 0
meth WeakIdKeyDictionary.copy 1 0 0
meth WeakIdKeyDictionary.deepcopy 2 0 0
meth WeakIdKeyDictionary.get 3 0 0
meth WeakIdKeyDictionary.contains 2 1 0
meth WeakIdKeyDictionary.items 1 0 0
meth WeakIdKeyDictionary.keys 1 0 0
meth WeakIdKeyDictionary.iter 1 0 0
meth WeakIdKeyDictionary.values 1 0 0
meth WeakIdKeyDictionary.keyrefs 1 0 0
meth WeakIdKeyDictionary.popitem 1 0 0
meth WeakIdKeyDictionary.pop 3 0 0
meth WeakIdKeyDictionary.setdefault 3 0 0
meth WeakIdKeyDictionary.update 3 1 0
meth WeakIdKeyDictionary.ior 2 0 0
meth WeakIdKeyDictionary.or 2 0 0
meth WeakIdKeyDictionary.ror 2 0 0
meth WeakIdKeyDictionary.eq 2 0 0
attr WeakIdKeyDictionary.data 1 0 0
attr WeakIdKeyDictionary.ref_type 1 0 0
meth WeakIdRef.init 3 1 0
meth WeakIdRef.call 1 0 0
meth WeakIdRef.hash 1 0 0
meth WeakIdRef.eq 2 0 0
meth TensorWeakRef.call 1 0 0

torch.xpu (10 missing, 7 any)

Symbol Typable Typed Any
meth device.init 2 2 1
meth device.enter 1 0 0
meth device.exit 4 3 3
attr device.idx 1 0 0
attr device.prev_idx 1 0 0
meth device_of.init 2 1 0
func is_initialized 1 0 0
meth StreamContext.enter 1 0 0
meth StreamContext.exit 4 3 3
attr StreamContext.stream 1 0 0
attr StreamContext.idx 1 0 0

torch.xpu.memory (1 missing, 0 any)

Symbol Typable Typed Any
meth XPUPluggableAllocator.init 4 3 0

torch.xpu.streams (19 missing, 0 any)

Symbol Typable Typed Any
meth Event.new 2 0 0
meth Event.record 2 1 0
meth Event.wait 2 1 0
meth Event.elapsed_time 2 0 0
prop Event.as_parameter 1 0 0
meth Stream.new 4 0 0
meth Stream.wait_event 2 1 0
meth Stream.wait_stream 2 1 0
meth Stream.record_event 2 0 0
meth Stream.eq 2 0 0
meth Stream.hash 1 0 0
prop Stream.as_parameter 1 0 0

Type-Ignore Comments

Flavor Count
type: ignore[override] 132
type: ignore[assignment] 65
type: ignore[attr-defined] 26
type: ignore[misc] 21
type: ignore[return] 17
type: ignore[import] 16
type: ignore[no-redef] 12
pyrefly: ignore 4
type: ignore[assignment, misc] 4
type: ignore[has-type] 4
type: ignore[var-annotated] 4
codespell: ignore 3
type: ignore[type-var] 3
type: ignore[arg-type] 2
type: ignore[overload-overlap] 2
type: ignore[possibly-undefined] 2
type: ignore[type-arg] 2
type: ignore[import-not-found] 1
type: ignore[misc, name-defined, valid-type] 1
type: ignore[misc, no-redef, valid-type] 1
type: ignore[misc, return, type-var] 1
type: ignore[name-defined] 1
type: ignore[no-overload-impl] 1
type: ignore[no-untyped-def] 1
type: ignore[valid-type] 1