1{ 2 "being_migrated": { 3 "torch.nn.intrinsic": "torch.ao.nn.intrinsic", 4 "torch.nn.intrinsic.modules": "torch.ao.nn.intrinsic.modules", 5 "torch.nn.intrinsic.modules.fused": "torch.ao.nn.intrinsic.modules.fused", 6 "torch.nn.intrinsic.qat": "torch.ao.nn.intrinsic.qat", 7 "torch.nn.intrinsic.qat.modules": "torch.ao.nn.intrinsic.qat.modules", 8 "torch.nn.intrinsic.qat.modules.conv_fused": "torch.ao.nn.intrinsic.qat.modules.conv_fused", 9 "torch.nn.intrinsic.qat.modules.linear_fused": "torch.ao.nn.intrinsic.qat.modules.linear_fused", 10 "torch.nn.intrinsic.qat.modules.linear_relu": "torch.ao.nn.intrinsic.qat.modules.linear_relu", 11 "torch.nn.intrinsic.quantized": "torch.ao.nn.intrinsic.quantized", 12 "torch.nn.intrinsic.quantized.modules": "torch.ao.nn.intrinsic.quantized.modules", 13 "torch.nn.intrinsic.quantized.modules.bn_relu": "torch.ao.nn.intrinsic.quantized.modules.bn_relu", 14 "torch.nn.intrinsic.quantized.modules.conv_relu": "torch.ao.nn.intrinsic.quantized.modules.conv_relu", 15 "torch.nn.intrinsic.quantized.modules.linear_relu": "torch.ao.nn.intrinsic.quantized.modules.linear_relu", 16 "torch.nn.intrinsic.quantized.dynamic": "torch.ao.nn.intrinsic.quantized.dynamic", 17 "torch.nn.intrinsic.quantized.dynamic.modules": "torch.ao.nn.intrinsic.quantized.dynamic.modules", 18 "torch.nn.intrinsic.quantized.dynamic.modules.linear_relu": "torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu", 19 "torch.nn.qat": "torch.ao.nn.qat", 20 "torch.nn.qat.dynamic": "torch.ao.nn.qat.dynamic", 21 "torch.nn.qat.dynamic.modules": "torch.ao.nn.qat.dynamic.modules", 22 "torch.nn.qat.dynamic.modules.linear": "torch.ao.nn.qat.dynamic.modules.linear", 23 "torch.nn.qat.modules": "torch.ao.nn.qat.modules", 24 "torch.nn.qat.modules.conv": "torch.ao.nn.qat.modules.conv", 25 "torch.nn.qat.modules.embedding_ops": "torch.ao.nn.qat.modules.embedding_ops", 26 "torch.nn.qat.modules.linear": "torch.ao.nn.qat.modules.linear", 27 "torch.nn.quantized.functional": "torch.ao.nn.quantized.functional", 28 "torch.nn.quantized": "torch.ao.nn.quantized", 29 "torch.nn.quantized.modules": "torch.ao.nn.quantized.modules", 30 "torch.nn.quantized.dynamic": "torch.ao.nn.quantized.dynamic", 31 "torch.nn.quantized.dynamic.modules": "torch.ao.nn.quantized.dynamic.modules", 32 "torch.nn.quantized.dynamic.modules.rnn": "torch.ao.nn.quantized.dynamic.modules.rnn", 33 "torch.nn.quantizable": "torch.ao.nn.quantizable", 34 "torch.nn.quantizable.modules": "torch.ao.nn.quantizable.modules", 35 "torch.nn.quantizable.modules.activation": "torch.ao.nn.quantizable.modules.activation", 36 "torch.nn.quantizable.modules.rnn": "torch.ao.nn.quantizable.modules.rnn", 37 "torch.distributed.tensor.device_mesh": "torch.distributed.device_mesh" 38 }, 39 "torch.backends": [ 40 "contextmanager" 41 ], 42 "torch.cuda.comm": [ 43 "broadcast", 44 "broadcast_coalesced", 45 "reduce_add", 46 "reduce_add_coalesced", 47 "scatter", 48 "gather" 49 ], 50 "torch.csrc.jit.tensorexpr.scripts.bisect": [ 51 "bisect" 52 ], 53 "torch.cuda.nccl": [ 54 "init_rank", 55 "is_available", 56 "unique_id", 57 "version" 58 ], 59 "torch.distributed": [ 60 "AllToAllOptions", 61 "AllreduceCoalescedOptions", 62 "AllreduceOptions", 63 "BarrierOptions", 64 "BroadcastOptions", 65 "BuiltinCommHookType", 66 "Callable", 67 "DebugLevel", 68 "Dict", 69 "Enum", 70 "FileStore", 71 "GatherOptions", 72 "GradBucket", 73 "HashStore", 74 "Logger", 75 "namedtuple", 76 "Optional", 77 "PrefixStore", 78 "ProcessGroup", 79 "ProcessGroupGloo", 80 "ReduceOp", 81 "ReduceOptions", 82 "ReduceScatterOptions", 83 "Reducer", 84 "ScatterOptions", 85 "Store", 86 "TCPStore", 87 "Tuple", 88 "Union", 89 "get_debug_level", 90 "set_debug_level", 91 "set_debug_level_from_env", 92 "timedelta", 93 "ProcessGroupMPI", 94 "ProcessGroupNCCL" 95 ], 96 "torch.distributed.checkpoint.state_dict": [ 97 "Any", 98 "Callable", 99 "DDP", 100 "DTensor", 101 "Dict", 102 "DictValueType", 103 "FQNS_T", 104 "FSDP", 105 "FullOptimStateDictConfig", 106 "FullStateDictConfig", 107 "Iterable", 108 "List", 109 "ListDictValueType", 110 "OptimStateDictConfig", 111 "OptimizerStateType", 112 "Optional", 113 "PrimitiveType", 114 "Set", 115 "ShardedOptimStateDictConfig", 116 "ShardedStateDictConfig", 117 "ShardedTensor", 118 "StateDictConfig", 119 "StateDictType", 120 "Tuple", 121 "Union", 122 "ValueType", 123 "asdict", 124 "cast", 125 "chain", 126 "dataclass", 127 "field", 128 "no_type_check" 129 ], 130 "torch.distributed.autograd": [ 131 "DistAutogradContext", 132 "backward", 133 "get_gradients" 134 ], 135 "torch.distributed.elastic.events": [ 136 "Dict", 137 "Enum", 138 "EventMetadataValue", 139 "Optional" 140 ], 141 "torch.distributed.elastic.events.handlers": [ 142 "Dict", 143 "Optional", 144 "ScubaLogHandler", 145 "ScubaRdzvLogHandler" 146 ], 147 "torch.distributed.elastic.metrics": [ 148 "Optional", 149 "get_logger", 150 "TorchElasticService" 151 ], 152 "torch.distributed.elastic.multiprocessing": [ 153 "Callable", 154 "Dict", 155 "Tuple", 156 "Union", 157 "get_logger" 158 ], 159 "torch.distributed.elastic.multiprocessing.redirects": [ 160 "contextmanager", 161 "partial", 162 "redirect_stderr", 163 "redirect_stdout" 164 ], 165 "torch.distributed.elastic.rendezvous": [ 166 "RendezvousHandlerCreator" 167 ], 168 "torch.distributed.elastic.rendezvous.api": [ 169 "ABC", 170 "Any", 171 "Callable", 172 "Dict", 173 "Optional", 174 "RendezvousHandlerCreator", 175 "Store", 176 "Tuple", 177 "abstractmethod" 178 ], 179 "torch.distributed.elastic.rendezvous.dynamic_rendezvous": [ 180 "get_method_name" 181 ], 182 "torch.distributed.elastic.utils.api": [ 183 "Any", 184 "List", 185 "Template" 186 ], 187 "torch.distributed.elastic.utils.data.elastic_distributed_sampler": [ 188 "DistributedSampler" 189 ], 190 "torch.distributed.elastic.utils.logging": [ 191 "Optional", 192 "get_log_level" 193 ], 194 "torch.distributed.elastic.utils.store": [ 195 "List", 196 "timedelta" 197 ], 198 "torch.distributed.nn": [ 199 "Function", 200 "ReduceOp", 201 "group" 202 ], 203 "torch.distributed.nn.functional": [ 204 "Function", 205 "ReduceOp", 206 "group" 207 ], 208 "torch.distributed.nn.jit.instantiator": [ 209 "Optional", 210 "get_remote_module_template" 211 ], 212 "torch.distributed.optim.utils": [ 213 "Type" 214 ], 215 "torch.distributed.remote_device": [ 216 "Optional", 217 "Union" 218 ], 219 "torch.distributed.rendezvous": [ 220 "Dict", 221 "FileStore", 222 "Iterable", 223 "Optional", 224 "PrefixStore", 225 "Store", 226 "TCPStore", 227 "Tuple", 228 "Union", 229 "cast", 230 "timedelta", 231 "urlparse", 232 "urlunparse" 233 ], 234 "torch.distributed.rpc": [], 235 "torch.fft": [ 236 "Tensor", 237 "fft", 238 "fft2", 239 "fftfreq", 240 "fftn", 241 "fftshift", 242 "hfft", 243 "ifft", 244 "ifft2", 245 "ifftn", 246 "ifftshift", 247 "ihfft", 248 "irfft", 249 "irfft2", 250 "irfftn", 251 "rfft", 252 "rfft2", 253 "rfftfreq", 254 "rfftn" 255 ], 256 "torch.functional": [ 257 "istft", 258 "pca_lowrank", 259 "svd_lowrank" 260 ], 261 "torch.futures": [ 262 "Future" 263 ], 264 "torch.fx": [ 265 "ProxyableClassMeta", 266 "Tracer", 267 "symbolic_trace", 268 "wrap" 269 ], 270 "torch.fx.experimental.migrate_gradual_types.z3_types": [ 271 "D" 272 ], 273 "torch.fx.experimental.unification.core": [ 274 "Iterator", 275 "assoc", 276 "dispatch", 277 "isvar", 278 "partial", 279 "unify", 280 "walk" 281 ], 282 "torch.fx.experimental.unification.dispatch": [ 283 "dispatch", 284 "partial" 285 ], 286 "torch.fx.experimental.unification.more": [ 287 "dispatch", 288 "reify", 289 "unify" 290 ], 291 "torch.fx.experimental.unification.unification_tools": [ 292 "first", 293 "getter", 294 "groupby" 295 ], 296 "torch.fx.experimental.unification.variable": [ 297 "contextmanager", 298 "dispatch", 299 "hashable", 300 "isvar" 301 ], 302 "torch.fx.proxy": [ 303 "assert_fn" 304 ], 305 "torch.hub": [ 306 "HTTPError", 307 "Path", 308 "Request", 309 "tqdm", 310 "urlopen", 311 "urlparse" 312 ], 313 "torch.jit": [ 314 "Attribute", 315 "Final", 316 "Iterator", 317 "ONNXTracedModule", 318 "RecursiveScriptClass", 319 "RecursiveScriptModule", 320 "ScriptModule", 321 "ScriptWarning", 322 "TopLevelTracedModule", 323 "TracedModule", 324 "TracerWarning", 325 "TracingCheckError", 326 "contextmanager", 327 "export", 328 "fork", 329 "freeze", 330 "fuser", 331 "ignore", 332 "interface", 333 "is_scripting", 334 "is_tracing", 335 "jit_module_from_flatbuffer", 336 "last_executed_optimized_graph", 337 "load", 338 "optimize_for_inference", 339 "optimized_execution", 340 "run_frozen_optimizations", 341 "save", 342 "save_jit_module_to_flatbuffer", 343 "script", 344 "script_method", 345 "set_fusion_strategy", 346 "set_module", 347 "trace", 348 "trace_module", 349 "unused", 350 "wait" 351 ], 352 "torch.jit.annotations": [ 353 "Any", 354 "AnyType", 355 "ComplexType", 356 "Dict", 357 "DictType", 358 "EvalEnv", 359 "FloatType", 360 "IntType", 361 "List", 362 "ListType", 363 "StringType", 364 "TensorType", 365 "Tuple", 366 "TupleType", 367 "get_enum_value_type", 368 "is_dict", 369 "is_function_or_method", 370 "is_list", 371 "is_optional", 372 "is_tensor", 373 "is_tuple", 374 "is_union", 375 "is_vararg" 376 ], 377 "torch.jit.frontend": [ 378 "Apply", 379 "Assert", 380 "Assign", 381 "Attribute", 382 "AugAssign", 383 "BinOp", 384 "Break", 385 "ClassDef", 386 "Const", 387 "Continue", 388 "Decl", 389 "Def", 390 "Delete", 391 "DictComp", 392 "DictLiteral", 393 "Dots", 394 "EmptyTypeAnnotation", 395 "ExprStmt", 396 "FalseLiteral", 397 "For", 398 "FunctionModifiers", 399 "Ident", 400 "If", 401 "List", 402 "ListComp", 403 "ListLiteral", 404 "NoneLiteral", 405 "Param", 406 "Pass", 407 "Property", 408 "Raise", 409 "Return", 410 "Select", 411 "SliceExpr", 412 "Starred", 413 "Stmt", 414 "StringLiteral", 415 "Subscript", 416 "TernaryIf", 417 "TrueLiteral", 418 "Tuple", 419 "TupleLiteral", 420 "UnaryOp", 421 "Var", 422 "While", 423 "With", 424 "WithItem", 425 "dedent", 426 "get_qualified_name", 427 "get_source_lines_and_file", 428 "is_static_fn", 429 "make_source_context", 430 "namedtuple", 431 "parse_def", 432 "should_drop", 433 "monkeytype_trace" 434 ], 435 "torch.linalg": [ 436 "LinAlgError", 437 "Tensor", 438 "cholesky", 439 "cholesky_ex", 440 "cond", 441 "cross", 442 "det", 443 "diagonal", 444 "eig", 445 "eigh", 446 "eigvals", 447 "eigvalsh", 448 "householder_product", 449 "inv", 450 "inv_ex", 451 "ldl_factor", 452 "ldl_factor_ex", 453 "ldl_solve", 454 "lstsq", 455 "lu", 456 "lu_factor", 457 "lu_factor_ex", 458 "lu_solve", 459 "matmul", 460 "matrix_exp", 461 "matrix_norm", 462 "matrix_power", 463 "matrix_rank", 464 "multi_dot", 465 "norm", 466 "pinv", 467 "qr", 468 "slogdet", 469 "solve", 470 "solve_ex", 471 "solve_triangular", 472 "svd", 473 "svdvals", 474 "tensorinv", 475 "tensorsolve", 476 "vander", 477 "vecdot", 478 "vector_norm" 479 ], 480 "torch.masked": [ 481 "amax", 482 "amin", 483 "argmax", 484 "argmin", 485 "as_masked_tensor", 486 "cumprod", 487 "cumsum", 488 "is_masked_tensor", 489 "log_softmax", 490 "logaddexp", 491 "logsumexp", 492 "masked_tensor", 493 "MaskedTensor", 494 "mean", 495 "median", 496 "norm", 497 "normalize", 498 "prod", 499 "softmax", 500 "softmin", 501 "std", 502 "sum", 503 "var" 504 ], 505 "torch.multiprocessing": [ 506 "Array", 507 "AuthenticationError", 508 "Barrier", 509 "BoundedSemaphore", 510 "BufferTooShort", 511 "Condition", 512 "Event", 513 "JoinableQueue", 514 "Lock", 515 "Manager", 516 "Pipe", 517 "Pool", 518 "Process", 519 "ProcessContext", 520 "ProcessError", 521 "ProcessExitedException", 522 "ProcessRaisedException", 523 "Queue", 524 "RLock", 525 "RawArray", 526 "RawValue", 527 "Semaphore", 528 "SimpleQueue", 529 "SpawnContext", 530 "TimeoutError", 531 "Value", 532 "active_children", 533 "allow_connection_pickling", 534 "cpu_count", 535 "current_process", 536 "freeze_support", 537 "get_all_start_methods", 538 "get_context", 539 "get_logger", 540 "get_start_method", 541 "init_reductions", 542 "log_to_stderr", 543 "set_executable", 544 "set_forkserver_preload", 545 "set_start_method", 546 "spawn", 547 "start_processes", 548 "parent_process" 549 ], 550 "torch.multiprocessing.reductions": [ 551 "ForkingPickler", 552 "Union", 553 "check_serializing_named_tensor", 554 "register_after_fork" 555 ], 556 "torch.multiprocessing.spawn": [ 557 "Optional" 558 ], 559 "torch.nested": [ 560 "nested_tensor", 561 "to_padded_tensor" 562 ], 563 "torch.nn.common_types": [ 564 "Optional", 565 "Tensor", 566 "Tuple", 567 "TypeVar", 568 "Union" 569 ], 570 "torch.nn.functional": [ 571 "Callable", 572 "DType", 573 "List", 574 "Optional", 575 "Tensor", 576 "Tuple", 577 "Union", 578 "adaptive_avg_pool1d", 579 "avg_pool1d", 580 "avg_pool2d", 581 "avg_pool3d", 582 "bilinear", 583 "boolean_dispatch", 584 "celu_", 585 "channel_shuffle", 586 "conv1d", 587 "conv2d", 588 "conv3d", 589 "conv_tbc", 590 "conv_transpose1d", 591 "conv_transpose2d", 592 "conv_transpose3d", 593 "cosine_similarity", 594 "elu_", 595 "gelu", 596 "handle_torch_function", 597 "hardshrink", 598 "hardtanh_", 599 "has_torch_function", 600 "has_torch_function_unary", 601 "has_torch_function_variadic", 602 "leaky_relu_", 603 "linear", 604 "logsigmoid", 605 "native_channel_shuffle", 606 "one_hot", 607 "pairwise_distance", 608 "pdist", 609 "pixel_shuffle", 610 "pixel_unshuffle", 611 "prelu", 612 "relu_", 613 "rrelu_", 614 "scaled_dot_product_attention", 615 "selu_", 616 "softplus", 617 "softshrink", 618 "threshold_" 619 ], 620 "torch.nn.init": [ 621 "Tensor" 622 ], 623 "torch.nn.intrinsic.modules": [ 624 "_FusedModule" 625 ], 626 "torch.nn.modules.linear": [ 627 "NonDynamicallyQuantizableLinear" 628 ], 629 "torch.nn.modules.rnn": [ 630 "apply_permutation" 631 ], 632 "torch.nn.parallel": [ 633 "DistributedDataParallelCPU" 634 ], 635 "torch.nn.parallel.comm": [ 636 "List" 637 ], 638 "torch.nn.parallel.parallel_apply": [ 639 "ExceptionWrapper", 640 "autocast" 641 ], 642 "torch.nn.parallel.replicate": [ 643 "OrderedDict" 644 ], 645 "torch.nn.parallel.scatter_gather": [ 646 "is_namedtuple" 647 ], 648 "torch.nn.parameter": [ 649 "OrderedDict" 650 ], 651 "torch.nn.utils.rnn": [ 652 "bind", 653 "PackedSequence_" 654 ], 655 "torch.nn.utils.convert_parameters": [ 656 "Iterable", 657 "Optional" 658 ], 659 "torch.onnx": [ 660 "Dict", 661 "OperatorExportTypes", 662 "Optional", 663 "TensorProtoDataType", 664 "TrainingMode" 665 ], 666 "torch.overrides": [ 667 "BaseTorchFunctionMode", 668 "TorchFunctionMode", 669 "TorchFunctionModeMeta", 670 "enable_torch_function_mode", 671 "get_default_nowrap_functions", 672 "has_torch_function" 673 ], 674 "torch.package.analyze.is_from_package": [ 675 "Any", 676 "ModuleType", 677 "is_mangled" 678 ], 679 "torch.package.find_file_dependencies": [ 680 "List", 681 "Optional", 682 "Tuple" 683 ], 684 "torch.package.glob_group": [ 685 "GlobPattern", 686 "Iterable", 687 "Union" 688 ], 689 "torch.profiler": [ 690 "DeviceType", 691 "ProfilerActivity", 692 "kineto_available", 693 "record_function" 694 ], 695 "torch.quantization": [ 696 "ABC", 697 "DeQuantStub", 698 "FakeQuantize", 699 "FakeQuantizeBase", 700 "FixedQParamsFakeQuantize", 701 "FusedMovingAvgObsFakeQuantize", 702 "HistogramObserver", 703 "MinMaxObserver", 704 "MovingAverageMinMaxObserver", 705 "MovingAveragePerChannelMinMaxObserver", 706 "NoopObserver", 707 "ObserverBase", 708 "PerChannelMinMaxObserver", 709 "PlaceholderObserver", 710 "QConfig", 711 "QConfigAny", 712 "QConfigDynamic", 713 "QuantStub", 714 "QuantType", 715 "QuantWrapper", 716 "RecordingObserver", 717 "_add_module_to_qconfig_obs_ctr", 718 "add_quant_dequant", 719 "_assert_valid_qconfig", 720 "convert", 721 "convert_dynamic_jit", 722 "convert_jit", 723 "default_fixed_qparams_range_0to1_fake_quant", 724 "default_affine_fixed_qparams_fake_quant", 725 "default_debug_observer", 726 "default_dynamic_quant_observer", 727 "default_fake_quant", 728 "default_float_qparams_observer", 729 "default_fused_act_fake_quant", 730 "default_fused_per_channel_wt_fake_quant", 731 "default_fused_wt_fake_quant", 732 "default_histogram_fake_quant", 733 "default_histogram_observer", 734 "default_observer", 735 "default_per_channel_weight_fake_quant", 736 "default_per_channel_weight_observer", 737 "default_placeholder_observer", 738 "default_fixed_qparams_range_neg1to1_fake_quant", 739 "default_symmetric_fixed_qparams_fake_quant", 740 "default_weight_fake_quant", 741 "default_weight_observer", 742 "disable_fake_quant", 743 "disable_observer", 744 "enable_fake_quant", 745 "enable_observer", 746 "fuse_conv_bn", 747 "fuse_conv_bn_jit", 748 "fuse_conv_bn_relu", 749 "fuse_linear_bn", 750 "fuse_modules", 751 "get_default_compare_output_module_list", 752 "get_default_dynamic_quant_module_mappings", 753 "get_default_float_to_quantized_operator_mappings", 754 "get_default_qat_module_mappings", 755 "get_default_qat_qconfig", 756 "get_default_qconfig", 757 "get_default_qconfig_propagation_list", 758 "get_default_static_quant_module_mappings", 759 "get_dynamic_quant_module_class", 760 "get_fuser_method", 761 "get_observer_state_dict", 762 "get_quantized_operator", 763 "get_static_quant_module_class", 764 "load_observer_state_dict", 765 "no_observer_set", 766 "prepare", 767 "prepare_dynamic_jit", 768 "prepare_jit", 769 "prepare_qat", 770 "propagate_qconfig_", 771 "qconfig_equals", 772 "_get_quant_type_to_str", 773 "quantize", 774 "quantize_dynamic", 775 "quantize_dynamic_jit", 776 "quantize_jit", 777 "quantize_qat", 778 "script_qconfig", 779 "script_qconfig_dict", 780 "swap_module" 781 ], 782 "torch.quantization.fake_quantize": [ 783 "FakeQuantize", 784 "FakeQuantizeBase", 785 "FixedQParamsFakeQuantize", 786 "FusedMovingAvgObsFakeQuantize", 787 "default_fixed_qparams_range_0to1_fake_quant", 788 "default_affine_fixed_qparams_fake_quant", 789 "default_fake_quant", 790 "default_fused_act_fake_quant", 791 "default_fused_per_channel_wt_fake_quant", 792 "default_fused_wt_fake_quant", 793 "default_histogram_fake_quant", 794 "default_per_channel_weight_fake_quant", 795 "default_fixed_qparams_range_neg1to1_fake_quant", 796 "default_symmetric_fixed_qparams_fake_quant", 797 "default_weight_fake_quant", 798 "disable_fake_quant", 799 "disable_observer", 800 "enable_fake_quant", 801 "enable_observer" 802 ], 803 "torch.quantization.fuse_modules": [ 804 "fuse_conv_bn", 805 "fuse_conv_bn_relu", 806 "fuse_known_modules", 807 "fuse_modules", 808 "get_fuser_method" 809 ], 810 "torch.quantization.fuser_method_mappings": [ 811 "fuse_conv_bn", 812 "fuse_conv_bn_relu", 813 "fuse_linear_bn", 814 "get_fuser_method" 815 ], 816 "torch.quantization.observer": [ 817 "ABC", 818 "HistogramObserver", 819 "MinMaxObserver", 820 "MovingAverageMinMaxObserver", 821 "MovingAveragePerChannelMinMaxObserver", 822 "NoopObserver", 823 "ObserverBase", 824 "PerChannelMinMaxObserver", 825 "PlaceholderObserver", 826 "RecordingObserver", 827 "default_debug_observer", 828 "default_dynamic_quant_observer", 829 "default_float_qparams_observer", 830 "default_histogram_observer", 831 "default_observer", 832 "default_per_channel_weight_observer", 833 "default_placeholder_observer", 834 "default_weight_observer", 835 "get_observer_state_dict", 836 "load_observer_state_dict" 837 ], 838 "torch.quantization.qconfig": [ 839 "QConfig", 840 "QConfigAny", 841 "QConfigDynamic", 842 "_add_module_to_qconfig_obs_ctr", 843 "_assert_valid_qconfig", 844 "get_default_qat_qconfig", 845 "get_default_qconfig", 846 "qconfig_equals" 847 ], 848 "torch.quantization.quant_type": [ 849 "QuantType", 850 "_get_quant_type_to_str" 851 ], 852 "torch.quantization.quantization_mappings": [ 853 "get_default_compare_output_module_list", 854 "get_default_dynamic_quant_module_mappings", 855 "get_default_float_to_quantized_operator_mappings", 856 "get_default_qat_module_mappings", 857 "get_default_qconfig_propagation_list", 858 "get_default_static_quant_module_mappings", 859 "get_dynamic_quant_module_class", 860 "get_quantized_operator", 861 "get_static_quant_module_class", 862 "no_observer_set" 863 ], 864 "torch.quantization.quantize": [ 865 "add_quant_dequant", 866 "convert", 867 "prepare", 868 "prepare_qat", 869 "propagate_qconfig_", 870 "quantize", 871 "quantize_dynamic", 872 "quantize_qat", 873 "swap_module" 874 ], 875 "torch.quantization.quantize_jit": [ 876 "convert_dynamic_jit", 877 "convert_jit", 878 "fuse_conv_bn_jit", 879 "prepare_dynamic_jit", 880 "prepare_jit", 881 "quantize_dynamic_jit", 882 "quantize_jit", 883 "script_qconfig", 884 "script_qconfig_dict" 885 ], 886 "torch.quantization.stubs": [ 887 "DeQuantStub", 888 "QuantStub", 889 "QuantWrapper" 890 ], 891 "torch.quasirandom": [ 892 "Optional" 893 ], 894 "torch.random": [ 895 "Generator" 896 ], 897 "torch.serialization": [ 898 "Any", 899 "BinaryIO", 900 "Dict", 901 "IO", 902 "Optional", 903 "Storage", 904 "Tuple", 905 "Type", 906 "Union", 907 "cast", 908 "closing", 909 "contextmanager", 910 "get_source_lines_and_file", 911 "get_default_mmap_options", 912 "set_default_mmap_options" 913 ], 914 "torch.sparse": [ 915 "BFloat16Tensor", 916 "ByteTensor", 917 "CharTensor", 918 "DoubleTensor", 919 "FloatTensor", 920 "HalfTensor", 921 "IntTensor", 922 "LongTensor", 923 "ShortTensor", 924 "addmm", 925 "log_softmax", 926 "mm", 927 "softmax" 928 ], 929 "torch.special": [ 930 "airy_ai", 931 "bessel_j0", 932 "bessel_j1", 933 "bessel_y0", 934 "bessel_y1", 935 "chebyshev_polynomial_t", 936 "chebyshev_polynomial_u", 937 "chebyshev_polynomial_v", 938 "chebyshev_polynomial_w", 939 "digamma", 940 "entr", 941 "erf", 942 "erfc", 943 "erfcx", 944 "erfinv", 945 "exp2", 946 "expit", 947 "expm1", 948 "gammainc", 949 "gammaincc", 950 "gammaln", 951 "hermite_polynomial_h", 952 "hermite_polynomial_he", 953 "i0", 954 "i0e", 955 "i1", 956 "i1e", 957 "laguerre_polynomial_l", 958 "legendre_polynomial_p", 959 "log1p", 960 "log_ndtr", 961 "log_softmax", 962 "logit", 963 "logsumexp", 964 "modified_bessel_i0", 965 "modified_bessel_i1", 966 "modified_bessel_k0", 967 "modified_bessel_k1", 968 "multigammaln", 969 "ndtr", 970 "ndtri", 971 "polygamma", 972 "psi", 973 "round", 974 "scaled_modified_bessel_k0", 975 "scaled_modified_bessel_k1", 976 "shifted_chebyshev_polynomial_t", 977 "shifted_chebyshev_polynomial_u", 978 "shifted_chebyshev_polynomial_v", 979 "shifted_chebyshev_polynomial_w", 980 "sinc", 981 "softmax", 982 "spherical_bessel_j0", 983 "xlog1py", 984 "xlogy", 985 "zeta" 986 ], 987 "torch.storage": [ 988 "Any", 989 "Storage", 990 "Type", 991 "TypeVar", 992 "Union", 993 "cast", 994 "lru_cache" 995 ], 996 "torch.testing": [ 997 "FileCheck", 998 "all_types", 999 "all_types_and", 1000 "all_types_and_complex", 1001 "all_types_and_complex_and", 1002 "all_types_and_half", 1003 "assert_allclose", 1004 "assert_close", 1005 "complex_types", 1006 "double_types", 1007 "empty_types", 1008 "floating_and_complex_types", 1009 "floating_and_complex_types_and", 1010 "floating_types", 1011 "floating_types_and", 1012 "floating_types_and_half", 1013 "get_all_complex_dtypes", 1014 "get_all_device_types", 1015 "get_all_dtypes", 1016 "get_all_fp_dtypes", 1017 "get_all_int_dtypes", 1018 "get_all_math_dtypes", 1019 "integral_types", 1020 "integral_types_and", 1021 "make_non_contiguous", 1022 "make_tensor", 1023 "rand", 1024 "randn" 1025 ], 1026 "torch.types": [ 1027 "Any", 1028 "Device", 1029 "List", 1030 "Number", 1031 "Sequence", 1032 "Tuple", 1033 "Union" 1034 ], 1035 "torch.utils.benchmark.utils.compare": [ 1036 "Colorize", 1037 "Table", 1038 "optional_min" 1039 ], 1040 "torch.utils.benchmark.utils.cpp_jit": [ 1041 "Any", 1042 "CallgrindModuleType", 1043 "List", 1044 "Optional", 1045 "TimeitModuleType" 1046 ], 1047 "torch.utils.benchmark.utils.fuzzer": [ 1048 "dtype_size", 1049 "prod" 1050 ], 1051 "torch.utils.benchmark.utils.sparse_fuzzer": [ 1052 "FuzzedTensor", 1053 "Number", 1054 "Optional", 1055 "Tuple", 1056 "Union" 1057 ], 1058 "torch.utils.benchmark.utils.timer": [ 1059 "CPPTimer", 1060 "timer" 1061 ], 1062 "torch.utils.benchmark.utils.valgrind_wrapper.timer_interface": [ 1063 "GlobalsBridge", 1064 "Serialization", 1065 "wrapper_singleton" 1066 ], 1067 "torch.utils.data": [ 1068 "_DatasetKind", 1069 "argument_validation", 1070 "default_collate", 1071 "default_convert", 1072 "functional_datapipe", 1073 "get_worker_info", 1074 "guaranteed_datapipes_determinism", 1075 "non_deterministic", 1076 "runtime_validation", 1077 "runtime_validation_disabled" 1078 ], 1079 "torch.utils.data.dataloader": [ 1080 "default_collate", 1081 "default_convert", 1082 "get_worker_info" 1083 ], 1084 "torch.utils.data.datapipes.dataframe": [ 1085 "DFIterDataPipe" 1086 ], 1087 "torch.utils.dlpack": [ 1088 "Any", 1089 "to_dlpack" 1090 ], 1091 "torch": [ 1092 "BFloat16Storage", 1093 "BFloat16Tensor", 1094 "ComplexDoubleStorage", 1095 "ComplexFloatStorage", 1096 "DisableTorchFunction", 1097 "DisableTorchFunctionSubclass", 1098 "Generator", 1099 "HalfStorage", 1100 "HalfTensor", 1101 "QInt32Storage", 1102 "QInt8Storage", 1103 "QUInt2x4Storage", 1104 "QUInt4x2Storage", 1105 "QUInt8Storage", 1106 "Storage", 1107 "TypedStorage", 1108 "_adaptive_avg_pool2d", 1109 "_adaptive_avg_pool3d", 1110 "_add_batch_dim", 1111 "_add_relu", 1112 "_add_relu_", 1113 "_addmm_activation", 1114 "_aminmax", 1115 "_amp_foreach_non_finite_check_and_unscale_", 1116 "_amp_update_scale_", 1117 "_assert_async", 1118 "_batch_norm_impl_index", 1119 "_cast_Byte", 1120 "_cast_Char", 1121 "_cast_Double", 1122 "_cast_Float", 1123 "_cast_Half", 1124 "_cast_Int", 1125 "_cast_Long", 1126 "_cast_Short", 1127 "_choose_qparams_per_tensor", 1128 "_coalesce", 1129 "_compute_linear_combination", 1130 "_conj", 1131 "_conj_copy", 1132 "_conj_physical", 1133 "_convert_indices_from_coo_to_csr", 1134 "_convert_indices_from_csr_to_coo", 1135 "_convolution", 1136 "_convolution_mode", 1137 "_copy_from", 1138 "_copy_from_and_resize", 1139 "_ctc_loss", 1140 "_cudnn_ctc_loss", 1141 "_cudnn_init_dropout_state", 1142 "_cudnn_rnn", 1143 "_cudnn_rnn_flatten_weight", 1144 "_cufft_clear_plan_cache", 1145 "_cufft_get_plan_cache_max_size", 1146 "_cufft_get_plan_cache_size", 1147 "_cufft_set_plan_cache_max_size", 1148 "_cummax_helper", 1149 "_cummin_helper", 1150 "_debug_has_internal_overlap", 1151 "_det_lu_based_helper_backward_helper", 1152 "_dim_arange", 1153 "_dirichlet_grad", 1154 "_disable_functionalization", 1155 "_efficientzerotensor", 1156 "_embedding_bag", 1157 "_embedding_bag_forward_only", 1158 "_empty_affine_quantized", 1159 "_empty_per_channel_affine_quantized", 1160 "_enable_functionalization", 1161 "_euclidean_dist", 1162 "_fake_quantize_learnable_per_channel_affine", 1163 "_fake_quantize_learnable_per_tensor_affine", 1164 "_fake_quantize_per_tensor_affine_cachemask_tensor_qparams", 1165 "_fft_c2c", 1166 "_fft_c2r", 1167 "_fft_r2c", 1168 "_foreach_abs", 1169 "_foreach_abs_", 1170 "_foreach_acos", 1171 "_foreach_acos_", 1172 "_foreach_add", 1173 "_foreach_add_", 1174 "_foreach_addcdiv", 1175 "_foreach_addcdiv_", 1176 "_foreach_addcmul", 1177 "_foreach_addcmul_", 1178 "_foreach_asin", 1179 "_foreach_asin_", 1180 "_foreach_atan", 1181 "_foreach_atan_", 1182 "_foreach_ceil", 1183 "_foreach_ceil_", 1184 "_foreach_cos", 1185 "_foreach_cos_", 1186 "_foreach_cosh", 1187 "_foreach_cosh_", 1188 "_foreach_div", 1189 "_foreach_div_", 1190 "_foreach_erf", 1191 "_foreach_erf_", 1192 "_foreach_erfc", 1193 "_foreach_erfc_", 1194 "_foreach_exp", 1195 "_foreach_exp_", 1196 "_foreach_expm1", 1197 "_foreach_expm1_", 1198 "_foreach_floor", 1199 "_foreach_floor_", 1200 "_foreach_frac", 1201 "_foreach_frac_", 1202 "_foreach_lgamma", 1203 "_foreach_lgamma_", 1204 "_foreach_log", 1205 "_foreach_log10", 1206 "_foreach_log10_", 1207 "_foreach_log1p", 1208 "_foreach_log1p_", 1209 "_foreach_log2", 1210 "_foreach_log2_", 1211 "_foreach_log_", 1212 "_foreach_maximum", 1213 "_foreach_minimum", 1214 "_foreach_mul", 1215 "_foreach_mul_", 1216 "_foreach_neg", 1217 "_foreach_neg_", 1218 "_foreach_norm", 1219 "_foreach_reciprocal", 1220 "_foreach_reciprocal_", 1221 "_foreach_round", 1222 "_foreach_round_", 1223 "_foreach_sigmoid", 1224 "_foreach_sigmoid_", 1225 "_foreach_sign", 1226 "_foreach_sign_", 1227 "_foreach_sin", 1228 "_foreach_sin_", 1229 "_foreach_sinh", 1230 "_foreach_sinh_", 1231 "_foreach_sqrt", 1232 "_foreach_sqrt_", 1233 "_foreach_sub", 1234 "_foreach_sub_", 1235 "_foreach_tan", 1236 "_foreach_tan_", 1237 "_foreach_tanh", 1238 "_foreach_tanh_", 1239 "_foreach_trunc", 1240 "_foreach_trunc_", 1241 "_foreach_zero_", 1242 "_from_functional_tensor", 1243 "_fused_dropout", 1244 "_fused_moving_avg_obs_fq_helper", 1245 "_fw_primal_copy", 1246 "_grid_sampler_2d_cpu_fallback", 1247 "_has_compatible_shallow_copy_type", 1248 "_histogramdd_bin_edges", 1249 "_histogramdd_from_bin_cts", 1250 "_histogramdd_from_bin_tensors", 1251 "_index_put_impl_", 1252 "_indices_copy", 1253 "_is_functional_tensor", 1254 "_is_zerotensor", 1255 "_linalg_check_errors", 1256 "_linalg_qr_helper", 1257 "_linalg_svd", 1258 "_linalg_solve_ex", 1259 "_log_softmax", 1260 "_log_softmax_backward_data", 1261 "_logcumsumexp", 1262 "_lu_with_info", 1263 "_make_dual", 1264 "_make_dual_copy", 1265 "_make_per_channel_quantized_tensor", 1266 "_make_per_tensor_quantized_tensor", 1267 "_masked_scale", 1268 "_masked_softmax", 1269 "_mkldnn_reshape", 1270 "_mkldnn_transpose", 1271 "_mkldnn_transpose_", 1272 "_neg_view", 1273 "_neg_view_copy", 1274 "_nested_from_padded", 1275 "_nested_from_padded_and_nested_example", 1276 "_nnpack_available", 1277 "_nnpack_spatial_convolution", 1278 "_pack_padded_sequence", 1279 "_pad_packed_sequence", 1280 "_pin_memory", 1281 "_remove_batch_dim", 1282 "_reshape_alias_copy", 1283 "_reshape_from_tensor", 1284 "_rowwise_prune", 1285 "_sample_dirichlet", 1286 "_saturate_weight_to_fp16", 1287 "_shape_as_tensor", 1288 "_sobol_engine_draw", 1289 "_sobol_engine_ff_", 1290 "_sobol_engine_initialize_state_", 1291 "_sobol_engine_scramble_", 1292 "_softmax", 1293 "_softmax_backward_data", 1294 "_sparse_broadcast_to", 1295 "_sparse_broadcast_to_copy", 1296 "_sparse_coo_tensor_unsafe", 1297 "_sparse_csr_prod", 1298 "_sparse_csr_sum", 1299 "_sparse_csr_tensor_unsafe", 1300 "_sparse_log_softmax_backward_data", 1301 "_sparse_softmax_backward_data", 1302 "_sparse_sparse_matmul", 1303 "_sparse_sum", 1304 "_stack", 1305 "_standard_gamma", 1306 "_standard_gamma_grad", 1307 "_sync", 1308 "_test_serialization_subcmul", 1309 "_to_cpu", 1310 "_to_functional_tensor", 1311 "_torch_cuda_cu_linker_symbol_op", 1312 "_trilinear", 1313 "_unique", 1314 "_unique2", 1315 "_unpack_dual", 1316 "_use_cudnn_ctc_loss", 1317 "_use_cudnn_rnn_flatten_weight", 1318 "_validate_sparse_compressed_tensor_args", 1319 "_validate_sparse_coo_tensor_args", 1320 "_validate_sparse_csr_tensor_args", 1321 "_values_copy", 1322 "_weight_norm", 1323 "_weight_norm_interface", 1324 "autocast", 1325 "broadcast_shapes", 1326 "compiled_with_cxx11_abi", 1327 "from_dlpack", 1328 "lobpcg", 1329 "lu", 1330 "segment_reduce", 1331 "set_default_dtype", 1332 "set_grad_enabled", 1333 "set_printoptions", 1334 "unique" 1335 ], 1336 "torch.ao.ns.fx.graph_matcher": [ 1337 "Any", 1338 "Dict", 1339 "FakeQuantizeBase", 1340 "Graph", 1341 "GraphModule", 1342 "List", 1343 "NSNodeTargetType", 1344 "NSSubgraph", 1345 "Node", 1346 "ObserverBase", 1347 "Optional", 1348 "Set", 1349 "Tuple", 1350 "end_node_matches_reversed_fusion", 1351 "get_base_name_to_sets_of_related_ops", 1352 "get_reversed_fusions", 1353 "get_type_a_related_to_b", 1354 "get_unmatchable_types_map", 1355 "getattr_from_fqn" 1356 ], 1357 "torch.ao.ns.fx.graph_passes": [ 1358 "Any", 1359 "Callable", 1360 "Dict", 1361 "Graph", 1362 "GraphModule", 1363 "List", 1364 "NSNodeTargetType", 1365 "NSSingleResultValuesType", 1366 "NSSubgraph", 1367 "Node", 1368 "NodeInputOrOutputType", 1369 "Optional", 1370 "Set", 1371 "Tuple", 1372 "Union", 1373 "get_arg_indices_of_inputs_to_log", 1374 "get_new_attr_name_with_prefix", 1375 "get_node_first_input_and_output_type", 1376 "get_node_input_qparams", 1377 "get_node_type_to_io_type_map", 1378 "get_normalized_nth_input", 1379 "get_number_of_non_param_args", 1380 "get_target_type_str", 1381 "getattr_from_fqn", 1382 "map_arg", 1383 "op_type_supports_shadowing", 1384 "return_first_non_observer_node" 1385 ], 1386 "torch.ao.ns.fx.mappings": [ 1387 "Callable", 1388 "Dict", 1389 "List", 1390 "NSNodeTargetType", 1391 "Optional", 1392 "Set", 1393 "Tuple", 1394 "get_native_backend_config" 1395 ], 1396 "torch.ao.ns.fx.n_shadows_utils": [ 1397 "Any", 1398 "Callable", 1399 "Dict", 1400 "Graph", 1401 "GraphModule", 1402 "List", 1403 "NSResultsType", 1404 "NSSingleResultValuesType", 1405 "Node", 1406 "Optional", 1407 "QConfigAny", 1408 "QConfigMapping", 1409 "Set", 1410 "Tuple", 1411 "get_normalized_nth_input", 1412 "get_target_type_str", 1413 "getattr_from_fqn", 1414 "tree_map" 1415 ], 1416 "torch.ao.ns.fx.ns_types": [ 1417 "Any", 1418 "Callable", 1419 "Dict", 1420 "List", 1421 "NSNodeTargetType", 1422 "NSResultsType", 1423 "NSSingleResultType", 1424 "NamedTuple", 1425 "Node", 1426 "Union" 1427 ], 1428 "torch.ao.ns.fx.pattern_utils": [ 1429 "Any", 1430 "Callable", 1431 "Dict", 1432 "FakeQuantizeBase", 1433 "GraphModule", 1434 "List", 1435 "NSFusionElType", 1436 "NSFusionType", 1437 "NSNodeTargetType", 1438 "Node", 1439 "ObserverBase", 1440 "Set", 1441 "Tuple", 1442 "Union", 1443 "get_native_backend_config", 1444 "getattr_from_fqn" 1445 ], 1446 "torch.ao.ns.fx.utils": [ 1447 "Callable", 1448 "Dict", 1449 "FakeQuantizeBase", 1450 "GraphModule", 1451 "List", 1452 "NSNodeTargetType", 1453 "NSResultsType", 1454 "Node", 1455 "ObserverBase", 1456 "Optional", 1457 "Set", 1458 "Tuple", 1459 "Union", 1460 "getattr_from_fqn" 1461 ], 1462 "torch.ao.ns.fx.weight_utils": [ 1463 "Callable", 1464 "Dict", 1465 "GraphModule", 1466 "List", 1467 "NSSingleResultType", 1468 "NSSingleResultValuesType", 1469 "Node", 1470 "Optional", 1471 "get_target_type_str", 1472 "getattr_from_fqn", 1473 "return_first_non_observer_node" 1474 ], 1475 "torch.ao.pruning": [ 1476 "get_dynamic_sparse_quantized_mapping", 1477 "get_static_sparse_quantized_mapping" 1478 ], 1479 "torch.ao.quantization.fx.lstm_utils": [ 1480 "Any", 1481 "BackendConfig", 1482 "Callable", 1483 "FakeQuantizeBase", 1484 "Optional", 1485 "QConfig", 1486 "QConfigMapping", 1487 "Tuple", 1488 "convert_to_reference_fx", 1489 "default_weight_fake_quant", 1490 "default_weight_observer", 1491 "prepare_fx" 1492 ], 1493 "torch.ao.quantization.fx.tracer": [ 1494 "ScopeContextManager" 1495 ], 1496 "torch.ao.quantization.pt2e.prepare": [ 1497 "Any", 1498 "Argument", 1499 "Dict", 1500 "EdgeOrNode", 1501 "FakeTensor", 1502 "GraphModule", 1503 "Node", 1504 "ObserverOrFakeQuantize", 1505 "PrepareCustomConfig", 1506 "QConfigAny", 1507 "QConfigMapping", 1508 "QuantizationAnnotation", 1509 "SharedQuantizationSpec", 1510 "Tuple", 1511 "Union" 1512 ], 1513 "torch.ao.quantization.pt2e.qat_utils": [ 1514 "Any", 1515 "Callable", 1516 "DerivedQuantizationSpec", 1517 "Dict", 1518 "EdgeOrNode", 1519 "Graph", 1520 "GraphModule", 1521 "List", 1522 "Node", 1523 "QuantizationSpecBase", 1524 "SharedQuantizationSpec", 1525 "Tuple", 1526 "fold_bn_weights_into_conv_node", 1527 "replace_pattern_with_filters" 1528 ], 1529 "torch.ao.quantization.quantize_fx": [ 1530 "Any", 1531 "BackendConfig", 1532 "ConvertCustomConfig", 1533 "Dict", 1534 "FuseCustomConfig", 1535 "GraphModule", 1536 "ObservedGraphModule", 1537 "Optional", 1538 "PrepareCustomConfig", 1539 "QConfigMapping", 1540 "QuantizationTracer", 1541 "Scope", 1542 "ScopeContextManager", 1543 "Tuple", 1544 "Union", 1545 "convert", 1546 "fuse", 1547 "get_custom_module_class_keys", 1548 "get_skipped_module_name_and_classes", 1549 "get_tensorrt_backend_config", 1550 "prepare" 1551 ], 1552 "torch.ao.quantization.quantizer.utils": [ 1553 "List", 1554 "Node", 1555 "QuantizationAnnotation" 1556 ], 1557 "torch.ao.quantization.quantizer.xnnpack_quantizer": [ 1558 "OperatorConfig", 1559 "OperatorPatternType", 1560 "QuantizationConfig", 1561 "propagate_annotation" 1562 ], 1563 "torch.ao.quantization.quantizer.xnnpack_quantizer_utils": [ 1564 "register_annotator" 1565 ], 1566 "torch.backends.xeon.run_cpu": [ 1567 "ArgumentParser", 1568 "Dict", 1569 "List", 1570 "RawTextHelpFormatter", 1571 "Std", 1572 "expanduser", 1573 "start_processes" 1574 ], 1575 "torch.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks": [ 1576 "Any", 1577 "Variable", 1578 "dataclass", 1579 "no_type_check" 1580 ], 1581 "torch.distributed.algorithms.model_averaging.hierarchical_model_averager": [ 1582 "Dict", 1583 "Iterable", 1584 "OrderedDict", 1585 "Union" 1586 ], 1587 "torch.distributed.argparse_util": [ 1588 "Action" 1589 ], 1590 "torch.distributed.collective_utils": [ 1591 "Any", 1592 "Callable", 1593 "Generic", 1594 "List", 1595 "Optional", 1596 "Tuple", 1597 "TypeVar", 1598 "Union", 1599 "cast", 1600 "dataclass" 1601 ], 1602 "torch.distributed.elastic.rendezvous.c10d_rendezvous_backend": [ 1603 "Any", 1604 "FileStore", 1605 "NodeState", 1606 "Optional", 1607 "RendezvousBackend", 1608 "RendezvousConnectionError", 1609 "RendezvousError", 1610 "RendezvousParameters", 1611 "RendezvousStateError", 1612 "Store", 1613 "TCPStore", 1614 "Token", 1615 "Tuple", 1616 "b64decode", 1617 "b64encode", 1618 "cast", 1619 "construct_and_record_rdzv_event", 1620 "parse_rendezvous_endpoint", 1621 "timedelta" 1622 ], 1623 "torch.distributed.elastic.rendezvous.etcd_rendezvous": [ 1624 "EtcdStore", 1625 "Optional", 1626 "RendezvousClosedError", 1627 "RendezvousError", 1628 "RendezvousHandler", 1629 "RendezvousParameters", 1630 "RendezvousTimeoutError", 1631 "cas_delay", 1632 "parse_rendezvous_endpoint" 1633 ], 1634 "torch.distributed.elastic.rendezvous.etcd_rendezvous_backend": [ 1635 "EtcdAlreadyExist", 1636 "EtcdClient", 1637 "EtcdCompareFailed", 1638 "EtcdException", 1639 "EtcdKeyNotFound", 1640 "EtcdResult", 1641 "EtcdStore", 1642 "Optional", 1643 "RendezvousBackend", 1644 "RendezvousConnectionError", 1645 "RendezvousParameters", 1646 "RendezvousStateError", 1647 "Store", 1648 "Token", 1649 "Tuple", 1650 "b64decode", 1651 "b64encode", 1652 "cast", 1653 "parse_rendezvous_endpoint" 1654 ], 1655 "torch.distributed.elastic.rendezvous.etcd_server": [ 1656 "Optional", 1657 "TextIO", 1658 "Union" 1659 ], 1660 "torch.distributed.elastic.rendezvous.etcd_store": [ 1661 "Optional", 1662 "Store", 1663 "b64decode", 1664 "b64encode" 1665 ], 1666 "torch.distributed.elastic.rendezvous.static_tcp_rendezvous": [ 1667 "Optional", 1668 "PrefixStore", 1669 "RendezvousHandler", 1670 "RendezvousParameters", 1671 "Store", 1672 "TCPStore", 1673 "Tuple", 1674 "cast", 1675 "parse_rendezvous_endpoint" 1676 ], 1677 "torch.distributed.elastic.utils.distributed": [ 1678 "closing", 1679 "get_logger" 1680 ], 1681 "torch.distributed.fsdp.sharded_grad_scaler": [ 1682 "Any", 1683 "Dict", 1684 "GradScaler", 1685 "Iterable", 1686 "List", 1687 "OptState", 1688 "Optional", 1689 "ProcessGroup", 1690 "Sequence", 1691 "Tuple", 1692 "Union", 1693 "defaultdict", 1694 "overload" 1695 ], 1696 "torch.distributed.launch": [ 1697 "get_args_parser", 1698 "run" 1699 ], 1700 "torch.distributed.rpc.rref_proxy": [ 1701 "Future", 1702 "partial", 1703 "rpc_async" 1704 ], 1705 "torch.distributed.run": [ 1706 "ArgumentParser", 1707 "Callable", 1708 "LaunchConfig", 1709 "List", 1710 "Std", 1711 "Tuple", 1712 "Union", 1713 "check_env", 1714 "elastic_launch", 1715 "env", 1716 "get_logger", 1717 "macros", 1718 "record", 1719 "DefaultLogsSpecs", 1720 "LogsSpecs", 1721 "Optional", 1722 "Set", 1723 "Type" 1724 ], 1725 "torch.fx.annotate": [ 1726 "Proxy", 1727 "compatibility" 1728 ], 1729 "torch.fx.experimental.accelerator_partitioner": [ 1730 "Deque", 1731 "Device", 1732 "Dict", 1733 "GraphModule", 1734 "List", 1735 "NamedTuple", 1736 "Node", 1737 "NodeLatency", 1738 "Partition", 1739 "PartitionMode", 1740 "PartitionerConfig", 1741 "Set", 1742 "Tuple", 1743 "deque", 1744 "get_extra_size_of", 1745 "get_latency_of_partitioned_graph", 1746 "get_partition_to_latency_mapping", 1747 "get_size_of_all_nodes", 1748 "map_arg", 1749 "split_module" 1750 ], 1751 "torch.fx.experimental.graph_gradual_typechecker": [ 1752 "BatchNorm2d", 1753 "Callable", 1754 "Conv2d", 1755 "Dict", 1756 "Equality", 1757 "Node", 1758 "Target", 1759 "TensorType", 1760 "Var", 1761 "is_consistent", 1762 "is_more_precise", 1763 "reduce" 1764 ], 1765 "torch.fx.experimental.merge_matmul": [ 1766 "Dict", 1767 "List", 1768 "Node", 1769 "Tuple", 1770 "legalize_graph", 1771 "symbolic_trace" 1772 ], 1773 "torch.fx.experimental.meta_tracer": [ 1774 "Any", 1775 "Callable", 1776 "Dict", 1777 "Optional", 1778 "Union" 1779 ], 1780 "torch.fx.experimental.migrate_gradual_types.constraint": [ 1781 "TensorType" 1782 ], 1783 "torch.fx.experimental.migrate_gradual_types.constraint_generator": [ 1784 "ApplyBroadcasting", 1785 "BatchNorm2d", 1786 "BinConstraintD", 1787 "BinConstraintT", 1788 "CalcConv", 1789 "CalcMaxPool", 1790 "CalcProduct", 1791 "Callable", 1792 "CanReshape", 1793 "Conj", 1794 "Conv2d", 1795 "DGreatestUpperBound", 1796 "DVar", 1797 "Dict", 1798 "Disj", 1799 "F", 1800 "GetItem", 1801 "GetItemTensor", 1802 "IndexSelect", 1803 "Iterable", 1804 "Node", 1805 "T", 1806 "TGreatestUpperBound", 1807 "TVar", 1808 "Target", 1809 "TensorType", 1810 "Transpose", 1811 "gen_bvar", 1812 "gen_dvar", 1813 "gen_nat_constraints", 1814 "gen_tensor_dims", 1815 "gen_tvar" 1816 ], 1817 "torch.fx.experimental.migrate_gradual_types.constraint_transformation": [ 1818 "ApplyBroadcasting", 1819 "BinConstraintD", 1820 "BinConstraintT", 1821 "CalcConv", 1822 "CalcMaxPool", 1823 "CalcProduct", 1824 "Callable", 1825 "CanReshape", 1826 "Conj", 1827 "Constraint", 1828 "DGreatestUpperBound", 1829 "DVar", 1830 "Dict", 1831 "Disj", 1832 "F", 1833 "GetItem", 1834 "GetItemTensor", 1835 "IndexSelect", 1836 "List", 1837 "Prod", 1838 "T", 1839 "TGreatestUpperBound", 1840 "TVar", 1841 "TensorType", 1842 "Transpose", 1843 "gen_dvar", 1844 "gen_nat_constraints", 1845 "gen_tensor_dims" 1846 ], 1847 "torch.fx.experimental.migrate_gradual_types.transform_to_z3": [ 1848 "BVar", 1849 "BinConstraintD", 1850 "BinConstraintT", 1851 "Conj", 1852 "ConstraintGenerator", 1853 "D", 1854 "DVar", 1855 "Disj", 1856 "F", 1857 "Prod", 1858 "T", 1859 "TVar", 1860 "TensorType", 1861 "is_algebraic_expression", 1862 "is_bool_expr", 1863 "is_dim", 1864 "transform_constraint" 1865 ], 1866 "torch.fx.experimental.migrate_gradual_types.util": [ 1867 "BVar", 1868 "BinConstraintD", 1869 "DVar", 1870 "TVar" 1871 ], 1872 "torch.fx.experimental.normalize": [ 1873 "AnnotateTypesWithSchema", 1874 "Any", 1875 "Argument", 1876 "Callable", 1877 "Dict", 1878 "Node", 1879 "Optional", 1880 "Proxy", 1881 "Target", 1882 "Transformer", 1883 "Tuple", 1884 "create_type_hint", 1885 "map_aggregate", 1886 "normalize_function", 1887 "normalize_module" 1888 ], 1889 "torch.fx.experimental.optimization": [ 1890 "Any", 1891 "Argument", 1892 "Dict", 1893 "Enum", 1894 "Iterable", 1895 "List", 1896 "Optional", 1897 "ShapeProp", 1898 "Target", 1899 "Tuple", 1900 "Type", 1901 "cast", 1902 "defaultdict", 1903 "fuse_conv_bn_eval" 1904 ], 1905 "torch.fx.experimental.partitioner_utils": [ 1906 "Dict", 1907 "Enum", 1908 "List", 1909 "NamedTuple", 1910 "Node", 1911 "Set", 1912 "map_arg" 1913 ], 1914 "torch.fx.experimental.proxy_tensor": [ 1915 "PreDispatchTorchFunctionMode", 1916 "ProxySymDispatchMode", 1917 "ProxyTorchDispatchMode", 1918 "decompose", 1919 "disable_autocast_cache", 1920 "disable_proxy_modes_tracing", 1921 "extract_val", 1922 "fake_signature", 1923 "fetch_sym_proxy", 1924 "fetch_object_proxy", 1925 "get_isolated_graphmodule", 1926 "get_proxy_slot", 1927 "get_torch_dispatch_modes", 1928 "has_proxy_slot", 1929 "is_sym_node", 1930 "maybe_handle_decomp", 1931 "proxy_call", 1932 "set_meta", 1933 "set_original_aten_op", 1934 "set_proxy_slot", 1935 "snapshot_fake", 1936 "thunkify", 1937 "track_tensor", 1938 "track_tensor_tree", 1939 "wrap_key", 1940 "wrapper_and_args_for_make_fx", 1941 "TorchFunctionMetadataMode" 1942 ], 1943 "torch.fx.experimental.rewriter": [ 1944 "Any", 1945 "Callable", 1946 "Dict", 1947 "FunctionType", 1948 "Graph", 1949 "Optional", 1950 "Tracer", 1951 "Union", 1952 "cast", 1953 "normalize_source_lines" 1954 ], 1955 "torch.fx.experimental.schema_type_annotation": [ 1956 "Any", 1957 "Argument", 1958 "Dict", 1959 "Optional", 1960 "Target", 1961 "Transformer", 1962 "Tuple" 1963 ], 1964 "torch.fx.experimental.sym_dispatch_mode": [ 1965 "sym_function_mode", 1966 "set_sym_function_mode" 1967 ], 1968 "torch.fx.experimental.sym_node": [ 1969 "SymNode", 1970 "method_to_operator", 1971 "magic_methods", 1972 "to_node", 1973 "wrap_node", 1974 "is_channels_last_contiguous_2d", 1975 "is_channels_last_contiguous_3d", 1976 "is_channels_last_strides_2d", 1977 "is_channels_last_strides_3d", 1978 "is_non_overlapping_and_dense_indicator", 1979 "sympy_is_channels_last_contiguous_2d", 1980 "sympy_is_channels_last_contiguous_3d", 1981 "sympy_is_channels_last_strides_2d", 1982 "sympy_is_channels_last_strides_3d", 1983 "sympy_is_channels_last_strides_generic", 1984 "is_contiguous", 1985 "sympy_is_contiguous", 1986 "sympy_is_contiguous_generic", 1987 "sym_sqrt" 1988 ], 1989 "torch.fx.experimental.symbolic_shapes": [ 1990 "Constraint", 1991 "ConstraintViolationError", 1992 "DimConstraints", 1993 "DimDynamic", 1994 "DynamicDimConstraintPrinter", 1995 "EqualityConstraint", 1996 "GuardOnDataDependentSymNode", 1997 "LoggingShapeGuardPrinter", 1998 "SymExprPrinter", 1999 "RelaxedUnspecConstraint", 2000 "RuntimeAssert", 2001 "ShapeGuardPrinter", 2002 "StrictMinMaxConstraint", 2003 "bind_symbols", 2004 "cast_symbool_to_symint_guardless", 2005 "constrain_range", 2006 "constrain_unify", 2007 "definitely_false", 2008 "definitely_true", 2009 "error", 2010 "eval_guards", 2011 "eval_is_non_overlapping_and_dense", 2012 "expect_true", 2013 "find_symbol_binding_fx_nodes", 2014 "free_unbacked_symbols", 2015 "fx_placeholder_targets", 2016 "fx_placeholder_vals", 2017 "guard_bool", 2018 "has_hint", 2019 "is_symbolic", 2020 "parallel_and", 2021 "parallel_or", 2022 "safe_expand", 2023 "uninteresting_files", 2024 "CallMethodKey", 2025 "DivideByKey", 2026 "InnerTensorKey", 2027 "PropagateUnbackedSymInts", 2028 "ShapeEnvSettings", 2029 "log_lru_cache_stats", 2030 "PendingUnbackedSymbolNotFound", 2031 "lru_cache" 2032 ], 2033 "torch.fx.experimental.unification.match": [ 2034 "first", 2035 "freeze", 2036 "groupby", 2037 "isvar", 2038 "reify", 2039 "unify" 2040 ], 2041 "torch.fx.experimental.unify_refinements": [ 2042 "Refine", 2043 "TensorType", 2044 "Var", 2045 "unify" 2046 ], 2047 "torch.fx.experimental.validator": [ 2048 "bisect" 2049 ], 2050 "torch.fx.passes.backends.cudagraphs": [ 2051 "CapabilityBasedPartitioner", 2052 "FakeTensorProp", 2053 "OperatorSupport", 2054 "tree_map" 2055 ], 2056 "torch.fx.passes.dialect.common.cse_pass": [ 2057 "Any", 2058 "Dict", 2059 "Graph", 2060 "GraphModule", 2061 "Node", 2062 "PassBase", 2063 "PassResult", 2064 "Tuple", 2065 "tree_flatten" 2066 ], 2067 "torch.fx.passes.infra.partitioner": [ 2068 "Deque", 2069 "Dict", 2070 "GraphModule", 2071 "Iterable", 2072 "List", 2073 "Node", 2074 "OperatorSupportBase", 2075 "Optional", 2076 "Sequence", 2077 "Set", 2078 "copy", 2079 "deque", 2080 "fuse_by_partitions" 2081 ], 2082 "torch.fx.passes.tests.test_pass_manager": [ 2083 "PassManager", 2084 "inplace_wrapper", 2085 "these_before_those_pass_constraint", 2086 "this_before_that_pass_constraint" 2087 ], 2088 "torch.fx.passes.utils.fuser_utils": [ 2089 "Dict", 2090 "Graph", 2091 "GraphModule", 2092 "List", 2093 "Node", 2094 "NodeList", 2095 "NodeSet", 2096 "SimpleQueue", 2097 "Tuple", 2098 "compatibility", 2099 "legalize_graph", 2100 "lift_subgraph_as_module" 2101 ], 2102 "torch.fx.tensor_type": [ 2103 "Var", 2104 "compatibility" 2105 ], 2106 "torch.jit.generate_bytecode": [ 2107 "List" 2108 ], 2109 "torch.jit.mobile": [ 2110 "validate_map_location" 2111 ], 2112 "torch.jit.quantized": [ 2113 "List", 2114 "Optional", 2115 "PackedSequence", 2116 "Tensor", 2117 "Tuple" 2118 ], 2119 "torch.jit.unsupported_tensor_ops": [ 2120 "Any", 2121 "Dict", 2122 "dedent" 2123 ], 2124 "torch.monitor": [ 2125 "Aggregation", 2126 "Event", 2127 "EventHandlerHandle", 2128 "Stat", 2129 "data_value_t", 2130 "log_event", 2131 "register_event_handler", 2132 "unregister_event_handler" 2133 ], 2134 "torch.multiprocessing.pool": [ 2135 "SimpleQueue" 2136 ], 2137 "torch.multiprocessing.queue": [ 2138 "ForkingPickler" 2139 ], 2140 "torch.nn.quantized.dynamic.modules.conv": [ 2141 "Conv1d", 2142 "Conv2d", 2143 "Conv3d", 2144 "ConvTranspose1d", 2145 "ConvTranspose2d", 2146 "ConvTranspose3d" 2147 ], 2148 "torch.nn.quantized.dynamic.modules.linear": [ 2149 "Linear" 2150 ], 2151 "torch.nn.quantized.modules.activation": [ 2152 "ELU", 2153 "Hardswish", 2154 "LeakyReLU", 2155 "MultiheadAttention", 2156 "PReLU", 2157 "ReLU6", 2158 "Sigmoid", 2159 "Softmax" 2160 ], 2161 "torch.nn.quantized.modules.batchnorm": [ 2162 "BatchNorm2d", 2163 "BatchNorm3d" 2164 ], 2165 "torch.nn.quantized.modules.conv": [ 2166 "Conv1d", 2167 "Conv2d", 2168 "Conv3d", 2169 "ConvTranspose1d", 2170 "ConvTranspose2d", 2171 "ConvTranspose3d" 2172 ], 2173 "torch.nn.quantized.modules.dropout": [ 2174 "Dropout" 2175 ], 2176 "torch.nn.quantized.modules.embedding_ops": [ 2177 "Embedding", 2178 "EmbeddingBag", 2179 "EmbeddingPackedParams" 2180 ], 2181 "torch.nn.quantized.modules.functional_modules": [ 2182 "FXFloatFunctional", 2183 "FloatFunctional", 2184 "QFunctional" 2185 ], 2186 "torch.nn.quantized.modules.linear": [ 2187 "Linear", 2188 "LinearPackedParams" 2189 ], 2190 "torch.nn.quantized.modules.normalization": [ 2191 "GroupNorm", 2192 "InstanceNorm1d", 2193 "InstanceNorm2d", 2194 "InstanceNorm3d", 2195 "LayerNorm" 2196 ], 2197 "torch.nn.quantized.modules.rnn": [ 2198 "LSTM" 2199 ], 2200 "torch.nn.quantized.modules.utils": [ 2201 "WeightedQuantizedModule" 2202 ], 2203 "torch.nn.utils.prune": [ 2204 "ABC", 2205 "Iterable", 2206 "Tuple", 2207 "abstractmethod" 2208 ], 2209 "torch.onnx.verification": [ 2210 "Any", 2211 "Callable", 2212 "Collection", 2213 "Dict", 2214 "FrozenSet", 2215 "List", 2216 "Mapping", 2217 "Number", 2218 "Optional", 2219 "Sequence", 2220 "Set", 2221 "Tuple", 2222 "Union" 2223 ], 2224 "torch.quantization.fx": [ 2225 "convert", 2226 "fuse", 2227 "prepare" 2228 ], 2229 "torch.quantization.fx.convert": [ 2230 "convert" 2231 ], 2232 "torch.quantization.fx.fuse": [ 2233 "fuse" 2234 ], 2235 "torch.quantization.fx.fusion_patterns": [ 2236 "DefaultFuseHandler", 2237 "FuseHandler" 2238 ], 2239 "torch.quantization.fx.graph_module": [ 2240 "FusedGraphModule", 2241 "GraphModule", 2242 "ObservedGraphModule", 2243 "ObservedStandaloneGraphModule", 2244 "QuantizedGraphModule" 2245 ], 2246 "torch.quantization.fx.match_utils": [ 2247 "MatchAllNode" 2248 ], 2249 "torch.quantization.fx.pattern_utils": [ 2250 "QuantizeHandler", 2251 "get_default_fusion_patterns", 2252 "get_default_output_activation_post_process_map", 2253 "get_default_quant_patterns" 2254 ], 2255 "torch.quantization.fx.prepare": [ 2256 "prepare" 2257 ], 2258 "torch.quantization.fx.quantization_patterns": [ 2259 "BatchNormQuantizeHandler", 2260 "BinaryOpQuantizeHandler", 2261 "CatQuantizeHandler", 2262 "ConvReluQuantizeHandler", 2263 "CopyNodeQuantizeHandler", 2264 "CustomModuleQuantizeHandler", 2265 "DefaultNodeQuantizeHandler", 2266 "EmbeddingQuantizeHandler", 2267 "FixedQParamsOpQuantizeHandler", 2268 "GeneralTensorShapeOpQuantizeHandler", 2269 "LinearReLUQuantizeHandler", 2270 "QuantizeHandler", 2271 "RNNDynamicQuantizeHandler", 2272 "StandaloneModuleQuantizeHandler" 2273 ], 2274 "torch.quantization.fx.quantization_types": [ 2275 "Pattern", 2276 "QuantizerCls" 2277 ], 2278 "torch.quantization.fx.utils": [ 2279 "all_node_args_have_no_tensors", 2280 "assert_and_get_unique_device", 2281 "create_getattr_from_value", 2282 "get_custom_module_class_keys", 2283 "get_linear_prepack_op_for_dtype", 2284 "get_new_attr_name_with_prefix", 2285 "get_non_observable_arg_indexes_and_types", 2286 "get_qconv_prepack_op", 2287 "graph_module_from_producer_nodes", 2288 "maybe_get_next_module" 2289 ], 2290 "torch.quantization.quantize_fx": [ 2291 "ObservedGraphModule", 2292 "QuantizationTracer", 2293 "Scope", 2294 "ScopeContextManager", 2295 "convert_fx", 2296 "fuse_fx", 2297 "prepare_fx", 2298 "prepare_qat_fx" 2299 ], 2300 "torch.quantization.utils": [ 2301 "activation_dtype", 2302 "activation_is_int8_quantized", 2303 "activation_is_statically_quantized", 2304 "calculate_qmin_qmax", 2305 "check_min_max_valid", 2306 "get_combined_dict", 2307 "get_qconfig_dtypes", 2308 "get_qparam_dict", 2309 "get_quant_type", 2310 "get_swapped_custom_module_class", 2311 "getattr_from_fqn", 2312 "is_per_channel", 2313 "is_per_tensor", 2314 "weight_dtype", 2315 "weight_is_quantized", 2316 "weight_is_statically_quantized" 2317 ], 2318 "torch.utils.benchmark": [ 2319 "Number", 2320 "Optional", 2321 "Tuple", 2322 "Union", 2323 "timer" 2324 ], 2325 "torch.utils.benchmark.examples.op_benchmark": [ 2326 "BinaryOpFuzzer", 2327 "Timer", 2328 "UnaryOpFuzzer" 2329 ], 2330 "torch.utils.benchmark.examples.spectral_ops_fuzz_test": [ 2331 "ArgumentParser", 2332 "Iterable", 2333 "SpectralOpFuzzer", 2334 "namedtuple" 2335 ], 2336 "torch.utils.benchmark.op_fuzzers.binary": [ 2337 "FuzzedParameter", 2338 "FuzzedTensor", 2339 "Fuzzer", 2340 "ParameterAlias" 2341 ], 2342 "torch.utils.benchmark.op_fuzzers.sparse_binary": [ 2343 "FuzzedParameter", 2344 "FuzzedSparseTensor", 2345 "Fuzzer", 2346 "ParameterAlias" 2347 ], 2348 "torch.utils.benchmark.op_fuzzers.sparse_unary": [ 2349 "FuzzedParameter", 2350 "FuzzedSparseTensor", 2351 "Fuzzer", 2352 "ParameterAlias" 2353 ], 2354 "torch.utils.benchmark.op_fuzzers.spectral": [ 2355 "power_range" 2356 ], 2357 "torch.utils.benchmark.op_fuzzers.unary": [ 2358 "FuzzedParameter", 2359 "FuzzedTensor", 2360 "Fuzzer", 2361 "ParameterAlias" 2362 ], 2363 "torch.utils.benchmark.utils.compile": [ 2364 "bench_loop" 2365 ], 2366 "torch.utils.bundled_inputs": [ 2367 "Any", 2368 "Callable", 2369 "Dict", 2370 "List", 2371 "ListType", 2372 "NamedTuple", 2373 "Optional", 2374 "Sequence", 2375 "Tuple", 2376 "TupleType", 2377 "TypeVar", 2378 "Union", 2379 "wrap_cpp_module" 2380 ], 2381 "torch.utils.collect_env": [ 2382 "namedtuple" 2383 ], 2384 "torch.utils.data.datapipes.gen_pyi": [ 2385 "Any", 2386 "Dict", 2387 "List", 2388 "Set", 2389 "Tuple", 2390 "Union", 2391 "defaultdict" 2392 ], 2393 "torch.utils.data.datapipes.utils.snapshot": [ 2394 "IterDataPipe", 2395 "apply_random_seed" 2396 ], 2397 "torch.utils.flop_counter": [ 2398 "addmm_flop", 2399 "baddbmm_flop", 2400 "bmm_flop", 2401 "conv_backward_flop", 2402 "conv_flop", 2403 "conv_flop_count", 2404 "convert_num_with_suffix", 2405 "convert_to_percent_str", 2406 "get_shape", 2407 "get_suffix_str", 2408 "mm_flop", 2409 "normalize_tuple", 2410 "sdpa_backward_flop", 2411 "sdpa_backward_flop_count", 2412 "sdpa_flop", 2413 "sdpa_flop_count", 2414 "shape_wrapper", 2415 "transpose_shape" 2416 ], 2417 "torch.utils.jit.log_extract": [ 2418 "Any", 2419 "List", 2420 "Timer", 2421 "Tuple", 2422 "cast", 2423 "contextmanager" 2424 ], 2425 "torch.utils.mobile_optimizer": [ 2426 "Enum", 2427 "List", 2428 "MobileOptimizerType", 2429 "Optional", 2430 "Set" 2431 ], 2432 "torch.utils.model_dump": [ 2433 "main" 2434 ], 2435 "torch.utils.model_zoo": [ 2436 "load_url", 2437 "tqdm" 2438 ], 2439 "torch.utils.tensorboard": [ 2440 "RecordWriter" 2441 ], 2442 "torch.ao.quantization.experimental.APoT_tensor": [ 2443 "APoTQuantizer" 2444 ], 2445 "torch.ao.quantization.experimental.fake_quantize": [ 2446 "APoTObserver", 2447 "FakeQuantizeBase", 2448 "Tensor" 2449 ], 2450 "torch.ao.quantization.experimental.fake_quantize_function": [ 2451 "dequantize_APoT", 2452 "quantize_APoT", 2453 "Tensor" 2454 ], 2455 "torch.ao.quantization.experimental.linear": [ 2456 "APoTObserver", 2457 "quantize_APoT", 2458 "WeightedQuantizedModule" 2459 ], 2460 "torch.ao.quantization.experimental.observer": [ 2461 "apot_to_float", 2462 "float_to_apot", 2463 "ObserverBase" 2464 ], 2465 "torch.ao.quantization.experimental.qconfig": [ 2466 "APoTFakeQuantize", 2467 "default_symmetric_fake_quant", 2468 "default_weight_symmetric_fake_quant", 2469 "FakeQuantize", 2470 "MinMaxObserver", 2471 "QConfig" 2472 ], 2473 "torch.ao.quantization.experimental.quantizer": [ 2474 "apot_to_float", 2475 "float_to_apot", 2476 "quant_dequant_util", 2477 "Tensor" 2478 ], 2479 "torch.ao.sparsity": [ 2480 "BaseScheduler", 2481 "BaseSparsifier", 2482 "CubicSL", 2483 "FakeSparsity", 2484 "fqn_to_module", 2485 "get_arg_info_from_tensor_fqn", 2486 "get_dynamic_sparse_quantized_mapping", 2487 "get_static_sparse_quantized_mapping", 2488 "LambdaSL", 2489 "module_to_fqn", 2490 "NearlyDiagonalSparsifier", 2491 "WeightNormSparsifier" 2492 ], 2493 "torch.ao.sparsity.scheduler.base_scheduler": [ 2494 "BaseScheduler" 2495 ], 2496 "torch.ao.sparsity.scheduler.cubic_scheduler": [ 2497 "CubicSL" 2498 ], 2499 "torch.ao.sparsity.scheduler.lambda_scheduler": [ 2500 "LambdaSL" 2501 ], 2502 "torch.ao.sparsity.sparsifier.base_sparsifier": [ 2503 "BaseSparsifier" 2504 ], 2505 "torch.ao.sparsity.sparsifier.nearly_diagonal_sparsifier": [ 2506 "NearlyDiagonalSparsifier" 2507 ], 2508 "torch.ao.sparsity.sparsifier.utils": [ 2509 "FakeSparsity", 2510 "fqn_to_module", 2511 "get_arg_info_from_tensor_fqn", 2512 "module_to_fqn" 2513 ], 2514 "torch.ao.sparsity.sparsifier.weight_norm_sparsifier": [ 2515 "WeightNormSparsifier" 2516 ], 2517 "torch.csrc.jit.tensorexpr.codegen_external": [ 2518 "FileManager", 2519 "parse_native_yaml" 2520 ], 2521 "torch.distributed.checkpoint.examples.async_checkpointing_example": [ 2522 "FSDP", 2523 "init_device_mesh" 2524 ], 2525 "torch.distributed.checkpoint.examples.fsdp_checkpoint_example": [ 2526 "FSDP", 2527 "load_sharded_optimizer_state_dict", 2528 "StateDictType" 2529 ], 2530 "torch.distributed.checkpoint.examples.stateful_example": [ 2531 "FSDP", 2532 "init_device_mesh" 2533 ], 2534 "torch.distributed.elastic.events.fb.scuba": [ 2535 "await_sync", 2536 "cast", 2537 "Dict", 2538 "Enum", 2539 "Event", 2540 "EventMetadataValue", 2541 "List", 2542 "Optional", 2543 "RdzvEvent", 2544 "RuntimeEnvironment", 2545 "TorchelasticRdzvLogEntry", 2546 "TorchelasticStatusLogEntry", 2547 "WhenceScribeLogged" 2548 ], 2549 "torch.distributed.elastic.metrics.fb.service_data_metrics": [ 2550 "MetricHandler", 2551 "ServiceDataMetrics" 2552 ], 2553 "torch.distributed.elastic.metrics.static_init": [ 2554 "configure", 2555 "get_logger", 2556 "MetricsConfig", 2557 "Optional", 2558 "ServiceDataMetricsHandler", 2559 "TorchElasticService" 2560 ], 2561 "torch.distributed.elastic.multiprocessing.errors.fb.error_handler_fb": [ 2562 "Any", 2563 "Dict", 2564 "ErrorHandler", 2565 "format_exception", 2566 "generate_python_trace", 2567 "MastReplyFileErrorCode", 2568 "Optional", 2569 "RuntimeEnvironment", 2570 "RuntimeEnvironmentScheduler", 2571 "write_formatted_message" 2572 ], 2573 "torch.distributed.elastic.multiprocessing.errors.handlers": [ 2574 "ErrorHandlerFB" 2575 ], 2576 "torch.distributed.elastic.rendezvous.fb.mast_rendezvous": [ 2577 "create_c10d_store", 2578 "DistNetworkError", 2579 "DistStoreError", 2580 "get_logger", 2581 "List", 2582 "Optional", 2583 "RendezvousClosedError", 2584 "RendezvousHandler", 2585 "RendezvousParameters", 2586 "RendezvousTimeoutError", 2587 "Tuple" 2588 ], 2589 "torch.distributed.elastic.rendezvous.fb.zeus": [ 2590 "gethostname", 2591 "get_logger", 2592 "namedtuple", 2593 "Optional", 2594 "RendezvousClosedError", 2595 "RendezvousHandler", 2596 "RendezvousParameters", 2597 "RendezvousTimeoutError" 2598 ], 2599 "torch.distributed.elastic.rendezvous.registry": [ 2600 "create_handler", 2601 "RendezvousHandler", 2602 "RendezvousParameters" 2603 ], 2604 "torch.distributed.logging_handlers": [ 2605 "C10D_CATEGORY", 2606 "Dict", 2607 "LogCategory", 2608 "Optional", 2609 "Sample", 2610 "ScubaData", 2611 "signpost", 2612 "SignpostType" 2613 ], 2614 "torch.utils.benchmark.examples.sparse.op_benchmark": [ 2615 "BinaryOpSparseFuzzer", 2616 "Timer", 2617 "UnaryOpSparseFuzzer" 2618 ], 2619 "torch.version": [ 2620 "get_file_path" 2621 ], 2622 "torch.ao.nn.intrinsic.modules": [ 2623 "_FusedModule" 2624 ], 2625 "torch.distributed.benchmarks.benchmark_ddp_rpc": [ 2626 "BackendType", 2627 "DDP", 2628 "DistributedOptimizer", 2629 "RRef", 2630 "TensorPipeRpcBackendOptions" 2631 ], 2632 "torch.distributed.pipelining": [ 2633 "Pipe", 2634 "PipelineStage", 2635 "SplitPoint", 2636 "pipe_split", 2637 "pipeline" 2638 ], 2639 "torch.distributed.pipelining.microbatch": [ 2640 "Any", 2641 "Dict", 2642 "List", 2643 "Optional", 2644 "Tuple", 2645 "tree_flatten", 2646 "tree_unflatten" 2647 ], 2648 "torch.export": [ 2649 "Constraint", 2650 "ShapesCollection" 2651 ], 2652 "torch.export.dynamic_shapes": [ 2653 "Constraint", 2654 "ShapesCollection" 2655 ], 2656 "torch.export.graph_signature": [ 2657 "TokenArgument" 2658 ], 2659 "torch.fx.experimental.shape_inference.infer_shape": [ 2660 "DimDynamic", 2661 "FakeTensorMode", 2662 "LocalSource", 2663 "ShapeEnv", 2664 "defaultdict", 2665 "infer_symbol_values", 2666 "make_fx" 2667 ], 2668 "torch.fx.experimental.shape_inference.infer_symbol_values": [ 2669 "Any", 2670 "DefaultDict", 2671 "Dict", 2672 "List", 2673 "Tuple", 2674 "Union" 2675 ], 2676 "torch.fx.passes.runtime_assert": [ 2677 "Any", 2678 "Dict", 2679 "GraphModule", 2680 "Optional", 2681 "Set", 2682 "ShapeEnv", 2683 "SymNode", 2684 "compatibility", 2685 "lazy_format_graph_code" 2686 ], 2687 "torch.library": [ 2688 "opcheck", 2689 "register_autograd", 2690 "register_kernel" 2691 ], 2692 "torch.mtia": [ 2693 "DeferredMtiaCallError", 2694 "StreamContext" 2695 ], 2696 "torch.onnx.symbolic_helper": [ 2697 "Any", 2698 "Callable", 2699 "List", 2700 "Literal", 2701 "NoReturn", 2702 "Number", 2703 "Optional", 2704 "Sequence", 2705 "Set", 2706 "Tuple", 2707 "Union" 2708 ], 2709 "torch.onnx.symbolic_opset18": [ 2710 "amax", 2711 "amin", 2712 "aminmax", 2713 "embedding_bag", 2714 "linalg_vector_norm", 2715 "max", 2716 "maximum", 2717 "min", 2718 "minimum" 2719 ], 2720 "torch.onnx.symbolic_opset20": [ 2721 "_affine_grid_generator", 2722 "_grid_sampler", 2723 "convert_grid_sample_mode" 2724 ], 2725 "torch.utils.data.datapipes.dataframe.dataframe_wrapper": [ 2726 "Any", 2727 "Optional" 2728 ], 2729 "torch.utils.hipify.hipify_python": [ 2730 "TrieNode" 2731 ] 2732} 2733