setup.py
torch/_VF.py
torch/__config__.py
torch/__future__.py
torch/__init__.py
torch/_appdirs.py
torch/_classes.py
torch/_compile.py
torch/_custom_ops.py
torch/_deploy.py
torch/_guards.py
torch/_jit_internal.py
torch/_linalg_utils.py
torch/_lobpcg.py
torch/_lowrank.py
torch/_meta_registrations.py
torch/_namedtensor_internals.py
torch/_ops.py
torch/_python_dispatcher.py
torch/_sources.py
torch/_storage_docs.py
torch/_tensor.py
torch/_tensor_docs.py
torch/_tensor_str.py
torch/_torch_docs.py
torch/_utils.py
torch/_utils_internal.py
torch/_vmap_internals.py
torch/_weights_only_unpickler.py
torch/functional.py
torch/hub.py
torch/library.py
torch/overrides.py
torch/quasirandom.py
torch/random.py
torch/return_types.py
torch/serialization.py
torch/storage.py
torch/torch_version.py
torch/types.py
torch/_awaits/__init__.py
torch/_custom_op/__init__.py
torch/_custom_op/autograd.py
torch/_custom_op/functional.py
torch/_custom_op/impl.py
torch/_decomp/__init__.py
torch/_decomp/decompositions.py
torch/_decomp/decompositions_for_jvp.py
torch/_decomp/decompositions_for_rng.py
torch/_dispatch/__init__.py
torch/_dispatch/python.py
torch/_dynamo/__init__.py
torch/_dynamo/allowed_functions.py
torch/_dynamo/bytecode_analysis.py
torch/_dynamo/bytecode_transformation.py
torch/_dynamo/cache_size.py
torch/_dynamo/codegen.py
torch/_dynamo/compiled_autograd.py
torch/_dynamo/comptime.py
torch/_dynamo/config.py
torch/_dynamo/config_utils.py
torch/_dynamo/convert_frame.py
torch/_dynamo/current_scope_id.py
torch/_dynamo/debug_utils.py
torch/_dynamo/decorators.py
torch/_dynamo/eval_frame.py
torch/_dynamo/exc.py
torch/_dynamo/external_utils.py
torch/_dynamo/guards.py
torch/_dynamo/hooks.py
torch/_dynamo/logging.py
torch/_dynamo/mutation_guard.py
torch/_dynamo/output_graph.py
torch/_dynamo/profiler.py
torch/_dynamo/replay_record.py
torch/_dynamo/resume_execution.py
torch/_dynamo/side_effects.py
torch/_dynamo/skipfiles.py
torch/_dynamo/source.py
torch/_dynamo/symbolic_convert.py
torch/_dynamo/test_case.py
torch/_dynamo/test_minifier_common.py
torch/_dynamo/testing.py
torch/_dynamo/types.py
torch/_dynamo/utils.py
torch/_dynamo/backends/__init__.py
torch/_dynamo/backends/common.py
torch/_dynamo/backends/cudagraphs.py
torch/_dynamo/backends/debugging.py
torch/_dynamo/backends/distributed.py
torch/_dynamo/backends/inductor.py
torch/_dynamo/backends/onnxrt.py
torch/_dynamo/backends/registry.py
torch/_dynamo/backends/tensorrt.py
torch/_dynamo/backends/torchxla.py
torch/_dynamo/backends/tvm.py
torch/_dynamo/repro/__init__.py
torch/_dynamo/repro/after_aot.py
torch/_dynamo/repro/after_dynamo.py
torch/_dynamo/variables/__init__.py
torch/_dynamo/variables/base.py
torch/_dynamo/variables/builder.py
torch/_dynamo/variables/builtin.py
torch/_dynamo/variables/constant.py
torch/_dynamo/variables/ctx_manager.py
torch/_dynamo/variables/dicts.py
torch/_dynamo/variables/distributed.py
torch/_dynamo/variables/functions.py
torch/_dynamo/variables/higher_order_ops.py
torch/_dynamo/variables/lists.py
torch/_dynamo/variables/misc.py
torch/_dynamo/variables/nn_module.py
torch/_dynamo/variables/optimizer.py
torch/_dynamo/variables/tensor.py
torch/_dynamo/variables/torch.py
torch/_dynamo/variables/user_defined.py
torch/_export/__init__.py
torch/_export/constraints.py
torch/_export/error.py
torch/_export/exported_program.py
torch/_export/pass_base.py
torch/_export/utils.py
torch/_export/verifier.py
torch/_export/wrappers.py
torch/_export/db/__init__.py
torch/_export/db/case.py
torch/_export/db/gen_example.py
torch/_export/db/logging.py
torch/_export/db/examples/__init__.py
torch/_export/db/examples/assume_constant_result.py
torch/_export/db/examples/autograd_function.py
torch/_export/db/examples/class_method.py
torch/_export/db/examples/cond_branch_class_method.py
torch/_export/db/examples/cond_branch_nested_function.py
torch/_export/db/examples/cond_branch_nonlocal_variables.py
torch/_export/db/examples/cond_closed_over_variable.py
torch/_export/db/examples/cond_operands.py
torch/_export/db/examples/cond_predicate.py
torch/_export/db/examples/constrain_as_size_example.py
torch/_export/db/examples/constrain_as_value_example.py
torch/_export/db/examples/decorator.py
torch/_export/db/examples/dictionary.py
torch/_export/db/examples/dynamic_shape_assert.py
torch/_export/db/examples/dynamic_shape_constructor.py
torch/_export/db/examples/dynamic_shape_if_guard.py
torch/_export/db/examples/dynamic_shape_map.py
torch/_export/db/examples/dynamic_shape_round.py
torch/_export/db/examples/dynamic_shape_slicing.py
torch/_export/db/examples/dynamic_shape_view.py
torch/_export/db/examples/fn_with_kwargs.py
torch/_export/db/examples/list_contains.py
torch/_export/db/examples/list_unpack.py
torch/_export/db/examples/nested_function.py
torch/_export/db/examples/null_context_manager.py
torch/_export/db/examples/pytree_flatten.py
torch/_export/db/examples/scalar_output.py
torch/_export/db/examples/specialized_attribute.py
torch/_export/db/examples/static_for_loop.py
torch/_export/db/examples/static_if.py
torch/_export/db/examples/tensor_setattr.py
torch/_export/db/examples/type_reflection_method.py
torch/_export/pass_infra/__init__.py
torch/_export/pass_infra/node_metadata.py
torch/_export/pass_infra/proxy_value.py
torch/_export/passes/__init__.py
torch/_export/passes/add_runtime_assertions_for_constraints_pass.py
torch/_export/passes/collect_tracepoints_pass.py
torch/_export/passes/functionalize_side_effectful_ops_pass.py
torch/_export/passes/lift_constant_tensor_pass.py
torch/_export/passes/replace_sym_size_ops_pass.py
torch/_export/passes/replace_view_ops_with_view_copy_ops_pass.py
torch/_export/serde/__init__.py
torch/_export/serde/schema.py
torch/_export/serde/serialize.py
torch/_export/serde/upgrade.py
torch/_functorch/__init__.py
torch/_functorch/aot_autograd.py
torch/_functorch/apis.py
torch/_functorch/autograd_function.py
torch/_functorch/batch_norm_replacement.py
torch/_functorch/benchmark_utils.py
torch/_functorch/compile_utils.py
torch/_functorch/compilers.py
torch/_functorch/config.py
torch/_functorch/deprecated.py
torch/_functorch/eager_transforms.py
torch/_functorch/functional_call.py
torch/_functorch/fx_minifier.py
torch/_functorch/make_functional.py
torch/_functorch/partitioners.py
torch/_functorch/pyfunctorch.py
torch/_functorch/python_key.py
torch/_functorch/pytree_hacks.py
torch/_functorch/top_operators_github_usage.py
torch/_functorch/utils.py
torch/_functorch/vmap.py
torch/_higher_order_ops/__init__.py
torch/_higher_order_ops/cond.py
torch/_higher_order_ops/out_dtype.py
torch/_higher_order_ops/utils.py
torch/_higher_order_ops/wrap.py
torch/_inductor/__init__.py
torch/_inductor/autotune_process.py
torch/_inductor/bounds.py
torch/_inductor/codecache.py
torch/_inductor/compile_fx.py
torch/_inductor/config.py
torch/_inductor/constant_folding.py
torch/_inductor/coordinate_descent_tuner.py
torch/_inductor/cuda_properties.py
torch/_inductor/cudagraph_trees.py
torch/_inductor/debug.py
torch/_inductor/decomposition.py
torch/_inductor/dependencies.py
torch/_inductor/exc.py
torch/_inductor/freezing.py
torch/_inductor/fx_utils.py
torch/_inductor/graph.py
torch/_inductor/hooks.py
torch/_inductor/index_propagation.py
torch/_inductor/inductor_prims.py
torch/_inductor/ir.py
torch/_inductor/lowering.py
torch/_inductor/metrics.py
torch/_inductor/optimize_indexing.py
torch/_inductor/pattern_matcher.py
torch/_inductor/quantized_lowerings.py
torch/_inductor/scheduler.py
torch/_inductor/select_algorithm.py
torch/_inductor/sizevars.py
torch/_inductor/test_operators.py
torch/_inductor/triton_helpers.py
torch/_inductor/triton_heuristics.py
torch/_inductor/utils.py
torch/_inductor/virtualized.py
torch/_inductor/wrapper_benchmark.py
torch/_inductor/codegen/__init__.py
torch/_inductor/codegen/common.py
torch/_inductor/codegen/cpp.py
torch/_inductor/codegen/triton.py
torch/_inductor/codegen/triton_foreach.py
torch/_inductor/codegen/triton_utils.py
torch/_inductor/codegen/wrapper.py
torch/_inductor/fx_passes/__init__.py
torch/_inductor/fx_passes/binary_folding.py
torch/_inductor/fx_passes/freezing_patterns.py
torch/_inductor/fx_passes/fuse_attention.py
torch/_inductor/fx_passes/group_batch_fusion.py
torch/_inductor/fx_passes/joint_graph.py
torch/_inductor/fx_passes/mkldnn_fusion.py
torch/_inductor/fx_passes/pad_mm.py
torch/_inductor/fx_passes/post_grad.py
torch/_inductor/fx_passes/pre_grad.py
torch/_inductor/fx_passes/quantization.py
torch/_inductor/fx_passes/replace_random.py
torch/_inductor/fx_passes/split_cat.py
torch/_inductor/kernel/__init__.py
torch/_inductor/kernel/bmm.py
torch/_inductor/kernel/conv.py
torch/_inductor/kernel/mm.py
torch/_inductor/kernel/mm_common.py
torch/_inductor/kernel/mm_plus_mm.py
torch/_inductor/kernel/unpack_mixed_mm.py
torch/_lazy/__init__.py
torch/_lazy/closure.py
torch/_lazy/computation.py
torch/_lazy/config.py
torch/_lazy/debug.py
torch/_lazy/device_context.py
torch/_lazy/extract_compiled_graph.py
torch/_lazy/ir_cache.py
torch/_lazy/metrics.py
torch/_lazy/tensor_factory_functions.py
torch/_lazy/ts_backend.py
torch/_logging/__init__.py
torch/_logging/_internal.py
torch/_logging/_registrations.py
torch/_numpy/__init__.py
torch/_numpy/_binary_ufuncs_impl.py
torch/_numpy/_casting_dicts.py
torch/_numpy/_dtypes.py
torch/_numpy/_dtypes_impl.py
torch/_numpy/_funcs.py
torch/_numpy/_funcs_impl.py
torch/_numpy/_getlimits.py
torch/_numpy/_ndarray.py
torch/_numpy/_normalizations.py
torch/_numpy/_reductions_impl.py
torch/_numpy/_ufuncs.py
torch/_numpy/_unary_ufuncs_impl.py
torch/_numpy/_util.py
torch/_numpy/fft.py
torch/_numpy/linalg.py
torch/_numpy/random.py
torch/_numpy/testing/__init__.py
torch/_numpy/testing/utils.py
torch/_prims/__init__.py
torch/_prims/context.py
torch/_prims/debug_prims.py
torch/_prims/executor.py
torch/_prims/rng_prims.py
torch/_prims_common/__init__.py
torch/_prims_common/wrappers.py
torch/_refs/__init__.py
torch/_refs/_conversions.py
torch/_refs/fft.py
torch/_refs/linalg/__init__.py
torch/_refs/nn/__init__.py
torch/_refs/nn/functional/__init__.py
torch/_refs/special/__init__.py
torch/_subclasses/__init__.py
torch/_subclasses/fake_tensor.py
torch/_subclasses/fake_utils.py
torch/_subclasses/meta_utils.py
torch/_subclasses/schema_check_mode.py
torch/amp/__init__.py
torch/amp/autocast_mode.py
torch/ao/__init__.py
torch/ao/nn/__init__.py
torch/ao/nn/intrinsic/__init__.py
torch/ao/nn/intrinsic/modules/__init__.py
torch/ao/nn/intrinsic/modules/fused.py
torch/ao/nn/intrinsic/qat/__init__.py
torch/ao/nn/intrinsic/qat/modules/__init__.py
torch/ao/nn/intrinsic/qat/modules/conv_fused.py
torch/ao/nn/intrinsic/qat/modules/linear_fused.py
torch/ao/nn/intrinsic/qat/modules/linear_relu.py
torch/ao/nn/intrinsic/quantized/__init__.py
torch/ao/nn/intrinsic/quantized/dynamic/__init__.py
torch/ao/nn/intrinsic/quantized/dynamic/modules/__init__.py
torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.py
torch/ao/nn/intrinsic/quantized/modules/__init__.py
torch/ao/nn/intrinsic/quantized/modules/bn_relu.py
torch/ao/nn/intrinsic/quantized/modules/conv_add.py
torch/ao/nn/intrinsic/quantized/modules/conv_relu.py
torch/ao/nn/intrinsic/quantized/modules/linear_relu.py
torch/ao/nn/qat/__init__.py
torch/ao/nn/qat/dynamic/__init__.py
torch/ao/nn/qat/dynamic/modules/__init__.py
torch/ao/nn/qat/dynamic/modules/linear.py
torch/ao/nn/qat/modules/__init__.py
torch/ao/nn/qat/modules/conv.py
torch/ao/nn/qat/modules/embedding_ops.py
torch/ao/nn/qat/modules/linear.py
torch/ao/nn/quantizable/__init__.py
torch/ao/nn/quantizable/modules/__init__.py
torch/ao/nn/quantizable/modules/activation.py
torch/ao/nn/quantizable/modules/rnn.py
torch/ao/nn/quantized/__init__.py
torch/ao/nn/quantized/functional.py
torch/ao/nn/quantized/dynamic/__init__.py
torch/ao/nn/quantized/dynamic/modules/__init__.py
torch/ao/nn/quantized/dynamic/modules/conv.py
torch/ao/nn/quantized/dynamic/modules/linear.py
torch/ao/nn/quantized/dynamic/modules/rnn.py
torch/ao/nn/quantized/modules/__init__.py
torch/ao/nn/quantized/modules/activation.py
torch/ao/nn/quantized/modules/batchnorm.py
torch/ao/nn/quantized/modules/conv.py
torch/ao/nn/quantized/modules/dropout.py
torch/ao/nn/quantized/modules/embedding_ops.py
torch/ao/nn/quantized/modules/functional_modules.py
torch/ao/nn/quantized/modules/linear.py
torch/ao/nn/quantized/modules/normalization.py
torch/ao/nn/quantized/modules/rnn.py
torch/ao/nn/quantized/modules/utils.py
torch/ao/nn/quantized/reference/__init__.py
torch/ao/nn/quantized/reference/modules/__init__.py
torch/ao/nn/quantized/reference/modules/conv.py
torch/ao/nn/quantized/reference/modules/linear.py
torch/ao/nn/quantized/reference/modules/rnn.py
torch/ao/nn/quantized/reference/modules/sparse.py
torch/ao/nn/quantized/reference/modules/utils.py
torch/ao/nn/sparse/__init__.py
torch/ao/nn/sparse/quantized/__init__.py
torch/ao/nn/sparse/quantized/linear.py
torch/ao/nn/sparse/quantized/utils.py
torch/ao/nn/sparse/quantized/dynamic/__init__.py
torch/ao/nn/sparse/quantized/dynamic/linear.py
torch/ao/ns/__init__.py
torch/ao/ns/_numeric_suite.py
torch/ao/ns/_numeric_suite_fx.py
torch/ao/ns/fx/__init__.py
torch/ao/ns/fx/graph_matcher.py
torch/ao/ns/fx/graph_passes.py
torch/ao/ns/fx/mappings.py
torch/ao/ns/fx/n_shadows_utils.py
torch/ao/ns/fx/ns_types.py
torch/ao/ns/fx/pattern_utils.py
torch/ao/ns/fx/qconfig_multi_mapping.py
torch/ao/ns/fx/utils.py
torch/ao/ns/fx/weight_utils.py
torch/ao/pruning/__init__.py
torch/ao/pruning/_mappings.py
torch/ao/pruning/_experimental/__init__.py
torch/ao/pruning/_experimental/activation_sparsifier/__init__.py
torch/ao/pruning/_experimental/activation_sparsifier/activation_sparsifier.py
torch/ao/pruning/_experimental/data_scheduler/__init__.py
torch/ao/pruning/_experimental/data_scheduler/base_data_scheduler.py
torch/ao/pruning/_experimental/data_sparsifier/__init__.py
torch/ao/pruning/_experimental/data_sparsifier/base_data_sparsifier.py
torch/ao/pruning/_experimental/data_sparsifier/data_norm_sparsifier.py
torch/ao/pruning/_experimental/data_sparsifier/quantization_utils.py
torch/ao/pruning/_experimental/data_sparsifier/lightning/__init__.py
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/__init__.py
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/_data_sparstity_utils.py
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/data_sparsity.py
torch/ao/pruning/_experimental/pruner/FPGM_pruner.py
torch/ao/pruning/_experimental/pruner/__init__.py
torch/ao/pruning/_experimental/pruner/base_structured_sparsifier.py
torch/ao/pruning/_experimental/pruner/lstm_saliency_pruner.py
torch/ao/pruning/_experimental/pruner/match_utils.py
torch/ao/pruning/_experimental/pruner/parametrization.py
torch/ao/pruning/_experimental/pruner/prune_functions.py
torch/ao/pruning/_experimental/pruner/saliency_pruner.py
torch/ao/pruning/scheduler/__init__.py
torch/ao/pruning/scheduler/base_scheduler.py
torch/ao/pruning/scheduler/cubic_scheduler.py
torch/ao/pruning/scheduler/lambda_scheduler.py
torch/ao/pruning/sparsifier/__init__.py
torch/ao/pruning/sparsifier/base_sparsifier.py
torch/ao/pruning/sparsifier/nearly_diagonal_sparsifier.py
torch/ao/pruning/sparsifier/utils.py
torch/ao/pruning/sparsifier/weight_norm_sparsifier.py
torch/ao/quantization/__init__.py
torch/ao/quantization/_correct_bias.py
torch/ao/quantization/_equalize.py
torch/ao/quantization/_learnable_fake_quantize.py
torch/ao/quantization/fake_quantize.py
torch/ao/quantization/fuse_modules.py
torch/ao/quantization/fuser_method_mappings.py
torch/ao/quantization/observer.py
torch/ao/quantization/qconfig.py
torch/ao/quantization/qconfig_mapping.py
torch/ao/quantization/quant_type.py
torch/ao/quantization/quantization_mappings.py
torch/ao/quantization/quantize.py
torch/ao/quantization/quantize_fx.py
torch/ao/quantization/quantize_jit.py
torch/ao/quantization/quantize_pt2e.py
torch/ao/quantization/stubs.py
torch/ao/quantization/utils.py
torch/ao/quantization/backend_config/__init__.py
torch/ao/quantization/backend_config/_common_operator_config_utils.py
torch/ao/quantization/backend_config/_qnnpack_pt2e.py
torch/ao/quantization/backend_config/backend_config.py
torch/ao/quantization/backend_config/executorch.py
torch/ao/quantization/backend_config/fbgemm.py
torch/ao/quantization/backend_config/native.py
torch/ao/quantization/backend_config/observation_type.py
torch/ao/quantization/backend_config/onednn.py
torch/ao/quantization/backend_config/qnnpack.py
torch/ao/quantization/backend_config/tensorrt.py
torch/ao/quantization/backend_config/utils.py
torch/ao/quantization/backend_config/x86.py
torch/ao/quantization/fx/__init__.py
torch/ao/quantization/fx/_decomposed.py
torch/ao/quantization/fx/_equalize.py
torch/ao/quantization/fx/_lower_to_native_backend.py
torch/ao/quantization/fx/convert.py
torch/ao/quantization/fx/custom_config.py
torch/ao/quantization/fx/fuse.py
torch/ao/quantization/fx/fuse_handler.py
torch/ao/quantization/fx/graph_module.py
torch/ao/quantization/fx/lower_to_fbgemm.py
torch/ao/quantization/fx/lower_to_qnnpack.py
torch/ao/quantization/fx/lstm_utils.py
torch/ao/quantization/fx/match_utils.py
torch/ao/quantization/fx/pattern_utils.py
torch/ao/quantization/fx/prepare.py
torch/ao/quantization/fx/qconfig_mapping_utils.py
torch/ao/quantization/fx/quantize_handler.py
torch/ao/quantization/fx/tracer.py
torch/ao/quantization/fx/utils.py
torch/ao/quantization/fx/_model_report/__init__.py
torch/ao/quantization/fx/_model_report/detector.py
torch/ao/quantization/fx/_model_report/model_report.py
torch/ao/quantization/fx/_model_report/model_report_observer.py
torch/ao/quantization/fx/_model_report/model_report_visualizer.py
torch/ao/quantization/pt2e/__init__.py
torch/ao/quantization/pt2e/duplicate_dq_pass.py
torch/ao/quantization/pt2e/eval_utils.py
torch/ao/quantization/pt2e/graph_utils.py
torch/ao/quantization/pt2e/port_metadata_pass.py
torch/ao/quantization/pt2e/prepare.py
torch/ao/quantization/pt2e/qat_utils.py
torch/ao/quantization/pt2e/utils.py
torch/ao/quantization/pt2e/representation/__init__.py
torch/ao/quantization/pt2e/representation/rewrite.py
torch/ao/quantization/quantizer/__init__.py
torch/ao/quantization/quantizer/composable_quantizer.py
torch/ao/quantization/quantizer/embedding_quantizer.py
torch/ao/quantization/quantizer/quantizer.py
torch/ao/quantization/quantizer/utils.py
torch/ao/quantization/quantizer/x86_inductor_quantizer.py
torch/ao/quantization/quantizer/xnnpack_quantizer.py
torch/ao/quantization/quantizer/xnnpack_quantizer_utils.py
torch/autograd/__init__.py
torch/autograd/anomaly_mode.py
torch/autograd/forward_ad.py
torch/autograd/function.py
torch/autograd/functional.py
torch/autograd/grad_mode.py
torch/autograd/gradcheck.py
torch/autograd/graph.py
torch/autograd/profiler.py
torch/autograd/profiler_legacy.py
torch/autograd/profiler_util.py
torch/autograd/variable.py
torch/autograd/_functions/__init__.py
torch/autograd/_functions/tensor.py
torch/autograd/_functions/utils.py
torch/backends/__init__.py
torch/backends/_coreml/__init__.py
torch/backends/_coreml/preprocess.py
torch/backends/_nnapi/__init__.py
torch/backends/_nnapi/prepare.py
torch/backends/_nnapi/serializer.py
torch/backends/cpu/__init__.py
torch/backends/cuda/__init__.py
torch/backends/cudnn/__init__.py
torch/backends/cudnn/rnn.py
torch/backends/mkl/__init__.py
torch/backends/mkldnn/__init__.py
torch/backends/mps/__init__.py
torch/backends/openmp/__init__.py
torch/backends/opt_einsum/__init__.py
torch/backends/quantized/__init__.py
torch/backends/xeon/__init__.py
torch/backends/xeon/run_cpu.py
torch/backends/xnnpack/__init__.py
torch/compiler/__init__.py
torch/contrib/__init__.py
torch/contrib/_tensorboard_vis.py
torch/cpu/__init__.py
torch/cpu/amp/__init__.py
torch/cpu/amp/autocast_mode.py
torch/cuda/__init__.py
torch/cuda/_memory_viz.py
torch/cuda/_sanitizer.py
torch/cuda/_utils.py
torch/cuda/comm.py
torch/cuda/error.py
torch/cuda/graphs.py
torch/cuda/jiterator.py
torch/cuda/memory.py
torch/cuda/nccl.py
torch/cuda/nvtx.py
torch/cuda/profiler.py
torch/cuda/random.py
torch/cuda/sparse.py
torch/cuda/streams.py
torch/cuda/amp/__init__.py
torch/cuda/amp/autocast_mode.py
torch/cuda/amp/common.py
torch/cuda/amp/grad_scaler.py
torch/distributed/__init__.py
torch/distributed/_composable_state.py
torch/distributed/_functional_collectives.py
torch/distributed/_functional_collectives_impl.py
torch/distributed/argparse_util.py
torch/distributed/c10d_logger.py
torch/distributed/collective_utils.py
torch/distributed/constants.py
torch/distributed/distributed_c10d.py
torch/distributed/launch.py
torch/distributed/logging_handlers.py
torch/distributed/remote_device.py
torch/distributed/rendezvous.py
torch/distributed/run.py
torch/distributed/utils.py
torch/distributed/_composable/__init__.py
torch/distributed/_composable/checkpoint_activation.py
torch/distributed/_composable/contract.py
torch/distributed/_composable/fully_shard.py
torch/distributed/_composable/replicate.py
torch/distributed/_shard/__init__.py
torch/distributed/_shard/_utils.py
torch/distributed/_shard/api.py
torch/distributed/_shard/common_op_utils.py
torch/distributed/_shard/metadata.py
torch/distributed/_shard/op_registry_utils.py
torch/distributed/_shard/sharder.py
torch/distributed/_shard/checkpoint/__init__.py
torch/distributed/_shard/sharded_optim/__init__.py
torch/distributed/_shard/sharded_optim/api.py
torch/distributed/_shard/sharded_tensor/__init__.py
torch/distributed/_shard/sharded_tensor/api.py
torch/distributed/_shard/sharded_tensor/logger.py
torch/distributed/_shard/sharded_tensor/logging_handlers.py
torch/distributed/_shard/sharded_tensor/metadata.py
torch/distributed/_shard/sharded_tensor/reshard.py
torch/distributed/_shard/sharded_tensor/shard.py
torch/distributed/_shard/sharded_tensor/utils.py
torch/distributed/_shard/sharded_tensor/_ops/__init__.py
torch/distributed/_shard/sharded_tensor/_ops/_common.py
torch/distributed/_shard/sharded_tensor/_ops/binary_cmp.py
torch/distributed/_shard/sharded_tensor/_ops/init.py
torch/distributed/_shard/sharded_tensor/_ops/misc_ops.py
torch/distributed/_shard/sharded_tensor/_ops/tensor_ops.py
torch/distributed/_shard/sharding_plan/__init__.py
torch/distributed/_shard/sharding_plan/api.py
torch/distributed/_shard/sharding_spec/__init__.py
torch/distributed/_shard/sharding_spec/_internals.py
torch/distributed/_shard/sharding_spec/api.py
torch/distributed/_shard/sharding_spec/chunk_sharding_spec.py
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/__init__.py
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/_common.py
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/embedding.py
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/embedding_bag.py
torch/distributed/_sharded_tensor/__init__.py
torch/distributed/_sharding_spec/__init__.py
torch/distributed/_spmd/__init__.py
torch/distributed/_spmd/api.py
torch/distributed/_spmd/batch_dim_utils.py
torch/distributed/_spmd/comm_tensor.py
torch/distributed/_spmd/config.py
torch/distributed/_spmd/data_parallel.py
torch/distributed/_spmd/distribute.py
torch/distributed/_spmd/experimental_ops.py
torch/distributed/_spmd/gm_transformation.py
torch/distributed/_spmd/graph_optimization.py
torch/distributed/_spmd/graph_utils.py
torch/distributed/_spmd/iter_graph_module.py
torch/distributed/_spmd/log_utils.py
torch/distributed/_spmd/parallel_mode.py
torch/distributed/_spmd/partial_lower.py
torch/distributed/_tensor/__init__.py
torch/distributed/_tensor/_collective_utils.py
torch/distributed/_tensor/_dynamo_utils.py
torch/distributed/_tensor/_utils.py
torch/distributed/_tensor/api.py
torch/distributed/_tensor/device_mesh.py
torch/distributed/_tensor/dispatch.py
torch/distributed/_tensor/op_schema.py
torch/distributed/_tensor/placement_types.py
torch/distributed/_tensor/random.py
torch/distributed/_tensor/redistribute.py
torch/distributed/_tensor/sharding_prop.py
torch/distributed/_tensor/debug/__init__.py
torch/distributed/_tensor/debug/op_coverage.py
torch/distributed/_tensor/ops/__init__.py
torch/distributed/_tensor/ops/basic_strategy.py
torch/distributed/_tensor/ops/common_rules.py
torch/distributed/_tensor/ops/embedding_ops.py
torch/distributed/_tensor/ops/math_ops.py
torch/distributed/_tensor/ops/matrix_ops.py
torch/distributed/_tensor/ops/pointwise_ops.py
torch/distributed/_tensor/ops/random_ops.py
torch/distributed/_tensor/ops/tensor_ops.py
torch/distributed/_tensor/ops/utils.py
torch/distributed/_tensor/ops/view_ops.py
torch/distributed/_tools/__init__.py
torch/distributed/_tools/memory_tracker.py
torch/distributed/algorithms/__init__.py
torch/distributed/algorithms/join.py
torch/distributed/algorithms/_checkpoint/__init__.py
torch/distributed/algorithms/_checkpoint/checkpoint_wrapper.py
torch/distributed/algorithms/_comm_hooks/__init__.py
torch/distributed/algorithms/_comm_hooks/default_hooks.py
torch/distributed/algorithms/_optimizer_overlap/__init__.py
torch/distributed/algorithms/_optimizer_overlap/optimizer_overlap.py
torch/distributed/algorithms/_quantization/__init__.py
torch/distributed/algorithms/_quantization/quantization.py
torch/distributed/algorithms/ddp_comm_hooks/__init__.py
torch/distributed/algorithms/ddp_comm_hooks/ddp_zero_hook.py
torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.py
torch/distributed/algorithms/ddp_comm_hooks/default_hooks.py
torch/distributed/algorithms/ddp_comm_hooks/mixed_precision_hooks.py
torch/distributed/algorithms/ddp_comm_hooks/optimizer_overlap_hooks.py
torch/distributed/algorithms/ddp_comm_hooks/post_localSGD_hook.py
torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.py
torch/distributed/algorithms/ddp_comm_hooks/quantization_hooks.py
torch/distributed/algorithms/model_averaging/__init__.py
torch/distributed/algorithms/model_averaging/averagers.py
torch/distributed/algorithms/model_averaging/hierarchical_model_averager.py
torch/distributed/algorithms/model_averaging/utils.py
torch/distributed/autograd/__init__.py
torch/distributed/checkpoint/__init__.py
torch/distributed/checkpoint/_dedup_tensors.py
torch/distributed/checkpoint/_fsspec_filesystem.py
torch/distributed/checkpoint/_nested_dict.py
torch/distributed/checkpoint/_sharded_tensor_utils.py
torch/distributed/checkpoint/_traverse.py
torch/distributed/checkpoint/api.py
torch/distributed/checkpoint/default_planner.py
torch/distributed/checkpoint/filesystem.py
torch/distributed/checkpoint/metadata.py
torch/distributed/checkpoint/optimizer.py
torch/distributed/checkpoint/planner.py
torch/distributed/checkpoint/planner_helpers.py
torch/distributed/checkpoint/resharding.py
torch/distributed/checkpoint/state_dict_loader.py
torch/distributed/checkpoint/state_dict_saver.py
torch/distributed/checkpoint/storage.py
torch/distributed/checkpoint/utils.py
torch/distributed/elastic/__init__.py
torch/distributed/elastic/agent/__init__.py
torch/distributed/elastic/agent/server/__init__.py
torch/distributed/elastic/agent/server/api.py
torch/distributed/elastic/agent/server/local_elastic_agent.py
torch/distributed/elastic/events/__init__.py
torch/distributed/elastic/events/api.py
torch/distributed/elastic/events/handlers.py
torch/distributed/elastic/metrics/__init__.py
torch/distributed/elastic/metrics/api.py
torch/distributed/elastic/multiprocessing/__init__.py
torch/distributed/elastic/multiprocessing/api.py
torch/distributed/elastic/multiprocessing/redirects.py
torch/distributed/elastic/multiprocessing/tail_log.py
torch/distributed/elastic/multiprocessing/errors/__init__.py
torch/distributed/elastic/multiprocessing/errors/error_handler.py
torch/distributed/elastic/multiprocessing/errors/handlers.py
torch/distributed/elastic/rendezvous/__init__.py
torch/distributed/elastic/rendezvous/api.py
torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.py
torch/distributed/elastic/rendezvous/dynamic_rendezvous.py
torch/distributed/elastic/rendezvous/etcd_rendezvous.py
torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.py
torch/distributed/elastic/rendezvous/etcd_server.py
torch/distributed/elastic/rendezvous/etcd_store.py
torch/distributed/elastic/rendezvous/registry.py
torch/distributed/elastic/rendezvous/static_tcp_rendezvous.py
torch/distributed/elastic/rendezvous/utils.py
torch/distributed/elastic/timer/__init__.py
torch/distributed/elastic/timer/api.py
torch/distributed/elastic/timer/file_based_local_timer.py
torch/distributed/elastic/timer/local_timer.py
torch/distributed/elastic/utils/__init__.py
torch/distributed/elastic/utils/api.py
torch/distributed/elastic/utils/distributed.py
torch/distributed/elastic/utils/log_level.py
torch/distributed/elastic/utils/logging.py
torch/distributed/elastic/utils/store.py
torch/distributed/elastic/utils/data/__init__.py
torch/distributed/elastic/utils/data/cycling_iterator.py
torch/distributed/elastic/utils/data/elastic_distributed_sampler.py
torch/distributed/fsdp/__init__.py
torch/distributed/fsdp/_common_utils.py
torch/distributed/fsdp/_debug_utils.py
torch/distributed/fsdp/_dynamo_utils.py
torch/distributed/fsdp/_exec_order_utils.py
torch/distributed/fsdp/_fsdp_extensions.py
torch/distributed/fsdp/_init_utils.py
torch/distributed/fsdp/_limiter_utils.py
torch/distributed/fsdp/_optim_utils.py
torch/distributed/fsdp/_runtime_utils.py
torch/distributed/fsdp/_shard_utils.py
torch/distributed/fsdp/_state_dict_utils.py
torch/distributed/fsdp/_trace_utils.py
torch/distributed/fsdp/_traversal_utils.py
torch/distributed/fsdp/_unshard_param_utils.py
torch/distributed/fsdp/_wrap_utils.py
torch/distributed/fsdp/api.py
torch/distributed/fsdp/flat_param.py
torch/distributed/fsdp/fully_sharded_data_parallel.py
torch/distributed/fsdp/sharded_grad_scaler.py
torch/distributed/fsdp/wrap.py
torch/distributed/launcher/__init__.py
torch/distributed/launcher/api.py
torch/distributed/nn/__init__.py
torch/distributed/nn/functional.py
torch/distributed/nn/api/__init__.py
torch/distributed/nn/api/remote_module.py
torch/distributed/nn/jit/__init__.py
torch/distributed/nn/jit/instantiator.py
torch/distributed/nn/jit/templates/__init__.py
torch/distributed/nn/jit/templates/remote_module_template.py
torch/distributed/optim/__init__.py
torch/distributed/optim/apply_optimizer_in_backward.py
torch/distributed/optim/functional_adadelta.py
torch/distributed/optim/functional_adagrad.py
torch/distributed/optim/functional_adam.py
torch/distributed/optim/functional_adamax.py
torch/distributed/optim/functional_adamw.py
torch/distributed/optim/functional_rmsprop.py
torch/distributed/optim/functional_rprop.py
torch/distributed/optim/functional_sgd.py
torch/distributed/optim/named_optimizer.py
torch/distributed/optim/optimizer.py
torch/distributed/optim/post_localSGD_optimizer.py
torch/distributed/optim/utils.py
torch/distributed/optim/zero_redundancy_optimizer.py
torch/distributed/pipeline/__init__.py
torch/distributed/pipeline/sync/__init__.py
torch/distributed/pipeline/sync/batchnorm.py
torch/distributed/pipeline/sync/checkpoint.py
torch/distributed/pipeline/sync/copy.py
torch/distributed/pipeline/sync/dependency.py
torch/distributed/pipeline/sync/microbatch.py
torch/distributed/pipeline/sync/phony.py
torch/distributed/pipeline/sync/pipe.py
torch/distributed/pipeline/sync/pipeline.py
torch/distributed/pipeline/sync/stream.py
torch/distributed/pipeline/sync/utils.py
torch/distributed/pipeline/sync/worker.py
torch/distributed/pipeline/sync/_balance/__init__.py
torch/distributed/pipeline/sync/_balance/blockpartition.py
torch/distributed/pipeline/sync/_balance/profile.py
torch/distributed/pipeline/sync/skip/__init__.py
torch/distributed/pipeline/sync/skip/layout.py
torch/distributed/pipeline/sync/skip/namespace.py
torch/distributed/pipeline/sync/skip/portal.py
torch/distributed/pipeline/sync/skip/skippable.py
torch/distributed/pipeline/sync/skip/tracker.py
torch/distributed/rpc/__init__.py
torch/distributed/rpc/_utils.py
torch/distributed/rpc/api.py
torch/distributed/rpc/backend_registry.py
torch/distributed/rpc/constants.py
torch/distributed/rpc/functions.py
torch/distributed/rpc/internal.py
torch/distributed/rpc/options.py
torch/distributed/rpc/rref_proxy.py
torch/distributed/rpc/server_process_global_profiler.py
torch/distributed/rpc/_testing/__init__.py
torch/distributed/rpc/_testing/faulty_agent_backend_registry.py
torch/distributed/tensor/__init__.py
torch/distributed/tensor/parallel/__init__.py
torch/distributed/tensor/parallel/_data_parallel_utils.py
torch/distributed/tensor/parallel/_utils.py
torch/distributed/tensor/parallel/_view_with_dim_change.py
torch/distributed/tensor/parallel/api.py
torch/distributed/tensor/parallel/ddp.py
torch/distributed/tensor/parallel/fsdp.py
torch/distributed/tensor/parallel/input_reshard.py
torch/distributed/tensor/parallel/style.py
torch/distributions/__init__.py
torch/distributions/bernoulli.py
torch/distributions/beta.py
torch/distributions/binomial.py
torch/distributions/categorical.py
torch/distributions/cauchy.py
torch/distributions/chi2.py
torch/distributions/constraint_registry.py
torch/distributions/constraints.py
torch/distributions/continuous_bernoulli.py
torch/distributions/dirichlet.py
torch/distributions/distribution.py
torch/distributions/exp_family.py
torch/distributions/exponential.py
torch/distributions/fishersnedecor.py
torch/distributions/gamma.py
torch/distributions/geometric.py
torch/distributions/gumbel.py
torch/distributions/half_cauchy.py
torch/distributions/half_normal.py
torch/distributions/independent.py
torch/distributions/kl.py
torch/distributions/kumaraswamy.py
torch/distributions/laplace.py
torch/distributions/lkj_cholesky.py
torch/distributions/log_normal.py
torch/distributions/logistic_normal.py
torch/distributions/lowrank_multivariate_normal.py
torch/distributions/mixture_same_family.py
torch/distributions/multinomial.py
torch/distributions/multivariate_normal.py
torch/distributions/negative_binomial.py
torch/distributions/normal.py
torch/distributions/one_hot_categorical.py
torch/distributions/pareto.py
torch/distributions/poisson.py
torch/distributions/relaxed_bernoulli.py
torch/distributions/relaxed_categorical.py
torch/distributions/studentT.py
torch/distributions/transformed_distribution.py
torch/distributions/transforms.py
torch/distributions/uniform.py
torch/distributions/utils.py
torch/distributions/von_mises.py
torch/distributions/weibull.py
torch/distributions/wishart.py
torch/export/__init__.py
torch/fft/__init__.py
torch/func/__init__.py
torch/futures/__init__.py
torch/fx/__init__.py
torch/fx/_compatibility.py
torch/fx/_pytree.py
torch/fx/_symbolic_trace.py
torch/fx/annotate.py
torch/fx/config.py
torch/fx/graph.py
torch/fx/graph_module.py
torch/fx/immutable_collections.py
torch/fx/interpreter.py
torch/fx/node.py
torch/fx/operator_schemas.py
torch/fx/proxy.py
torch/fx/subgraph_rewriter.py
torch/fx/tensor_type.py
torch/fx/traceback.py
torch/fx/experimental/__init__.py
torch/fx/experimental/accelerator_partitioner.py
torch/fx/experimental/const_fold.py
torch/fx/experimental/debug.py
torch/fx/experimental/graph_gradual_typechecker.py
torch/fx/experimental/merge_matmul.py
torch/fx/experimental/meta_tracer.py
torch/fx/experimental/normalize.py
torch/fx/experimental/optimization.py
torch/fx/experimental/partitioner_utils.py
torch/fx/experimental/proxy_tensor.py
torch/fx/experimental/refinement_types.py
torch/fx/experimental/rewriter.py
torch/fx/experimental/schema_type_annotation.py
torch/fx/experimental/symbolic_shapes.py
torch/fx/experimental/unify_refinements.py
torch/fx/experimental/validator.py
torch/fx/experimental/migrate_gradual_types/__init__.py
torch/fx/experimental/migrate_gradual_types/constraint.py
torch/fx/experimental/migrate_gradual_types/constraint_generator.py
torch/fx/experimental/migrate_gradual_types/constraint_transformation.py
torch/fx/experimental/migrate_gradual_types/operation.py
torch/fx/experimental/migrate_gradual_types/transform_to_z3.py
torch/fx/experimental/migrate_gradual_types/util.py
torch/fx/experimental/migrate_gradual_types/z3_types.py
torch/fx/experimental/unification/__init__.py
torch/fx/experimental/unification/core.py
torch/fx/experimental/unification/dispatch.py
torch/fx/experimental/unification/match.py
torch/fx/experimental/unification/more.py
torch/fx/experimental/unification/unification_tools.py
torch/fx/experimental/unification/utils.py
torch/fx/experimental/unification/variable.py
torch/fx/experimental/unification/multipledispatch/__init__.py
torch/fx/experimental/unification/multipledispatch/conflict.py
torch/fx/experimental/unification/multipledispatch/core.py
torch/fx/experimental/unification/multipledispatch/dispatcher.py
torch/fx/experimental/unification/multipledispatch/utils.py
torch/fx/experimental/unification/multipledispatch/variadic.py
torch/fx/passes/__init__.py
torch/fx/passes/annotate_getitem_nodes.py
torch/fx/passes/fake_tensor_prop.py
torch/fx/passes/graph_drawer.py
torch/fx/passes/graph_manipulation.py
torch/fx/passes/net_min_base.py
torch/fx/passes/operator_support.py
torch/fx/passes/param_fetch.py
torch/fx/passes/pass_manager.py
torch/fx/passes/reinplace.py
torch/fx/passes/shape_prop.py
torch/fx/passes/split_module.py
torch/fx/passes/split_utils.py
torch/fx/passes/splitter_base.py
torch/fx/passes/tools_common.py
torch/fx/passes/backends/__init__.py
torch/fx/passes/backends/cudagraphs.py
torch/fx/passes/dialect/__init__.py
torch/fx/passes/dialect/common/__init__.py
torch/fx/passes/dialect/common/cse_pass.py
torch/fx/passes/infra/__init__.py
torch/fx/passes/infra/partitioner.py
torch/fx/passes/infra/pass_base.py
torch/fx/passes/infra/pass_manager.py
torch/fx/passes/tests/__init__.py
torch/fx/passes/tests/test_pass_manager.py
torch/fx/passes/utils/__init__.py
torch/fx/passes/utils/common.py
torch/fx/passes/utils/fuser_utils.py
torch/fx/passes/utils/matcher_utils.py
torch/fx/passes/utils/source_matcher_utils.py
torch/jit/__init__.py
torch/jit/_async.py
torch/jit/_await.py
torch/jit/_builtins.py
torch/jit/_check.py
torch/jit/_dataclass_impls.py
torch/jit/_decomposition_utils.py
torch/jit/_decompositions.py
torch/jit/_freeze.py
torch/jit/_fuser.py
torch/jit/_ir_utils.py
torch/jit/_logging.py
torch/jit/_monkeytype_config.py
torch/jit/_pickle.py
torch/jit/_recursive.py
torch/jit/_script.py
torch/jit/_serialization.py
torch/jit/_shape_functions.py
torch/jit/_state.py
torch/jit/_trace.py
torch/jit/annotations.py
torch/jit/frontend.py
torch/jit/generate_bytecode.py
torch/jit/quantized.py
torch/jit/supported_ops.py
torch/jit/unsupported_tensor_ops.py
torch/jit/_passes/__init__.py
torch/jit/_passes/_property_propagation.py
torch/jit/mobile/__init__.py
torch/linalg/__init__.py
torch/masked/__init__.py
torch/masked/_docs.py
torch/masked/_ops.py
torch/masked/maskedtensor/__init__.py
torch/masked/maskedtensor/_ops_refs.py
torch/masked/maskedtensor/binary.py
torch/masked/maskedtensor/core.py
torch/masked/maskedtensor/creation.py
torch/masked/maskedtensor/passthrough.py
torch/masked/maskedtensor/reductions.py
torch/masked/maskedtensor/unary.py
torch/monitor/__init__.py
torch/mps/__init__.py
torch/mps/event.py
torch/mps/profiler.py
torch/multiprocessing/__init__.py
torch/multiprocessing/_atfork.py
torch/multiprocessing/pool.py
torch/multiprocessing/queue.py
torch/multiprocessing/reductions.py
torch/multiprocessing/spawn.py
torch/nested/__init__.py
torch/nn/__init__.py
torch/nn/_reduction.py
torch/nn/common_types.py
torch/nn/cpp.py
torch/nn/functional.py
torch/nn/grad.py
torch/nn/init.py
torch/nn/parameter.py
torch/nn/backends/__init__.py
torch/nn/backends/thnn.py
torch/nn/intrinsic/__init__.py
torch/nn/intrinsic/modules/__init__.py
torch/nn/intrinsic/modules/fused.py
torch/nn/intrinsic/qat/__init__.py
torch/nn/intrinsic/qat/modules/__init__.py
torch/nn/intrinsic/qat/modules/conv_fused.py
torch/nn/intrinsic/qat/modules/linear_fused.py
torch/nn/intrinsic/qat/modules/linear_relu.py
torch/nn/intrinsic/quantized/__init__.py
torch/nn/intrinsic/quantized/dynamic/__init__.py
torch/nn/intrinsic/quantized/dynamic/modules/__init__.py
torch/nn/intrinsic/quantized/dynamic/modules/linear_relu.py
torch/nn/intrinsic/quantized/modules/__init__.py
torch/nn/intrinsic/quantized/modules/bn_relu.py
torch/nn/intrinsic/quantized/modules/conv_relu.py
torch/nn/intrinsic/quantized/modules/linear_relu.py
torch/nn/modules/__init__.py
torch/nn/modules/_functions.py
torch/nn/modules/activation.py
torch/nn/modules/adaptive.py
torch/nn/modules/batchnorm.py
torch/nn/modules/channelshuffle.py
torch/nn/modules/container.py
torch/nn/modules/conv.py
torch/nn/modules/distance.py
torch/nn/modules/dropout.py
torch/nn/modules/flatten.py
torch/nn/modules/fold.py
torch/nn/modules/instancenorm.py
torch/nn/modules/lazy.py
torch/nn/modules/linear.py
torch/nn/modules/loss.py
torch/nn/modules/module.py
torch/nn/modules/normalization.py
torch/nn/modules/padding.py
torch/nn/modules/pixelshuffle.py
torch/nn/modules/pooling.py
torch/nn/modules/rnn.py
torch/nn/modules/sparse.py
torch/nn/modules/transformer.py
torch/nn/modules/upsampling.py
torch/nn/modules/utils.py
torch/nn/parallel/__init__.py
torch/nn/parallel/_functions.py
torch/nn/parallel/comm.py
torch/nn/parallel/data_parallel.py
torch/nn/parallel/distributed.py
torch/nn/parallel/parallel_apply.py
torch/nn/parallel/replicate.py
torch/nn/parallel/scatter_gather.py
torch/nn/qat/__init__.py
torch/nn/qat/dynamic/__init__.py
torch/nn/qat/dynamic/modules/__init__.py
torch/nn/qat/dynamic/modules/linear.py
torch/nn/qat/modules/__init__.py
torch/nn/qat/modules/conv.py
torch/nn/qat/modules/embedding_ops.py
torch/nn/qat/modules/linear.py
torch/nn/quantizable/__init__.py
torch/nn/quantizable/modules/__init__.py
torch/nn/quantizable/modules/activation.py
torch/nn/quantizable/modules/rnn.py
torch/nn/quantized/__init__.py
torch/nn/quantized/functional.py
torch/nn/quantized/_reference/__init__.py
torch/nn/quantized/_reference/modules/__init__.py
torch/nn/quantized/_reference/modules/conv.py
torch/nn/quantized/_reference/modules/linear.py
torch/nn/quantized/_reference/modules/rnn.py
torch/nn/quantized/_reference/modules/sparse.py
torch/nn/quantized/_reference/modules/utils.py
torch/nn/quantized/dynamic/__init__.py
torch/nn/quantized/dynamic/modules/__init__.py
torch/nn/quantized/dynamic/modules/conv.py
torch/nn/quantized/dynamic/modules/linear.py
torch/nn/quantized/dynamic/modules/rnn.py
torch/nn/quantized/modules/__init__.py
torch/nn/quantized/modules/activation.py
torch/nn/quantized/modules/batchnorm.py
torch/nn/quantized/modules/conv.py
torch/nn/quantized/modules/dropout.py
torch/nn/quantized/modules/embedding_ops.py
torch/nn/quantized/modules/functional_modules.py
torch/nn/quantized/modules/linear.py
torch/nn/quantized/modules/normalization.py
torch/nn/quantized/modules/rnn.py
torch/nn/quantized/modules/utils.py
torch/nn/utils/__init__.py
torch/nn/utils/_deprecation_utils.py
torch/nn/utils/_named_member_accessor.py
torch/nn/utils/_per_sample_grad.py
torch/nn/utils/clip_grad.py
torch/nn/utils/convert_parameters.py
torch/nn/utils/fusion.py
torch/nn/utils/init.py
torch/nn/utils/memory_format.py
torch/nn/utils/parametrizations.py
torch/nn/utils/parametrize.py
torch/nn/utils/prune.py
torch/nn/utils/rnn.py
torch/nn/utils/spectral_norm.py
torch/nn/utils/stateless.py
torch/nn/utils/weight_norm.py
torch/nn/utils/_expanded_weights/__init__.py
torch/nn/utils/_expanded_weights/conv_expanded_weights.py
torch/nn/utils/_expanded_weights/conv_utils.py
torch/nn/utils/_expanded_weights/embedding_expanded_weights.py
torch/nn/utils/_expanded_weights/expanded_weights_impl.py
torch/nn/utils/_expanded_weights/expanded_weights_utils.py
torch/nn/utils/_expanded_weights/group_norm_expanded_weights.py
torch/nn/utils/_expanded_weights/instance_norm_expanded_weights.py
torch/nn/utils/_expanded_weights/layer_norm_expanded_weights.py
torch/nn/utils/_expanded_weights/linear_expanded_weights.py
torch/onnx/__init__.py
torch/onnx/_constants.py
torch/onnx/_deprecation.py
torch/onnx/_experimental.py
torch/onnx/_exporter_states.py
torch/onnx/_globals.py
torch/onnx/_onnx_supported_ops.py
torch/onnx/_type_utils.py
torch/onnx/errors.py
torch/onnx/operators.py
torch/onnx/symbolic_caffe2.py
torch/onnx/symbolic_helper.py
torch/onnx/symbolic_opset10.py
torch/onnx/symbolic_opset11.py
torch/onnx/symbolic_opset12.py
torch/onnx/symbolic_opset13.py
torch/onnx/symbolic_opset14.py
torch/onnx/symbolic_opset15.py
torch/onnx/symbolic_opset16.py
torch/onnx/symbolic_opset17.py
torch/onnx/symbolic_opset18.py
torch/onnx/symbolic_opset7.py
torch/onnx/symbolic_opset8.py
torch/onnx/symbolic_opset9.py
torch/onnx/utils.py
torch/onnx/verification.py
torch/onnx/_internal/__init__.py
torch/onnx/_internal/_beartype.py
torch/onnx/_internal/exporter.py
torch/onnx/_internal/io_adapter.py
torch/onnx/_internal/jit_utils.py
torch/onnx/_internal/onnx_proto_utils.py
torch/onnx/_internal/onnxruntime.py
torch/onnx/_internal/registration.py
torch/onnx/_internal/diagnostics/__init__.py
torch/onnx/_internal/diagnostics/_diagnostic.py
torch/onnx/_internal/diagnostics/_rules.py
torch/onnx/_internal/diagnostics/infra/__init__.py
torch/onnx/_internal/diagnostics/infra/_infra.py
torch/onnx/_internal/diagnostics/infra/context.py
torch/onnx/_internal/diagnostics/infra/decorator.py
torch/onnx/_internal/diagnostics/infra/formatter.py
torch/onnx/_internal/diagnostics/infra/utils.py
torch/onnx/_internal/diagnostics/infra/sarif/__init__.py
torch/onnx/_internal/diagnostics/infra/sarif/_address.py
torch/onnx/_internal/diagnostics/infra/sarif/_artifact.py
torch/onnx/_internal/diagnostics/infra/sarif/_artifact_change.py
torch/onnx/_internal/diagnostics/infra/sarif/_artifact_content.py
torch/onnx/_internal/diagnostics/infra/sarif/_artifact_location.py
torch/onnx/_internal/diagnostics/infra/sarif/_attachment.py
torch/onnx/_internal/diagnostics/infra/sarif/_code_flow.py
torch/onnx/_internal/diagnostics/infra/sarif/_configuration_override.py
torch/onnx/_internal/diagnostics/infra/sarif/_conversion.py
torch/onnx/_internal/diagnostics/infra/sarif/_edge.py
torch/onnx/_internal/diagnostics/infra/sarif/_edge_traversal.py
torch/onnx/_internal/diagnostics/infra/sarif/_exception.py
torch/onnx/_internal/diagnostics/infra/sarif/_external_properties.py
torch/onnx/_internal/diagnostics/infra/sarif/_external_property_file_reference.py
torch/onnx/_internal/diagnostics/infra/sarif/_external_property_file_references.py
torch/onnx/_internal/diagnostics/infra/sarif/_fix.py
torch/onnx/_internal/diagnostics/infra/sarif/_graph.py
torch/onnx/_internal/diagnostics/infra/sarif/_graph_traversal.py
torch/onnx/_internal/diagnostics/infra/sarif/_invocation.py
torch/onnx/_internal/diagnostics/infra/sarif/_location.py
torch/onnx/_internal/diagnostics/infra/sarif/_location_relationship.py
torch/onnx/_internal/diagnostics/infra/sarif/_logical_location.py
torch/onnx/_internal/diagnostics/infra/sarif/_message.py
torch/onnx/_internal/diagnostics/infra/sarif/_multiformat_message_string.py
torch/onnx/_internal/diagnostics/infra/sarif/_node.py
torch/onnx/_internal/diagnostics/infra/sarif/_notification.py
torch/onnx/_internal/diagnostics/infra/sarif/_physical_location.py
torch/onnx/_internal/diagnostics/infra/sarif/_property_bag.py
torch/onnx/_internal/diagnostics/infra/sarif/_rectangle.py
torch/onnx/_internal/diagnostics/infra/sarif/_region.py
torch/onnx/_internal/diagnostics/infra/sarif/_replacement.py
torch/onnx/_internal/diagnostics/infra/sarif/_reporting_configuration.py
torch/onnx/_internal/diagnostics/infra/sarif/_reporting_descriptor.py
torch/onnx/_internal/diagnostics/infra/sarif/_reporting_descriptor_reference.py
torch/onnx/_internal/diagnostics/infra/sarif/_reporting_descriptor_relationship.py
torch/onnx/_internal/diagnostics/infra/sarif/_result.py
torch/onnx/_internal/diagnostics/infra/sarif/_result_provenance.py
torch/onnx/_internal/diagnostics/infra/sarif/_run.py
torch/onnx/_internal/diagnostics/infra/sarif/_run_automation_details.py
torch/onnx/_internal/diagnostics/infra/sarif/_sarif_log.py
torch/onnx/_internal/diagnostics/infra/sarif/_special_locations.py
torch/onnx/_internal/diagnostics/infra/sarif/_stack.py
torch/onnx/_internal/diagnostics/infra/sarif/_stack_frame.py
torch/onnx/_internal/diagnostics/infra/sarif/_suppression.py
torch/onnx/_internal/diagnostics/infra/sarif/_thread_flow.py
torch/onnx/_internal/diagnostics/infra/sarif/_thread_flow_location.py
torch/onnx/_internal/diagnostics/infra/sarif/_tool.py
torch/onnx/_internal/diagnostics/infra/sarif/_tool_component.py
torch/onnx/_internal/diagnostics/infra/sarif/_tool_component_reference.py
torch/onnx/_internal/diagnostics/infra/sarif/_translation_metadata.py
torch/onnx/_internal/diagnostics/infra/sarif/_version_control_details.py
torch/onnx/_internal/diagnostics/infra/sarif/_web_request.py
torch/onnx/_internal/diagnostics/infra/sarif/_web_response.py
torch/onnx/_internal/diagnostics/infra/sarif/version.py
torch/onnx/_internal/fx/__init__.py
torch/onnx/_internal/fx/_pass.py
torch/onnx/_internal/fx/decomposition_table.py
torch/onnx/_internal/fx/diagnostics.py
torch/onnx/_internal/fx/dynamo_graph_extractor.py
torch/onnx/_internal/fx/fx_onnx_interpreter.py
torch/onnx/_internal/fx/fx_symbolic_graph_extractor.py
torch/onnx/_internal/fx/onnxfunction_dispatcher.py
torch/onnx/_internal/fx/op_validation.py
torch/onnx/_internal/fx/patcher.py
torch/onnx/_internal/fx/registration.py
torch/onnx/_internal/fx/serialization.py
torch/onnx/_internal/fx/type_utils.py
torch/onnx/_internal/fx/analysis/__init__.py
torch/onnx/_internal/fx/analysis/unsupported_nodes.py
torch/onnx/_internal/fx/passes/__init__.py
torch/onnx/_internal/fx/passes/_utils.py
torch/onnx/_internal/fx/passes/decomp.py
torch/onnx/_internal/fx/passes/functionalization.py
torch/onnx/_internal/fx/passes/modularization.py
torch/onnx/_internal/fx/passes/readability.py
torch/onnx/_internal/fx/passes/type_promotion.py
torch/onnx/_internal/fx/passes/virtualization.py
torch/optim/__init__.py
torch/optim/_functional.py
torch/optim/adadelta.py
torch/optim/adagrad.py
torch/optim/adam.py
torch/optim/adamax.py
torch/optim/adamw.py
torch/optim/asgd.py
torch/optim/lbfgs.py
torch/optim/lr_scheduler.py
torch/optim/nadam.py
torch/optim/optimizer.py
torch/optim/radam.py
torch/optim/rmsprop.py
torch/optim/rprop.py
torch/optim/sgd.py
torch/optim/sparse_adam.py
torch/optim/swa_utils.py
torch/optim/_multi_tensor/__init__.py
torch/package/__init__.py
torch/package/_digraph.py
torch/package/_directory_reader.py
torch/package/_importlib.py
torch/package/_mangling.py
torch/package/_mock.py
torch/package/_package_pickler.py
torch/package/_package_unpickler.py
torch/package/_stdlib.py
torch/package/file_structure_representation.py
torch/package/find_file_dependencies.py
torch/package/glob_group.py
torch/package/importer.py
torch/package/package_exporter.py
torch/package/package_importer.py
torch/package/analyze/__init__.py
torch/package/analyze/find_first_use_of_broken_modules.py
torch/package/analyze/is_from_package.py
torch/package/analyze/trace_dependencies.py
torch/profiler/__init__.py
torch/profiler/_memory_profiler.py
torch/profiler/_pattern_matcher.py
torch/profiler/_utils.py
torch/profiler/itt.py
torch/profiler/profiler.py
torch/profiler/python_tracer.py
torch/quantization/__init__.py
torch/quantization/_numeric_suite.py
torch/quantization/_numeric_suite_fx.py
torch/quantization/fake_quantize.py
torch/quantization/fuse_modules.py
torch/quantization/fuser_method_mappings.py
torch/quantization/observer.py
torch/quantization/qconfig.py
torch/quantization/quant_type.py
torch/quantization/quantization_mappings.py
torch/quantization/quantize.py
torch/quantization/quantize_fx.py
torch/quantization/quantize_jit.py
torch/quantization/stubs.py
torch/quantization/utils.py
torch/quantization/fx/__init__.py
torch/quantization/fx/_equalize.py
torch/quantization/fx/convert.py
torch/quantization/fx/fuse.py
torch/quantization/fx/fusion_patterns.py
torch/quantization/fx/graph_module.py
torch/quantization/fx/match_utils.py
torch/quantization/fx/pattern_utils.py
torch/quantization/fx/prepare.py
torch/quantization/fx/quantization_patterns.py
torch/quantization/fx/quantization_types.py
torch/quantization/fx/utils.py
torch/signal/__init__.py
torch/signal/windows/__init__.py
torch/signal/windows/windows.py
torch/sparse/__init__.py
torch/sparse/_semi_structured_conversions.py
torch/sparse/_triton_ops.py
torch/sparse/semi_structured.py
torch/special/__init__.py
torch/testing/__init__.py
torch/testing/_comparison.py
torch/testing/_creation.py
torch/testing/_internal/__init__.py
torch/testing/_internal/autocast_test_lists.py
torch/testing/_internal/autograd_function_db.py
torch/testing/_internal/check_kernel_launches.py
torch/testing/_internal/common_cuda.py
torch/testing/_internal/common_device_type.py
torch/testing/_internal/common_dist_composable.py
torch/testing/_internal/common_distributed.py
torch/testing/_internal/common_dtype.py
torch/testing/_internal/common_fsdp.py
torch/testing/_internal/common_jit.py
torch/testing/_internal/common_methods_invocations.py
torch/testing/_internal/common_modules.py
torch/testing/_internal/common_nn.py
torch/testing/_internal/common_pruning.py
torch/testing/_internal/common_quantization.py
torch/testing/_internal/common_quantized.py
torch/testing/_internal/common_subclass.py
torch/testing/_internal/common_utils.py
torch/testing/_internal/composite_compliance.py
torch/testing/_internal/control_flow_opinfo_db.py
torch/testing/_internal/custom_op_db.py
torch/testing/_internal/dist_utils.py
torch/testing/_internal/hypothesis_utils.py
torch/testing/_internal/inductor_utils.py
torch/testing/_internal/jit_metaprogramming_utils.py
torch/testing/_internal/jit_utils.py
torch/testing/_internal/logging_tensor.py
torch/testing/_internal/logging_utils.py
torch/testing/_internal/quantization_torch_package_models.py
torch/testing/_internal/two_tensor.py
torch/testing/_internal/codegen/__init__.py
torch/testing/_internal/codegen/random_topo_test.py
torch/testing/_internal/data/__init__.py
torch/testing/_internal/data/network1.py
torch/testing/_internal/data/network2.py
torch/testing/_internal/distributed/__init__.py
torch/testing/_internal/distributed/checkpoint_utils.py
torch/testing/_internal/distributed/ddp_under_dist_autograd_test.py
torch/testing/_internal/distributed/distributed_test.py
torch/testing/_internal/distributed/distributed_utils.py
torch/testing/_internal/distributed/fake_pg.py
torch/testing/_internal/distributed/multi_threaded_pg.py
torch/testing/_internal/distributed/pipe_with_ddp_test.py
torch/testing/_internal/distributed/rpc_utils.py
torch/testing/_internal/distributed/_shard/__init__.py
torch/testing/_internal/distributed/_shard/test_common.py
torch/testing/_internal/distributed/_shard/sharded_tensor/__init__.py
torch/testing/_internal/distributed/_shard/sharded_tensor/_test_ops_common.py
torch/testing/_internal/distributed/_shard/sharded_tensor/_test_st_common.py
torch/testing/_internal/distributed/_tensor/__init__.py
torch/testing/_internal/distributed/_tensor/common_dtensor.py
torch/testing/_internal/distributed/nn/__init__.py
torch/testing/_internal/distributed/nn/api/__init__.py
torch/testing/_internal/distributed/nn/api/remote_module_test.py
torch/testing/_internal/distributed/pipeline/__init__.py
torch/testing/_internal/distributed/rpc/__init__.py
torch/testing/_internal/distributed/rpc/dist_autograd_test.py
torch/testing/_internal/distributed/rpc/dist_optimizer_test.py
torch/testing/_internal/distributed/rpc/faulty_agent_rpc_test.py
torch/testing/_internal/distributed/rpc/faulty_rpc_agent_test_fixture.py
torch/testing/_internal/distributed/rpc/rpc_agent_test_fixture.py
torch/testing/_internal/distributed/rpc/rpc_test.py
torch/testing/_internal/distributed/rpc/tensorpipe_rpc_agent_test_fixture.py
torch/testing/_internal/distributed/rpc/examples/__init__.py
torch/testing/_internal/distributed/rpc/examples/parameter_server_test.py
torch/testing/_internal/distributed/rpc/examples/reinforcement_learning_rpc_test.py
torch/testing/_internal/distributed/rpc/jit/__init__.py
torch/testing/_internal/distributed/rpc/jit/dist_autograd_test.py
torch/testing/_internal/distributed/rpc/jit/rpc_test.py
torch/testing/_internal/distributed/rpc/jit/rpc_test_faulty.py
torch/testing/_internal/generated/__init__.py
torch/testing/_internal/opinfo/__init__.py
torch/testing/_internal/opinfo/core.py
torch/testing/_internal/opinfo/refs.py
torch/testing/_internal/opinfo/utils.py
torch/testing/_internal/opinfo/definitions/__init__.py
torch/testing/_internal/opinfo/definitions/_masked.py
torch/testing/_internal/opinfo/definitions/fft.py
torch/testing/_internal/opinfo/definitions/linalg.py
torch/testing/_internal/opinfo/definitions/signal.py
torch/testing/_internal/opinfo/definitions/sparse.py
torch/testing/_internal/opinfo/definitions/special.py
torch/testing/_internal/optests/__init__.py
torch/testing/_internal/optests/aot_autograd.py
torch/testing/_internal/optests/autograd_registration.py
torch/testing/_internal/optests/compile_check.py
torch/testing/_internal/optests/fake_tensor.py
torch/testing/_internal/optests/generate_tests.py
torch/testing/_internal/optests/make_fx.py
torch/testing/_internal/test_module/__init__.py
torch/testing/_internal/test_module/future_div.py
torch/testing/_internal/test_module/no_future_div.py
torch/utils/__init__.py
torch/utils/_content_store.py
torch/utils/_contextlib.py
torch/utils/_cpp_extension_versioner.py
torch/utils/_cuda_trace.py
torch/utils/_device.py
torch/utils/_foreach_utils.py
torch/utils/_freeze.py
torch/utils/_mode_utils.py
torch/utils/_python_dispatch.py
torch/utils/_pytree.py
torch/utils/_stats.py
torch/utils/_traceback.py
torch/utils/_zip.py
torch/utils/backend_registration.py
torch/utils/bundled_inputs.py
torch/utils/checkpoint.py
torch/utils/collect_env.py
torch/utils/cpp_backtrace.py
torch/utils/cpp_extension.py
torch/utils/dlpack.py
torch/utils/file_baton.py
torch/utils/flop_counter.py
torch/utils/hooks.py
torch/utils/mkldnn.py
torch/utils/mobile_optimizer.py
torch/utils/model_zoo.py
torch/utils/show_pickle.py
torch/utils/throughput_benchmark.py
torch/utils/weak.py
torch/utils/_sympy/__init__.py
torch/utils/_sympy/functions.py
torch/utils/_sympy/interp.py
torch/utils/_sympy/reference.py
torch/utils/_sympy/solve.py
torch/utils/_sympy/value_ranges.py
torch/utils/backcompat/__init__.py
torch/utils/benchmark/__init__.py
torch/utils/benchmark/examples/__init__.py
torch/utils/benchmark/examples/blas_compare.py
torch/utils/benchmark/examples/blas_compare_setup.py
torch/utils/benchmark/examples/compare.py
torch/utils/benchmark/examples/end_to_end.py
torch/utils/benchmark/examples/fuzzer.py
torch/utils/benchmark/examples/op_benchmark.py
torch/utils/benchmark/examples/simple_timeit.py
torch/utils/benchmark/examples/spectral_ops_fuzz_test.py
torch/utils/benchmark/op_fuzzers/__init__.py
torch/utils/benchmark/op_fuzzers/binary.py
torch/utils/benchmark/op_fuzzers/sparse_binary.py
torch/utils/benchmark/op_fuzzers/sparse_unary.py
torch/utils/benchmark/op_fuzzers/spectral.py
torch/utils/benchmark/op_fuzzers/unary.py
torch/utils/benchmark/utils/__init__.py
torch/utils/benchmark/utils/_stubs.py
torch/utils/benchmark/utils/common.py
torch/utils/benchmark/utils/compare.py
torch/utils/benchmark/utils/compile.py
torch/utils/benchmark/utils/cpp_jit.py
torch/utils/benchmark/utils/fuzzer.py
torch/utils/benchmark/utils/sparse_fuzzer.py
torch/utils/benchmark/utils/timer.py
torch/utils/benchmark/utils/valgrind_wrapper/__init__.py
torch/utils/benchmark/utils/valgrind_wrapper/timer_interface.py
torch/utils/bottleneck/__init__.py
torch/utils/bottleneck/__main__.py
torch/utils/data/__init__.py
torch/utils/data/backward_compatibility.py
torch/utils/data/dataloader.py
torch/utils/data/dataset.py
torch/utils/data/distributed.py
torch/utils/data/graph.py
torch/utils/data/graph_settings.py
torch/utils/data/sampler.py
torch/utils/data/_utils/__init__.py
torch/utils/data/_utils/collate.py
torch/utils/data/_utils/fetch.py
torch/utils/data/_utils/pin_memory.py
torch/utils/data/_utils/serialization.py
torch/utils/data/_utils/signal_handling.py
torch/utils/data/_utils/worker.py
torch/utils/data/datapipes/__init__.py
torch/utils/data/datapipes/_decorator.py
torch/utils/data/datapipes/_hook_iterator.py
torch/utils/data/datapipes/_typing.py
torch/utils/data/datapipes/datapipe.py
torch/utils/data/datapipes/gen_pyi.py
torch/utils/data/datapipes/dataframe/__init__.py
torch/utils/data/datapipes/dataframe/dataframe_wrapper.py
torch/utils/data/datapipes/dataframe/dataframes.py
torch/utils/data/datapipes/dataframe/datapipes.py
torch/utils/data/datapipes/dataframe/structures.py
torch/utils/data/datapipes/iter/__init__.py
torch/utils/data/datapipes/iter/callable.py
torch/utils/data/datapipes/iter/combinatorics.py
torch/utils/data/datapipes/iter/combining.py
torch/utils/data/datapipes/iter/filelister.py
torch/utils/data/datapipes/iter/fileopener.py
torch/utils/data/datapipes/iter/grouping.py
torch/utils/data/datapipes/iter/routeddecoder.py
torch/utils/data/datapipes/iter/selecting.py
torch/utils/data/datapipes/iter/sharding.py
torch/utils/data/datapipes/iter/streamreader.py
torch/utils/data/datapipes/iter/utils.py
torch/utils/data/datapipes/map/__init__.py
torch/utils/data/datapipes/map/callable.py
torch/utils/data/datapipes/map/combinatorics.py
torch/utils/data/datapipes/map/combining.py
torch/utils/data/datapipes/map/grouping.py
torch/utils/data/datapipes/map/utils.py
torch/utils/data/datapipes/utils/__init__.py
torch/utils/data/datapipes/utils/common.py
torch/utils/data/datapipes/utils/decoder.py
torch/utils/data/datapipes/utils/snapshot.py
torch/utils/hipify/__init__.py
torch/utils/hipify/constants.py
torch/utils/hipify/cuda_to_hip_mappings.py
torch/utils/hipify/hipify_python.py
torch/utils/hipify/version.py
torch/utils/jit/__init__.py
torch/utils/jit/log_extract.py
torch/utils/model_dump/__init__.py
torch/utils/model_dump/__main__.py
torch/utils/tensorboard/__init__.py
torch/utils/tensorboard/_caffe2_graph.py
torch/utils/tensorboard/_convert_np.py
torch/utils/tensorboard/_embedding.py
torch/utils/tensorboard/_onnx_graph.py
torch/utils/tensorboard/_proto_graph.py
torch/utils/tensorboard/_pytorch_graph.py
torch/utils/tensorboard/_utils.py
torch/utils/tensorboard/summary.py
torch/utils/tensorboard/writer.py
torch/utils/viz/__init__.py
torch/utils/viz/_cycles.py
torch_build_optim.egg-info/PKG-INFO
torch_build_optim.egg-info/SOURCES.txt
torch_build_optim.egg-info/dependency_links.txt
torch_build_optim.egg-info/requires.txt
torch_build_optim.egg-info/top_level.txt