Path Lines of Code torch/_C/_VariableFunctions.pyi.in 11 torch/_C/__init__.pyi.in 981 torch/_C/_autograd.pyi 87 torch/_C/_cudnn.pyi 12 torch/_C/_distributed_autograd.pyi 21 torch/_C/_distributed_c10d.pyi 360 torch/_C/_distributed_rpc.pyi 156 torch/_C/_distributed_rpc_testing.pyi 34 torch/_C/_functions.pyi 10 torch/_C/_monitor.pyi 40 torch/_C/_nn.pyi.in 25 torch/_C/_nvtx.pyi 3 torch/_C/_onnx.pyi 30 torch/_VF.py 11 torch/__config__.py 8 torch/__future__.py 6 torch/__init__.py 416 torch/_appdirs.py 309 torch/_classes.py 25 torch/_deploy.py 75 torch/_jit_internal.py 674 torch/_linalg_utils.py 48 torch/_lobpcg.py 576 torch/_lowrank.py 111 torch/_masked/__init__.py 528 torch/_namedtensor_internals.py 75 torch/_ops.py 46 torch/_python_dispatcher.py 64 torch/_six.py 12 torch/_sources.py 64 torch/_storage_docs.py 14 torch/_tensor.py 665 torch/_tensor_docs.py 1512 torch/_tensor_str.py 309 torch/_torch_docs.py 1102 torch/_utils.py 266 torch/_utils_internal.py 26 torch/_vmap_internals.py 137 torch/abi-check.cpp 8 torch/ao/__init__.py 1 torch/ao/nn/__init__.py 1 torch/ao/nn/sparse/__init__.py 1 torch/ao/nn/sparse/quantized/__init__.py 8 torch/ao/nn/sparse/quantized/dynamic/__init__.py 4 torch/ao/nn/sparse/quantized/dynamic/linear.py 96 torch/ao/nn/sparse/quantized/linear.py 150 torch/ao/nn/sparse/quantized/utils.py 26 torch/ao/ns/__init__.py 1 torch/ao/ns/_numeric_suite.py 276 torch/ao/ns/_numeric_suite_dbr.py 80 torch/ao/ns/_numeric_suite_fx.py 382 torch/ao/ns/fx/__init__.py 1 torch/ao/ns/fx/graph_matcher.py 251 torch/ao/ns/fx/graph_passes.py 539 torch/ao/ns/fx/mappings.py 596 torch/ao/ns/fx/ns_types.py 15 torch/ao/ns/fx/pattern_utils.py 125 torch/ao/ns/fx/utils.py 301 torch/ao/ns/fx/weight_utils.py 226 torch/ao/quantization/__init__.py 14 torch/ao/quantization/_correct_bias.py 103 torch/ao/quantization/_dbr/__init__.py 1 torch/ao/quantization/_dbr/auto_trace.py 357 torch/ao/quantization/_dbr/auto_trace_rewriter.py 180 torch/ao/quantization/_dbr/fusion.py 38 torch/ao/quantization/_dbr/mappings.py 119 torch/ao/quantization/_dbr/model_utils.py 92 torch/ao/quantization/_dbr/module_swap_utils.py 32 torch/ao/quantization/_dbr/qconfig_dict_utils.py 15 torch/ao/quantization/_dbr/quantization_state.py 556 torch/ao/quantization/_dbr/utils.py 439 torch/ao/quantization/_equalize.py 121 torch/ao/quantization/_learnable_fake_quantize.py 108 torch/ao/quantization/_quantize_dbr.py 74 torch/ao/quantization/_quantize_fx_do_not_use.py 26 torch/ao/quantization/fake_quantize.py 279 torch/ao/quantization/fuse_modules.py 59 torch/ao/quantization/fuser_method_mappings.py 134 torch/ao/quantization/fx/__init__.py 4 torch/ao/quantization/fx/_convert_do_not_use.py 210 torch/ao/quantization/fx/_equalize.py 410 torch/ao/quantization/fx/_lower_to_native_backend.py 64 torch/ao/quantization/fx/backend_config/__init__.py 3 torch/ao/quantization/fx/backend_config/fuse_handler.py 3 torch/ao/quantization/fx/backend_config/observation_type.py 4 torch/ao/quantization/fx/backend_config/quantize_handler.py 14 torch/ao/quantization/fx/backend_config/tensorrt.py 190 torch/ao/quantization/fx/backend_config/utils.py 66 torch/ao/quantization/fx/common_quantization_patterns.py 60 torch/ao/quantization/fx/convert.py 476 torch/ao/quantization/fx/fuse.py 92 torch/ao/quantization/fx/fusion_patterns.py 97 torch/ao/quantization/fx/graph_module.py 79 torch/ao/quantization/fx/lower_to_fbgemm.py 4 torch/ao/quantization/fx/lower_to_qnnpack.py 4 torch/ao/quantization/fx/match_utils.py 119 torch/ao/quantization/fx/pattern_utils.py 38 torch/ao/quantization/fx/prepare.py 935 torch/ao/quantization/fx/qconfig_utils.py 199 torch/ao/quantization/fx/quantization_patterns.py 1545 torch/ao/quantization/fx/quantization_types.py 5 torch/ao/quantization/fx/quantized_fusion_patterns_and_replacements.py 23 torch/ao/quantization/fx/subgraph_rewriter_FORKED_DO_NOT_USE.py 180 torch/ao/quantization/fx/utils.py 364 torch/ao/quantization/observer.py 877 torch/ao/quantization/qconfig.py 213 torch/ao/quantization/qconfig_dict_utils.py 69 torch/ao/quantization/quant_type.py 14 torch/ao/quantization/quantization_mappings.py 228 torch/ao/quantization/quantize.py 322 torch/ao/quantization/quantize_fx.py 288 torch/ao/quantization/quantize_jit.py 98 torch/ao/quantization/stubs.py 31 torch/ao/quantization/utils.py 158 torch/ao/sparsity/__init__.py 14 torch/ao/sparsity/_mappings.py 12 torch/ao/sparsity/experimental/__init__.py 1 torch/ao/sparsity/experimental/pruner/__init__.py 1 torch/ao/sparsity/experimental/pruner/base_pruner.py 166 torch/ao/sparsity/experimental/pruner/parametrization.py 54 torch/ao/sparsity/scheduler/__init__.py 1 torch/ao/sparsity/scheduler/base_scheduler.py 88 torch/ao/sparsity/scheduler/lambda_scheduler.py 20 torch/ao/sparsity/sparsifier/__init__.py 1 torch/ao/sparsity/sparsifier/base_sparsifier.py 153 torch/ao/sparsity/sparsifier/utils.py 27 torch/ao/sparsity/sparsifier/weight_norm_sparsifier.py 72 torch/autocast_mode.py 87 torch/autograd/__init__.py 160 torch/autograd/_functions/__init__.py 1 torch/autograd/_functions/tensor.py 42 torch/autograd/_functions/utils.py 41 torch/autograd/anomaly_mode.py 23 torch/autograd/forward_ad.py 49 torch/autograd/function.py 232 torch/autograd/functional.py 412 torch/autograd/grad_mode.py 90 torch/autograd/gradcheck.py 946 torch/autograd/graph.py 26 torch/autograd/profiler.py 346 torch/autograd/profiler_legacy.py 227 torch/autograd/profiler_util.py 684 torch/autograd/variable.py 9 torch/backends/__init__.py 35 torch/backends/_coreml/__init__.py 1 torch/backends/_coreml/preprocess.py 87 torch/backends/_nnapi/__init__.py 1 torch/backends/_nnapi/prepare.py 109 torch/backends/_nnapi/serializer.py 1641 torch/backends/cuda/__init__.py 96 torch/backends/cudnn/__init__.py 97 torch/backends/cudnn/rnn.py 40 torch/backends/mkl/__init__.py 4 torch/backends/mkldnn/__init__.py 25 torch/backends/openmp/__init__.py 4 torch/backends/quantized/__init__.py 40 torch/backends/xnnpack/__init__.py 16 torch/contrib/__init__.py 1 torch/contrib/_tensorboard_vis.py 86 torch/cpu/__init__.py 1 torch/cpu/amp/__init__.py 1 torch/cpu/amp/autocast_mode.py 23 torch/csrc/CudaIPCTypes.cpp 214 torch/csrc/CudaIPCTypes.h 113 torch/csrc/DataLoader.cpp 161 torch/csrc/DataLoader.h 3 torch/csrc/Device.cpp 206 torch/csrc/Device.h 14 torch/csrc/Dtype.cpp 126 torch/csrc/Dtype.h 21 torch/csrc/DynamicTypes.cpp 160 torch/csrc/DynamicTypes.h 26 torch/csrc/Exceptions.cpp 236 torch/csrc/Exceptions.h 308 torch/csrc/Export.h 7 torch/csrc/Generator.cpp 230 torch/csrc/Generator.h 15 torch/csrc/Layout.cpp 73 torch/csrc/Layout.h 16 torch/csrc/MemoryFormat.cpp 73 torch/csrc/MemoryFormat.h 16 torch/csrc/Module.cpp 915 torch/csrc/Module.h 4 torch/csrc/PythonTypes.h 17 torch/csrc/QScheme.cpp 82 torch/csrc/QScheme.h 16 torch/csrc/Size.cpp 202 torch/csrc/Size.h 9 torch/csrc/Storage.cpp 23 torch/csrc/Storage.h 14 torch/csrc/StorageDefs.h 5 torch/csrc/Stream.cpp 103 torch/csrc/Stream.h 13 torch/csrc/THCGenerateByteType.h 18 torch/csrc/THConcat.h 13 torch/csrc/THGenerateByteType.h 16 torch/csrc/THP.h 21 torch/csrc/TypeInfo.cpp 301 torch/csrc/TypeInfo.h 17 torch/csrc/Types.h 8 torch/csrc/api/include/torch/all.h 19 torch/csrc/api/include/torch/arg.h 20 torch/csrc/api/include/torch/autograd.h 4 torch/csrc/api/include/torch/cuda.h 14 torch/csrc/api/include/torch/data.h 11 torch/csrc/api/include/torch/data/dataloader.h 44 torch/csrc/api/include/torch/data/dataloader/base.h 163 torch/csrc/api/include/torch/data/dataloader/stateful.h 39 torch/csrc/api/include/torch/data/dataloader/stateless.h 56 torch/csrc/api/include/torch/data/dataloader_options.h 35 torch/csrc/api/include/torch/data/datasets.h 8 torch/csrc/api/include/torch/data/datasets/base.h 70 torch/csrc/api/include/torch/data/datasets/chunk.h 323 torch/csrc/api/include/torch/data/datasets/map.h 80 torch/csrc/api/include/torch/data/datasets/mnist.h 25 torch/csrc/api/include/torch/data/datasets/shared.h 49 torch/csrc/api/include/torch/data/datasets/stateful.h 42 torch/csrc/api/include/torch/data/datasets/tensor.h 24 torch/csrc/api/include/torch/data/detail/data_shuttle.h 50 torch/csrc/api/include/torch/data/detail/queue.h 55 torch/csrc/api/include/torch/data/detail/sequencers.h 66 torch/csrc/api/include/torch/data/example.h 34 torch/csrc/api/include/torch/data/iterator.h 115 torch/csrc/api/include/torch/data/samplers.h 8 torch/csrc/api/include/torch/data/samplers/base.h 28 torch/csrc/api/include/torch/data/samplers/custom_batch_request.h 13 torch/csrc/api/include/torch/data/samplers/distributed.h 88 torch/csrc/api/include/torch/data/samplers/random.h 33 torch/csrc/api/include/torch/data/samplers/sequential.h 30 torch/csrc/api/include/torch/data/samplers/serialize.h 23 torch/csrc/api/include/torch/data/samplers/stream.h 35 torch/csrc/api/include/torch/data/transforms.h 6 torch/csrc/api/include/torch/data/transforms/base.h 34 torch/csrc/api/include/torch/data/transforms/collate.h 14 torch/csrc/api/include/torch/data/transforms/lambda.h 38 torch/csrc/api/include/torch/data/transforms/stack.h 39 torch/csrc/api/include/torch/data/transforms/tensor.h 51 torch/csrc/api/include/torch/data/worker_exception.h 25 torch/csrc/api/include/torch/detail/TensorDataContainer.h 222 torch/csrc/api/include/torch/detail/static.h 40 torch/csrc/api/include/torch/enum.h 139 torch/csrc/api/include/torch/expanding_array.h 104 torch/csrc/api/include/torch/fft.h 131 torch/csrc/api/include/torch/imethod.h 21 torch/csrc/api/include/torch/jit.h 10 torch/csrc/api/include/torch/linalg.h 342 torch/csrc/api/include/torch/nn.h 9 torch/csrc/api/include/torch/nn/cloneable.h 71 torch/csrc/api/include/torch/nn/functional.h 16 torch/csrc/api/include/torch/nn/functional/activation.h 587 torch/csrc/api/include/torch/nn/functional/batchnorm.h 53 torch/csrc/api/include/torch/nn/functional/conv.h 184 torch/csrc/api/include/torch/nn/functional/distance.h 55 torch/csrc/api/include/torch/nn/functional/dropout.h 94 torch/csrc/api/include/torch/nn/functional/embedding.h 139 torch/csrc/api/include/torch/nn/functional/fold.h 68 torch/csrc/api/include/torch/nn/functional/instancenorm.h 26 torch/csrc/api/include/torch/nn/functional/linear.h 23 torch/csrc/api/include/torch/nn/functional/loss.h 581 torch/csrc/api/include/torch/nn/functional/normalization.h 117 torch/csrc/api/include/torch/nn/functional/padding.h 81 torch/csrc/api/include/torch/nn/functional/pixelshuffle.h 33 torch/csrc/api/include/torch/nn/functional/pooling.h 659 torch/csrc/api/include/torch/nn/functional/upsampling.h 180 torch/csrc/api/include/torch/nn/functional/vision.h 88 torch/csrc/api/include/torch/nn/init.h 54 torch/csrc/api/include/torch/nn/module.h 219 torch/csrc/api/include/torch/nn/modules.h 30 torch/csrc/api/include/torch/nn/modules/_functions.h 20 torch/csrc/api/include/torch/nn/modules/activation.h 261 torch/csrc/api/include/torch/nn/modules/adaptive.h 38 torch/csrc/api/include/torch/nn/modules/batchnorm.h 115 torch/csrc/api/include/torch/nn/modules/common.h 26 torch/csrc/api/include/torch/nn/modules/container/any.h 196 torch/csrc/api/include/torch/nn/modules/container/any_module_holder.h 87 torch/csrc/api/include/torch/nn/modules/container/any_value.h 80 torch/csrc/api/include/torch/nn/modules/container/functional.h 35 torch/csrc/api/include/torch/nn/modules/container/moduledict.h 115 torch/csrc/api/include/torch/nn/modules/container/modulelist.h 131 torch/csrc/api/include/torch/nn/modules/container/named_any.h 49 torch/csrc/api/include/torch/nn/modules/container/parameterdict.h 88 torch/csrc/api/include/torch/nn/modules/container/parameterlist.h 105 torch/csrc/api/include/torch/nn/modules/container/sequential.h 176 torch/csrc/api/include/torch/nn/modules/conv.h 260 torch/csrc/api/include/torch/nn/modules/distance.h 29 torch/csrc/api/include/torch/nn/modules/dropout.h 67 torch/csrc/api/include/torch/nn/modules/embedding.h 79 torch/csrc/api/include/torch/nn/modules/fold.h 33 torch/csrc/api/include/torch/nn/modules/instancenorm.h 56 torch/csrc/api/include/torch/nn/modules/linear.h 69 torch/csrc/api/include/torch/nn/modules/loss.h 202 torch/csrc/api/include/torch/nn/modules/normalization.h 64 torch/csrc/api/include/torch/nn/modules/padding.h 98 torch/csrc/api/include/torch/nn/modules/pixelshuffle.h 26 torch/csrc/api/include/torch/nn/modules/pooling.h 228 torch/csrc/api/include/torch/nn/modules/rnn.h 170 torch/csrc/api/include/torch/nn/modules/transformer.h 41 torch/csrc/api/include/torch/nn/modules/transformercoder.h 60 torch/csrc/api/include/torch/nn/modules/transformerlayer.h 77 torch/csrc/api/include/torch/nn/modules/upsampling.h 22 torch/csrc/api/include/torch/nn/modules/utils.h 39 torch/csrc/api/include/torch/nn/options.h 17 torch/csrc/api/include/torch/nn/options/activation.h 196 torch/csrc/api/include/torch/nn/options/adaptive.h 16 torch/csrc/api/include/torch/nn/options/batchnorm.h 28 torch/csrc/api/include/torch/nn/options/conv.h 133 torch/csrc/api/include/torch/nn/options/distance.h 23 torch/csrc/api/include/torch/nn/options/dropout.h 36 torch/csrc/api/include/torch/nn/options/embedding.h 74 torch/csrc/api/include/torch/nn/options/fold.h 33 torch/csrc/api/include/torch/nn/options/instancenorm.h 31 torch/csrc/api/include/torch/nn/options/linear.h 36 torch/csrc/api/include/torch/nn/options/loss.h 187 torch/csrc/api/include/torch/nn/options/normalization.h 65 torch/csrc/api/include/torch/nn/options/padding.h 53 torch/csrc/api/include/torch/nn/options/pixelshuffle.h 22 torch/csrc/api/include/torch/nn/options/pooling.h 150 torch/csrc/api/include/torch/nn/options/rnn.h 91 torch/csrc/api/include/torch/nn/options/transformer.h 25 torch/csrc/api/include/torch/nn/options/transformercoder.h 28 torch/csrc/api/include/torch/nn/options/transformerlayer.h 26 torch/csrc/api/include/torch/nn/options/upsampling.h 42 torch/csrc/api/include/torch/nn/options/vision.h 18 torch/csrc/api/include/torch/nn/parallel/data_parallel.h 175 torch/csrc/api/include/torch/nn/pimpl-inl.h 35 torch/csrc/api/include/torch/nn/pimpl.h 121 torch/csrc/api/include/torch/nn/utils.h 4 torch/csrc/api/include/torch/nn/utils/clip_grad.h 91 torch/csrc/api/include/torch/nn/utils/convert_parameters.h 49 torch/csrc/api/include/torch/nn/utils/rnn.h 150 torch/csrc/api/include/torch/optim.h 10 torch/csrc/api/include/torch/optim/adagrad.h 72 torch/csrc/api/include/torch/optim/adam.h 66 torch/csrc/api/include/torch/optim/adamw.h 66 torch/csrc/api/include/torch/optim/lbfgs.h 80 torch/csrc/api/include/torch/optim/optimizer.h 110 torch/csrc/api/include/torch/optim/rmsprop.h 68 torch/csrc/api/include/torch/optim/schedulers/lr_scheduler.h 20 torch/csrc/api/include/torch/optim/schedulers/step_lr.h 16 torch/csrc/api/include/torch/optim/serialize.h 227 torch/csrc/api/include/torch/optim/sgd.h 62 torch/csrc/api/include/torch/ordered_dict.h 316 torch/csrc/api/include/torch/python.h 165 torch/csrc/api/include/torch/python/init.h 7 torch/csrc/api/include/torch/serialize.h 45 torch/csrc/api/include/torch/serialize/archive.h 3 torch/csrc/api/include/torch/serialize/input-archive.h 57 torch/csrc/api/include/torch/serialize/output-archive.h 48 torch/csrc/api/include/torch/serialize/tensor.h 17 torch/csrc/api/include/torch/sparse.h 5 torch/csrc/api/include/torch/special.h 221 torch/csrc/api/include/torch/torch.h 5 torch/csrc/api/include/torch/types.h 29 torch/csrc/api/include/torch/utils.h 37 torch/csrc/api/include/torch/version.h.in 6 torch/csrc/api/src/cuda.cpp 45 torch/csrc/api/src/data/datasets/mnist.cpp 98 torch/csrc/api/src/data/samplers/distributed.cpp 140 torch/csrc/api/src/data/samplers/random.cpp 59 torch/csrc/api/src/data/samplers/sequential.cpp 47 torch/csrc/api/src/data/samplers/stream.cpp 51 torch/csrc/api/src/enum.cpp 39 torch/csrc/api/src/imethod.cpp 12 torch/csrc/api/src/jit.cpp 18 torch/csrc/api/src/nn/init.cpp 205 torch/csrc/api/src/nn/module.cpp 360 torch/csrc/api/src/nn/modules/_functions.cpp 89 torch/csrc/api/src/nn/modules/activation.cpp 387 torch/csrc/api/src/nn/modules/adaptive.cpp 152 torch/csrc/api/src/nn/modules/batchnorm.cpp 48 torch/csrc/api/src/nn/modules/container/functional.cpp 23 torch/csrc/api/src/nn/modules/conv.cpp 263 torch/csrc/api/src/nn/modules/distance.cpp 31 torch/csrc/api/src/nn/modules/dropout.cpp 55 torch/csrc/api/src/nn/modules/embedding.cpp 144 torch/csrc/api/src/nn/modules/fold.cpp 40 torch/csrc/api/src/nn/modules/instancenorm.cpp 41 torch/csrc/api/src/nn/modules/linear.cpp 123 torch/csrc/api/src/nn/modules/loss.cpp 282 torch/csrc/api/src/nn/modules/normalization.cpp 96 torch/csrc/api/src/nn/modules/padding.cpp 69 torch/csrc/api/src/nn/modules/pixelshuffle.cpp 28 torch/csrc/api/src/nn/modules/pooling.cpp 307 torch/csrc/api/src/nn/modules/rnn.cpp 686 torch/csrc/api/src/nn/modules/transformer.cpp 317 torch/csrc/api/src/nn/modules/upsampling.cpp 41 torch/csrc/api/src/nn/options/activation.cpp 38 torch/csrc/api/src/nn/options/adaptive.cpp 8 torch/csrc/api/src/nn/options/batchnorm.cpp 6 torch/csrc/api/src/nn/options/conv.cpp 16 torch/csrc/api/src/nn/options/dropout.cpp 6 torch/csrc/api/src/nn/options/embedding.cpp 9 torch/csrc/api/src/nn/options/instancenorm.cpp 6 torch/csrc/api/src/nn/options/linear.cpp 15 torch/csrc/api/src/nn/options/normalization.cpp 13 torch/csrc/api/src/nn/options/padding.cpp 16 torch/csrc/api/src/nn/options/pooling.cpp 22 torch/csrc/api/src/nn/options/rnn.cpp 25 torch/csrc/api/src/nn/options/transformer.cpp 30 torch/csrc/api/src/nn/options/vision.cpp 1 torch/csrc/api/src/optim/adagrad.cpp 125 torch/csrc/api/src/optim/adam.cpp 146 torch/csrc/api/src/optim/adamw.cpp 146 torch/csrc/api/src/optim/lbfgs.cpp 464 torch/csrc/api/src/optim/optimizer.cpp 149 torch/csrc/api/src/optim/rmsprop.cpp 148 torch/csrc/api/src/optim/schedulers/lr_scheduler.cpp 32 torch/csrc/api/src/optim/schedulers/step_lr.cpp 21 torch/csrc/api/src/optim/serialize.cpp 48 torch/csrc/api/src/optim/sgd.cpp 117 torch/csrc/api/src/python/init.cpp 63 torch/csrc/api/src/serialize.cpp 12 torch/csrc/api/src/serialize/input-archive.cpp 152 torch/csrc/api/src/serialize/output-archive.cpp 40 torch/csrc/autograd/FunctionsManual.cpp 3181 torch/csrc/autograd/FunctionsManual.h 440 torch/csrc/autograd/InferenceMode.h 6 torch/csrc/autograd/TraceTypeManual.cpp 253 torch/csrc/autograd/VariableTypeManual.cpp 297 torch/csrc/autograd/VariableTypeUtils.h 274 torch/csrc/autograd/anomaly_mode.cpp 61 torch/csrc/autograd/anomaly_mode.h 31 torch/csrc/autograd/autograd.cpp 147 torch/csrc/autograd/autograd.h 23 torch/csrc/autograd/autograd_meta.cpp 127 torch/csrc/autograd/autograd_not_implemented_fallback.cpp 255 torch/csrc/autograd/autograd_not_implemented_fallback.h 7 torch/csrc/autograd/cpp_hook.cpp 37 torch/csrc/autograd/cpp_hook.h 13 torch/csrc/autograd/custom_function.cpp 327 torch/csrc/autograd/custom_function.h 234 torch/csrc/autograd/edge.h 34 torch/csrc/autograd/engine.cpp 832 torch/csrc/autograd/engine.h 237 torch/csrc/autograd/forward_grad.cpp 55 torch/csrc/autograd/forward_grad.h 73 torch/csrc/autograd/function.cpp 58 torch/csrc/autograd/function.h 298 torch/csrc/autograd/function_hook.cpp 5 torch/csrc/autograd/function_hook.h 18 torch/csrc/autograd/functions/accumulate_grad.cpp 37 torch/csrc/autograd/functions/accumulate_grad.h 101 torch/csrc/autograd/functions/basic_ops.cpp 44 torch/csrc/autograd/functions/basic_ops.h 65 torch/csrc/autograd/functions/comm.cpp 116 torch/csrc/autograd/functions/comm.h 36 torch/csrc/autograd/functions/init.cpp 118 torch/csrc/autograd/functions/pybind.h 9 torch/csrc/autograd/functions/tensor.cpp 88 torch/csrc/autograd/functions/tensor.h 28 torch/csrc/autograd/functions/utils.cpp 54 torch/csrc/autograd/functions/utils.h 72 torch/csrc/autograd/grad_mode.h 7 torch/csrc/autograd/init.cpp 499 torch/csrc/autograd/input_buffer.cpp 141 torch/csrc/autograd/input_buffer.h 27 torch/csrc/autograd/input_metadata.h 53 torch/csrc/autograd/profiler.h 3 torch/csrc/autograd/profiler_kineto.cpp 638 torch/csrc/autograd/profiler_kineto.h 291 torch/csrc/autograd/profiler_legacy.cpp 442 torch/csrc/autograd/profiler_legacy.h 332 torch/csrc/autograd/profiler_python.cpp 471 torch/csrc/autograd/profiler_python.h 4 torch/csrc/autograd/python_anomaly_mode.cpp 90 torch/csrc/autograd/python_anomaly_mode.h 30 torch/csrc/autograd/python_autograd.h 11 torch/csrc/autograd/python_cpp_function.cpp 229 torch/csrc/autograd/python_cpp_function.h 53 torch/csrc/autograd/python_engine.cpp 328 torch/csrc/autograd/python_engine.h 33 torch/csrc/autograd/python_fft_functions.h 4 torch/csrc/autograd/python_function.cpp 809 torch/csrc/autograd/python_function.h 59 torch/csrc/autograd/python_hook.cpp 150 torch/csrc/autograd/python_hook.h 19 torch/csrc/autograd/python_legacy_variable.cpp 118 torch/csrc/autograd/python_legacy_variable.h 5 torch/csrc/autograd/python_linalg_functions.h 4 torch/csrc/autograd/python_mode.cpp 22 torch/csrc/autograd/python_mode.h 9 torch/csrc/autograd/python_nn_functions.h 4 torch/csrc/autograd/python_return_types.h 5 torch/csrc/autograd/python_saved_variable_hooks.cpp 60 torch/csrc/autograd/python_saved_variable_hooks.h 25 torch/csrc/autograd/python_sparse_functions.h 4 torch/csrc/autograd/python_special_functions.h 4 torch/csrc/autograd/python_torch_functions.h 16 torch/csrc/autograd/python_torch_functions_manual.cpp 723 torch/csrc/autograd/python_variable.cpp 1321 torch/csrc/autograd/python_variable.h 42 torch/csrc/autograd/python_variable_indexing.cpp 370 torch/csrc/autograd/python_variable_indexing.h 7 torch/csrc/autograd/record_function_ops.cpp 72 torch/csrc/autograd/record_function_ops.h 13 torch/csrc/autograd/saved_variable.cpp 175 torch/csrc/autograd/saved_variable.h 47 torch/csrc/autograd/saved_variable_hooks.h 9 torch/csrc/autograd/symbolic.h 12 torch/csrc/autograd/utils/error_messages.h 14 torch/csrc/autograd/utils/grad_layout_contract.h 24 torch/csrc/autograd/utils/lambda_post_hook.h 23 torch/csrc/autograd/utils/python_arg_parsing.h 29 torch/csrc/autograd/utils/warnings.cpp 19 torch/csrc/autograd/utils/warnings.h 22 torch/csrc/autograd/utils/wrap_outputs.h 179 torch/csrc/autograd/variable.cpp 510 torch/csrc/autograd/variable.h 258 torch/csrc/copy_utils.h 33 torch/csrc/cuda/Event.cpp 202 torch/csrc/cuda/Event.h 14 torch/csrc/cuda/Graph.cpp 33 torch/csrc/cuda/Module.cpp 528 torch/csrc/cuda/Module.h 10 torch/csrc/cuda/Storage.cpp 13 torch/csrc/cuda/Storage.h 14 torch/csrc/cuda/Stream.cpp 164 torch/csrc/cuda/Stream.h 14 torch/csrc/cuda/THCP.h 11 torch/csrc/cuda/Tensor.cpp 13 torch/csrc/cuda/comm.cpp 415 torch/csrc/cuda/comm.h 42 torch/csrc/cuda/device_set.h 6 torch/csrc/cuda/nccl.cpp 694 torch/csrc/cuda/nccl.h 147 torch/csrc/cuda/override_macros.h 30 torch/csrc/cuda/python_comm.cpp 101 torch/csrc/cuda/python_comm.h 4 torch/csrc/cuda/python_nccl.cpp 268 torch/csrc/cuda/python_nccl.h 10 torch/csrc/cuda/restore_macros.h 13 torch/csrc/cuda/serialization.cpp 7 torch/csrc/cuda/serialization.h 6 torch/csrc/cuda/shared/cudart.cpp 50 torch/csrc/cuda/shared/cudnn.cpp 78 torch/csrc/cuda/shared/nvtx.cpp 13 torch/csrc/cuda/undef_macros.h 29 torch/csrc/cuda/utils.cpp 29 torch/csrc/cuda/utils.h 12 torch/csrc/deploy/deploy.cpp 278 torch/csrc/deploy/deploy.h 247 torch/csrc/deploy/elf_file.cpp 43 torch/csrc/deploy/elf_file.h 51 torch/csrc/deploy/environment.h 53 torch/csrc/deploy/example/benchmark.cpp 284 torch/csrc/deploy/example/examples.py 113 torch/csrc/deploy/example/fx/examples.py 13 torch/csrc/deploy/example/fx/some_dependency.py 2 torch/csrc/deploy/example/generate_examples.py 70 torch/csrc/deploy/example/gpu_wrapper.py 54 torch/csrc/deploy/example/simple.pt 23 torch/csrc/deploy/example/tensorrt_example.py 48 torch/csrc/deploy/interactive_embedded_interpreter.cpp 26 torch/csrc/deploy/interpreter/builtin_registry.cpp 208 torch/csrc/deploy/interpreter/builtin_registry.h 56 torch/csrc/deploy/interpreter/import_find_sharedfuncptr.cpp 33 torch/csrc/deploy/interpreter/interpreter_impl.cpp 330 torch/csrc/deploy/interpreter/interpreter_impl.h 130 torch/csrc/deploy/interpreter/register_frozenpython.cpp 78 torch/csrc/deploy/interpreter/register_numpy.cpp 48 torch/csrc/deploy/interpreter/register_pyyaml.cpp 4 torch/csrc/deploy/loader.cpp 963 torch/csrc/deploy/loader.h 40 torch/csrc/deploy/mem_file.h 52 torch/csrc/deploy/noop_environment.h 10 torch/csrc/deploy/path_environment.cpp 10 torch/csrc/deploy/path_environment.h 14 torch/csrc/deploy/remove_dt_needed.cpp 74 torch/csrc/deploy/unity/example.py 9 torch/csrc/deploy/unity/main.cpp 27 torch/csrc/deploy/unity/xar_environment.cpp 91 torch/csrc/deploy/unity/xar_environment.h 24 torch/csrc/distributed/autograd/autograd.cpp 23 torch/csrc/distributed/autograd/autograd.h 14 torch/csrc/distributed/autograd/context/container.cpp 261 torch/csrc/distributed/autograd/context/container.h 59 torch/csrc/distributed/autograd/context/context.cpp 232 torch/csrc/distributed/autograd/context/context.h 81 torch/csrc/distributed/autograd/engine/dist_engine.cpp 407 torch/csrc/distributed/autograd/engine/dist_engine.h 73 torch/csrc/distributed/autograd/functions/recvrpc_backward.cpp 52 torch/csrc/distributed/autograd/functions/recvrpc_backward.h 27 torch/csrc/distributed/autograd/functions/sendrpc_backward.cpp 23 torch/csrc/distributed/autograd/functions/sendrpc_backward.h 17 torch/csrc/distributed/autograd/init.cpp 202 torch/csrc/distributed/autograd/python_autograd.h 9 torch/csrc/distributed/autograd/rpc_messages/autograd_metadata.cpp 12 torch/csrc/distributed/autograd/rpc_messages/autograd_metadata.h 14 torch/csrc/distributed/autograd/rpc_messages/cleanup_autograd_context_req.cpp 35 torch/csrc/distributed/autograd/rpc_messages/cleanup_autograd_context_req.h 20 torch/csrc/distributed/autograd/rpc_messages/cleanup_autograd_context_resp.cpp 19 torch/csrc/distributed/autograd/rpc_messages/cleanup_autograd_context_resp.h 16 torch/csrc/distributed/autograd/rpc_messages/propagate_gradients_req.cpp 69 torch/csrc/distributed/autograd/rpc_messages/propagate_gradients_req.h 28 torch/csrc/distributed/autograd/rpc_messages/propagate_gradients_resp.cpp 17 torch/csrc/distributed/autograd/rpc_messages/propagate_gradients_resp.h 16 torch/csrc/distributed/autograd/rpc_messages/rpc_with_autograd.cpp 141 torch/csrc/distributed/autograd/rpc_messages/rpc_with_autograd.h 47 torch/csrc/distributed/autograd/rpc_messages/rpc_with_profiling_req.cpp 115 torch/csrc/distributed/autograd/rpc_messages/rpc_with_profiling_req.h 44 torch/csrc/distributed/autograd/rpc_messages/rpc_with_profiling_resp.cpp 129 torch/csrc/distributed/autograd/rpc_messages/rpc_with_profiling_resp.h 44 torch/csrc/distributed/autograd/rpc_messages/rref_backward_req.cpp 57 torch/csrc/distributed/autograd/rpc_messages/rref_backward_req.h 27 torch/csrc/distributed/autograd/rpc_messages/rref_backward_resp.cpp 18 torch/csrc/distributed/autograd/rpc_messages/rref_backward_resp.h 16 torch/csrc/distributed/autograd/utils.cpp 145 torch/csrc/distributed/autograd/utils.h 35 torch/csrc/distributed/c10d/FileStore.cpp 372 torch/csrc/distributed/c10d/FileStore.hpp 39 torch/csrc/distributed/c10d/GlooDeviceFactory.cpp 131 torch/csrc/distributed/c10d/GlooDeviceFactory.hpp 21 torch/csrc/distributed/c10d/HashStore.cpp 104 torch/csrc/distributed/c10d/HashStore.hpp 32 torch/csrc/distributed/c10d/NCCLUtils.cpp 44 torch/csrc/distributed/c10d/NCCLUtils.hpp 161 torch/csrc/distributed/c10d/ParamCommsUtils.cpp 19 torch/csrc/distributed/c10d/ParamCommsUtils.hpp 64 torch/csrc/distributed/c10d/PrefixStore.cpp 65 torch/csrc/distributed/c10d/PrefixStore.hpp 34 torch/csrc/distributed/c10d/ProcessGroup.cpp 153 torch/csrc/distributed/c10d/ProcessGroup.hpp 283 torch/csrc/distributed/c10d/ProcessGroupGloo.cpp 2263 torch/csrc/distributed/c10d/ProcessGroupGloo.hpp 213 torch/csrc/distributed/c10d/ProcessGroupMPI.cpp 786 torch/csrc/distributed/c10d/ProcessGroupMPI.hpp 172 torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp 1723 torch/csrc/distributed/c10d/ProcessGroupNCCL.hpp 276 torch/csrc/distributed/c10d/ProcessGroupRoundRobin.cpp 111 torch/csrc/distributed/c10d/ProcessGroupRoundRobin.hpp 77 torch/csrc/distributed/c10d/ProcessGroupWrapper.cpp 288 torch/csrc/distributed/c10d/ProcessGroupWrapper.hpp 87 torch/csrc/distributed/c10d/PyProcessGroup.hpp 103 torch/csrc/distributed/c10d/Store.cpp 12 torch/csrc/distributed/c10d/Store.hpp 52 torch/csrc/distributed/c10d/TCPStore.cpp 873 torch/csrc/distributed/c10d/TCPStore.hpp 73 torch/csrc/distributed/c10d/TraceUtils.h 231 torch/csrc/distributed/c10d/Types.hpp 55 torch/csrc/distributed/c10d/UnixSockUtils.hpp 18 torch/csrc/distributed/c10d/Utils.cpp 78 torch/csrc/distributed/c10d/Utils.hpp 554 torch/csrc/distributed/c10d/WinSockUtils.hpp 18 torch/csrc/distributed/c10d/c10d.h 9 torch/csrc/distributed/c10d/comm.cpp 71 torch/csrc/distributed/c10d/comm.hpp 88 torch/csrc/distributed/c10d/default_comm_hooks.cpp 42 torch/csrc/distributed/c10d/default_comm_hooks.hpp 31 torch/csrc/distributed/c10d/error.h 26 torch/csrc/distributed/c10d/exception.cpp 5 torch/csrc/distributed/c10d/exception.h 23 torch/csrc/distributed/c10d/init.cpp 1477 torch/csrc/distributed/c10d/logger.cpp 316 torch/csrc/distributed/c10d/logger.hpp 54 torch/csrc/distributed/c10d/logging.h 9 torch/csrc/distributed/c10d/python_comm_hook.cpp 40 torch/csrc/distributed/c10d/python_comm_hook.h 19 torch/csrc/distributed/c10d/quantization/quantization.cpp 85 torch/csrc/distributed/c10d/quantization/quantization.h 13 torch/csrc/distributed/c10d/quantization/quantization_gpu.cu 134 torch/csrc/distributed/c10d/quantization/quantization_gpu.h 13 torch/csrc/distributed/c10d/quantization/quantization_utils.h 23 torch/csrc/distributed/c10d/reducer.cpp 1456 torch/csrc/distributed/c10d/reducer.hpp 282 torch/csrc/distributed/c10d/reducer_cuda.cpp 60 torch/csrc/distributed/c10d/sequence_num.cpp 49 torch/csrc/distributed/c10d/sequence_num.hpp 44 torch/csrc/distributed/c10d/socket.cpp 607 torch/csrc/distributed/c10d/socket.h 59 torch/csrc/distributed/rpc/agent_utils.cpp 72 torch/csrc/distributed/rpc/agent_utils.h 18 torch/csrc/distributed/rpc/init.cpp 732 torch/csrc/distributed/rpc/message.cpp 88 torch/csrc/distributed/rpc/message.h 109 torch/csrc/distributed/rpc/metrics/RpcMetricsHandler.h 25 torch/csrc/distributed/rpc/metrics/registry.cpp 10 torch/csrc/distributed/rpc/profiler/remote_profiler_manager.cpp 77 torch/csrc/distributed/rpc/profiler/remote_profiler_manager.h 40 torch/csrc/distributed/rpc/profiler/server_process_global_profiler.cpp 51 torch/csrc/distributed/rpc/profiler/server_process_global_profiler.h 70 torch/csrc/distributed/rpc/py_rref.cpp 283 torch/csrc/distributed/rpc/py_rref.h 51 torch/csrc/distributed/rpc/python_call.cpp 43 torch/csrc/distributed/rpc/python_call.h 22 torch/csrc/distributed/rpc/python_functions.cpp 359 torch/csrc/distributed/rpc/python_functions.h 52 torch/csrc/distributed/rpc/python_remote_call.cpp 58 torch/csrc/distributed/rpc/python_remote_call.h 39 torch/csrc/distributed/rpc/python_resp.cpp 27 torch/csrc/distributed/rpc/python_resp.h 18 torch/csrc/distributed/rpc/python_rpc_handler.cpp 152 torch/csrc/distributed/rpc/python_rpc_handler.h 54 torch/csrc/distributed/rpc/request_callback.cpp 15 torch/csrc/distributed/rpc/request_callback.h 19 torch/csrc/distributed/rpc/request_callback_impl.cpp 287 torch/csrc/distributed/rpc/request_callback_impl.h 48 torch/csrc/distributed/rpc/request_callback_no_python.cpp 471 torch/csrc/distributed/rpc/request_callback_no_python.h 84 torch/csrc/distributed/rpc/rpc.h 9 torch/csrc/distributed/rpc/rpc_agent.cpp 212 torch/csrc/distributed/rpc/rpc_agent.h 183 torch/csrc/distributed/rpc/rpc_command_base.h 19 torch/csrc/distributed/rpc/rref_context.cpp 558 torch/csrc/distributed/rpc/rref_context.h 133 torch/csrc/distributed/rpc/rref_impl.cpp 217 torch/csrc/distributed/rpc/rref_impl.h 163 torch/csrc/distributed/rpc/rref_proto.cpp 157 torch/csrc/distributed/rpc/rref_proto.h 117 torch/csrc/distributed/rpc/script_call.cpp 119 torch/csrc/distributed/rpc/script_call.h 46 torch/csrc/distributed/rpc/script_remote_call.cpp 71 torch/csrc/distributed/rpc/script_remote_call.h 40 torch/csrc/distributed/rpc/script_resp.cpp 35 torch/csrc/distributed/rpc/script_resp.h 19 torch/csrc/distributed/rpc/tensorpipe_agent.cpp 1019 torch/csrc/distributed/rpc/tensorpipe_agent.h 281 torch/csrc/distributed/rpc/tensorpipe_cuda.cpp 87 torch/csrc/distributed/rpc/tensorpipe_utils.cpp 259 torch/csrc/distributed/rpc/tensorpipe_utils.h 77 torch/csrc/distributed/rpc/testing/faulty_tensorpipe_agent.cpp 126 torch/csrc/distributed/rpc/testing/faulty_tensorpipe_agent.h 72 torch/csrc/distributed/rpc/testing/init.cpp 126 torch/csrc/distributed/rpc/testing/testing.h 11 torch/csrc/distributed/rpc/torchscript_functions.cpp 135 torch/csrc/distributed/rpc/torchscript_functions.h 26 torch/csrc/distributed/rpc/types.cpp 88 torch/csrc/distributed/rpc/types.h 49 torch/csrc/distributed/rpc/unpickled_python_call.cpp 29 torch/csrc/distributed/rpc/unpickled_python_call.h 25 torch/csrc/distributed/rpc/unpickled_python_remote_call.cpp 23 torch/csrc/distributed/rpc/unpickled_python_remote_call.h 24 torch/csrc/distributed/rpc/utils.cpp 489 torch/csrc/distributed/rpc/utils.h 49 torch/csrc/dl.c 23 torch/csrc/empty.c 1 torch/csrc/generic/Storage.cpp 311 torch/csrc/generic/Storage.h 11 torch/csrc/generic/StorageMethods.cpp 419 torch/csrc/generic/StorageSharing.cpp 486 torch/csrc/generic/serialization.cpp 162 torch/csrc/generic/serialization.h 9 torch/csrc/generic/utils.h 28 torch/csrc/jit/api/compilation_unit.h 251 torch/csrc/jit/api/function_impl.cpp 111 torch/csrc/jit/api/function_impl.h 115 torch/csrc/jit/api/method.h 48 torch/csrc/jit/api/module.cpp 503 torch/csrc/jit/api/module.h 422 torch/csrc/jit/api/module_save.cpp 35 torch/csrc/jit/api/object.cpp 34 torch/csrc/jit/api/object.h 142 torch/csrc/jit/backends/backend.h 101 torch/csrc/jit/backends/backend_debug_handler.cpp 24 torch/csrc/jit/backends/backend_debug_handler.h 22 torch/csrc/jit/backends/backend_debug_info.cpp 21 torch/csrc/jit/backends/backend_debug_info.h 30 torch/csrc/jit/backends/backend_detail.cpp 263 torch/csrc/jit/backends/backend_detail.h 28 torch/csrc/jit/backends/backend_exception.h 35 torch/csrc/jit/backends/backend_init.cpp 130 torch/csrc/jit/backends/backend_init.h 8 torch/csrc/jit/backends/backend_interface.cpp 7 torch/csrc/jit/backends/backend_interface.h 18 torch/csrc/jit/backends/backend_preprocess.h 16 torch/csrc/jit/backends/backend_resolver.cpp 52 torch/csrc/jit/backends/backend_resolver.h 7 torch/csrc/jit/backends/coreml/cpp/backend.cpp 25 torch/csrc/jit/backends/coreml/cpp/context.cpp 24 torch/csrc/jit/backends/coreml/cpp/context.h 24 torch/csrc/jit/backends/coreml/cpp/preprocess.cpp 26 torch/csrc/jit/backends/coreml/objc/PTMCoreMLBackend.mm 244 torch/csrc/jit/backends/coreml/objc/PTMCoreMLExecutor.h 28 torch/csrc/jit/backends/coreml/objc/PTMCoreMLExecutor.mm 230 torch/csrc/jit/backends/nnapi/nnapi_backend_lib.cpp 92 torch/csrc/jit/backends/nnapi/nnapi_backend_preprocess.cpp 77 torch/csrc/jit/codegen/cuda/arith.cpp 1171 torch/csrc/jit/codegen/cuda/arith.h 353 torch/csrc/jit/codegen/cuda/codegen.cpp 1091 torch/csrc/jit/codegen/cuda/codegen.h 17 torch/csrc/jit/codegen/cuda/compute_at.cpp 651 torch/csrc/jit/codegen/cuda/compute_at.h 66 torch/csrc/jit/codegen/cuda/compute_at_map.cpp 498 torch/csrc/jit/codegen/cuda/compute_at_map.h 46 torch/csrc/jit/codegen/cuda/disjoint_set.h 106 torch/csrc/jit/codegen/cuda/dispatch.cpp 357 torch/csrc/jit/codegen/cuda/dispatch.h 317 torch/csrc/jit/codegen/cuda/evaluator_common.cpp 440 torch/csrc/jit/codegen/cuda/evaluator_common.h 162 torch/csrc/jit/codegen/cuda/executor.cpp 781 torch/csrc/jit/codegen/cuda/executor.h 141 torch/csrc/jit/codegen/cuda/executor_kernel_arg.cpp 181 torch/csrc/jit/codegen/cuda/executor_kernel_arg.h 120 torch/csrc/jit/codegen/cuda/executor_launch_params.cpp 123 torch/csrc/jit/codegen/cuda/executor_launch_params.h 102 torch/csrc/jit/codegen/cuda/executor_utils.cpp 1025 torch/csrc/jit/codegen/cuda/executor_utils.h 190 torch/csrc/jit/codegen/cuda/expr_evaluator.cpp 120 torch/csrc/jit/codegen/cuda/expr_evaluator.h 40 torch/csrc/jit/codegen/cuda/fusion.cpp 573 torch/csrc/jit/codegen/cuda/fusion.h 126 torch/csrc/jit/codegen/cuda/fusion_segmenter.cpp 2148 torch/csrc/jit/codegen/cuda/fusion_segmenter.h 316 torch/csrc/jit/codegen/cuda/graph_fuser.cpp 1427 torch/csrc/jit/codegen/cuda/index_compute.cpp 2108 torch/csrc/jit/codegen/cuda/index_compute.h 156 torch/csrc/jit/codegen/cuda/index_reference_replay.cpp 277 torch/csrc/jit/codegen/cuda/index_reference_replay.h 63 torch/csrc/jit/codegen/cuda/instrumentation.cpp 59 torch/csrc/jit/codegen/cuda/instrumentation.h 65 torch/csrc/jit/codegen/cuda/interface.cpp 228 torch/csrc/jit/codegen/cuda/interface.h 35 torch/csrc/jit/codegen/cuda/ir_all_nodes.h 5 torch/csrc/jit/codegen/cuda/ir_base_nodes.cpp 182 torch/csrc/jit/codegen/cuda/ir_base_nodes.h 205 torch/csrc/jit/codegen/cuda/ir_cloner.cpp 140 torch/csrc/jit/codegen/cuda/ir_cloner.h 73 torch/csrc/jit/codegen/cuda/ir_graphviz.cpp 374 torch/csrc/jit/codegen/cuda/ir_graphviz.h 84 torch/csrc/jit/codegen/cuda/ir_interface_nodes.h 215 torch/csrc/jit/codegen/cuda/ir_internal_nodes.h 602 torch/csrc/jit/codegen/cuda/ir_iostream.cpp 418 torch/csrc/jit/codegen/cuda/ir_iostream.h 77 torch/csrc/jit/codegen/cuda/ir_nodes.cpp 1269 torch/csrc/jit/codegen/cuda/ir_printer.h 29 torch/csrc/jit/codegen/cuda/ir_utils.cpp 345 torch/csrc/jit/codegen/cuda/ir_utils.h 117 torch/csrc/jit/codegen/cuda/iter_visitor.cpp 549 torch/csrc/jit/codegen/cuda/iter_visitor.h 111 torch/csrc/jit/codegen/cuda/kernel.cpp 211 torch/csrc/jit/codegen/cuda/kernel.h 103 torch/csrc/jit/codegen/cuda/kernel_cache.cpp 468 torch/csrc/jit/codegen/cuda/kernel_cache.h 201 torch/csrc/jit/codegen/cuda/kernel_expr_evaluator.cpp 142 torch/csrc/jit/codegen/cuda/kernel_expr_evaluator.h 39 torch/csrc/jit/codegen/cuda/kernel_ir.cpp 597 torch/csrc/jit/codegen/cuda/kernel_ir.h 1221 torch/csrc/jit/codegen/cuda/kernel_ir_builder.cpp 233 torch/csrc/jit/codegen/cuda/kernel_ir_builder.h 73 torch/csrc/jit/codegen/cuda/kernel_ir_printer.cpp 376 torch/csrc/jit/codegen/cuda/kernel_ir_printer.h 70 torch/csrc/jit/codegen/cuda/lower2device.cpp 533 torch/csrc/jit/codegen/cuda/lower2device.h 109 torch/csrc/jit/codegen/cuda/lower_alias_memory.cpp 798 torch/csrc/jit/codegen/cuda/lower_alias_memory.h 15 torch/csrc/jit/codegen/cuda/lower_allocation.cpp 482 torch/csrc/jit/codegen/cuda/lower_allocation.h 22 torch/csrc/jit/codegen/cuda/lower_expr_sort.cpp 889 torch/csrc/jit/codegen/cuda/lower_expr_sort.h 11 torch/csrc/jit/codegen/cuda/lower_index.cpp 361 torch/csrc/jit/codegen/cuda/lower_index.h 46 torch/csrc/jit/codegen/cuda/lower_insert_syncs.cpp 451 torch/csrc/jit/codegen/cuda/lower_insert_syncs.h 17 torch/csrc/jit/codegen/cuda/lower_loops.cpp 229 torch/csrc/jit/codegen/cuda/lower_loops.h 31 torch/csrc/jit/codegen/cuda/lower_magic_zero.cpp 116 torch/csrc/jit/codegen/cuda/lower_magic_zero.h 15 torch/csrc/jit/codegen/cuda/lower_misaligned_vectorization.cpp 429 torch/csrc/jit/codegen/cuda/lower_misaligned_vectorization.h 16 torch/csrc/jit/codegen/cuda/lower_predicate.cpp 422 torch/csrc/jit/codegen/cuda/lower_predicate.h 33 torch/csrc/jit/codegen/cuda/lower_shift.cpp 596 torch/csrc/jit/codegen/cuda/lower_shift.h 82 torch/csrc/jit/codegen/cuda/lower_thread_predicate.cpp 246 torch/csrc/jit/codegen/cuda/lower_thread_predicate.h 50 torch/csrc/jit/codegen/cuda/lower_trivial_reductions.cpp 100 torch/csrc/jit/codegen/cuda/lower_trivial_reductions.h 30 torch/csrc/jit/codegen/cuda/lower_unroll.cpp 214 torch/csrc/jit/codegen/cuda/lower_unroll.h 36 torch/csrc/jit/codegen/cuda/lower_utils.cpp 499 torch/csrc/jit/codegen/cuda/lower_utils.h 77 torch/csrc/jit/codegen/cuda/lower_validation.cpp 583 torch/csrc/jit/codegen/cuda/lower_validation.h 15 torch/csrc/jit/codegen/cuda/lower_warp_reduce.cpp 394 torch/csrc/jit/codegen/cuda/lower_warp_reduce.h 16 torch/csrc/jit/codegen/cuda/manager.cpp 159 torch/csrc/jit/codegen/cuda/manager.h 16 torch/csrc/jit/codegen/cuda/mutator.cpp 196 torch/csrc/jit/codegen/cuda/mutator.h 13 torch/csrc/jit/codegen/cuda/non_divisible_split.cpp 121 torch/csrc/jit/codegen/cuda/non_divisible_split.h 37 torch/csrc/jit/codegen/cuda/ops/all_ops.h 4 torch/csrc/jit/codegen/cuda/ops/composite.cpp 158 torch/csrc/jit/codegen/cuda/ops/composite.h 41 torch/csrc/jit/codegen/cuda/ops/normalization.cpp 456 torch/csrc/jit/codegen/cuda/ops/normalization.h 79 torch/csrc/jit/codegen/cuda/parallel_dimension_map.cpp 242 torch/csrc/jit/codegen/cuda/parallel_dimension_map.h 39 torch/csrc/jit/codegen/cuda/parallel_type_bitmap.cpp 25 torch/csrc/jit/codegen/cuda/parallel_type_bitmap.h 203 torch/csrc/jit/codegen/cuda/parser.cpp 2466 torch/csrc/jit/codegen/cuda/parser.h 28 torch/csrc/jit/codegen/cuda/partial_split_map.cpp 72 torch/csrc/jit/codegen/cuda/partial_split_map.h 27 torch/csrc/jit/codegen/cuda/partition.cpp 242 torch/csrc/jit/codegen/cuda/partition.h 15 torch/csrc/jit/codegen/cuda/predicate_compute.cpp 519 torch/csrc/jit/codegen/cuda/predicate_compute.h 119 torch/csrc/jit/codegen/cuda/reference_tensor.h 17 torch/csrc/jit/codegen/cuda/register_interface.cpp 28 torch/csrc/jit/codegen/cuda/root_domain_map.cpp 773 torch/csrc/jit/codegen/cuda/root_domain_map.h 213 torch/csrc/jit/codegen/cuda/runtime/bf16_support.cu 27 torch/csrc/jit/codegen/cuda/runtime/block_reduction.cu 108 torch/csrc/jit/codegen/cuda/runtime/block_sync_atomic.cu 44 torch/csrc/jit/codegen/cuda/runtime/block_sync_default.cu 8 torch/csrc/jit/codegen/cuda/runtime/broadcast.cu 30 torch/csrc/jit/codegen/cuda/runtime/fp16_support.cu 35 torch/csrc/jit/codegen/cuda/runtime/grid_broadcast.cu 65 torch/csrc/jit/codegen/cuda/runtime/grid_reduction.cu 259 torch/csrc/jit/codegen/cuda/runtime/grid_sync.cu 59 torch/csrc/jit/codegen/cuda/runtime/helpers.cu 228 torch/csrc/jit/codegen/cuda/runtime/index_utils.cu 67 torch/csrc/jit/codegen/cuda/runtime/random_numbers.cu 102 torch/csrc/jit/codegen/cuda/runtime/tensor.cu 18 torch/csrc/jit/codegen/cuda/runtime/warp.cu 63 torch/csrc/jit/codegen/cuda/runtime/welford.cu 342 torch/csrc/jit/codegen/cuda/scheduler/all_schedulers.h 17 torch/csrc/jit/codegen/cuda/scheduler/compile_time_info.h 108 torch/csrc/jit/codegen/cuda/scheduler/normalization.cpp 660 torch/csrc/jit/codegen/cuda/scheduler/normalization.h 25 torch/csrc/jit/codegen/cuda/scheduler/pointwise.cpp 584 torch/csrc/jit/codegen/cuda/scheduler/pointwise.h 28 torch/csrc/jit/codegen/cuda/scheduler/pointwise_heuristic.h 65 torch/csrc/jit/codegen/cuda/scheduler/reduction.cpp 612 torch/csrc/jit/codegen/cuda/scheduler/reduction.h 25 torch/csrc/jit/codegen/cuda/scheduler/reduction_heuristic.h 173 torch/csrc/jit/codegen/cuda/scheduler/reduction_utils.cpp 621 torch/csrc/jit/codegen/cuda/scheduler/reduction_utils.h 28 torch/csrc/jit/codegen/cuda/scheduler/registry.cpp 967 torch/csrc/jit/codegen/cuda/scheduler/registry.h 121 torch/csrc/jit/codegen/cuda/scheduler/utils.cpp 1032 torch/csrc/jit/codegen/cuda/scheduler/utils.h 122 torch/csrc/jit/codegen/cuda/tensor_view.cpp 654 torch/csrc/jit/codegen/cuda/tools/stringify_file.py 21 torch/csrc/jit/codegen/cuda/transform_iter.cpp 525 torch/csrc/jit/codegen/cuda/transform_iter.h 117 torch/csrc/jit/codegen/cuda/transform_replay.cpp 540 torch/csrc/jit/codegen/cuda/transform_replay.h 52 torch/csrc/jit/codegen/cuda/transform_rfactor.cpp 296 torch/csrc/jit/codegen/cuda/transform_rfactor.h 19 torch/csrc/jit/codegen/cuda/transform_view.cpp 520 torch/csrc/jit/codegen/cuda/transform_view.h 27 torch/csrc/jit/codegen/cuda/type.cpp 664 torch/csrc/jit/codegen/cuda/type.h 207 torch/csrc/jit/codegen/cuda/type_inference.cpp 524 torch/csrc/jit/codegen/cuda/type_inference.h 12 torch/csrc/jit/codegen/cuda/type_promotion.cpp 178 torch/csrc/jit/codegen/cuda/type_promotion.h 36 torch/csrc/jit/codegen/cuda/utils.cpp 149 torch/csrc/jit/codegen/cuda/utils.h 75 torch/csrc/jit/codegen/fuser/arg_spec.h 42 torch/csrc/jit/codegen/fuser/codegen.cpp 576 torch/csrc/jit/codegen/fuser/codegen.h 23 torch/csrc/jit/codegen/fuser/compiler.cpp 229 torch/csrc/jit/codegen/fuser/compiler.h 44 torch/csrc/jit/codegen/fuser/cpu/fused_kernel.cpp 299 torch/csrc/jit/codegen/fuser/cpu/fused_kernel.h 39 torch/csrc/jit/codegen/fuser/cpu/resource_strings.h 81 torch/csrc/jit/codegen/fuser/cpu/temp_file.h 115 torch/csrc/jit/codegen/fuser/cuda/fused_kernel.cpp 215 torch/csrc/jit/codegen/fuser/cuda/fused_kernel.h 49 torch/csrc/jit/codegen/fuser/cuda/resource_strings.h 347 torch/csrc/jit/codegen/fuser/executor.cpp 317 torch/csrc/jit/codegen/fuser/executor.h 16 torch/csrc/jit/codegen/fuser/fallback.cpp 41 torch/csrc/jit/codegen/fuser/fallback.h 10 torch/csrc/jit/codegen/fuser/fused_kernel.h 65 torch/csrc/jit/codegen/fuser/interface.cpp 81 torch/csrc/jit/codegen/fuser/interface.h 27 torch/csrc/jit/codegen/fuser/kernel_cache.cpp 67 torch/csrc/jit/codegen/fuser/kernel_cache.h 19 torch/csrc/jit/codegen/fuser/kernel_spec.h 114 torch/csrc/jit/codegen/fuser/partition_desc.h 45 torch/csrc/jit/codegen/fuser/tensor_desc.h 79 torch/csrc/jit/codegen/fuser/tensor_info.h 21 torch/csrc/jit/cuda/cuda.h 141 torch/csrc/jit/frontend/builtin_functions.cpp 211 torch/csrc/jit/frontend/builtin_functions.h 8 torch/csrc/jit/frontend/canonicalize_modified_loop.cpp 56 torch/csrc/jit/frontend/canonicalize_modified_loop.h 9 torch/csrc/jit/frontend/concrete_module_type.cpp 311 torch/csrc/jit/frontend/concrete_module_type.h 124 torch/csrc/jit/frontend/convert_to_ssa.cpp 264 torch/csrc/jit/frontend/convert_to_ssa.h 11 torch/csrc/jit/frontend/edit_distance.cpp 42 torch/csrc/jit/frontend/edit_distance.h 11 torch/csrc/jit/frontend/error_report.cpp 67 torch/csrc/jit/frontend/error_report.h 36 torch/csrc/jit/frontend/exit_transforms.cpp 473 torch/csrc/jit/frontend/exit_transforms.h 8 torch/csrc/jit/frontend/function_schema_parser.cpp 337 torch/csrc/jit/frontend/function_schema_parser.h 13 torch/csrc/jit/frontend/inline_loop_condition.cpp 41 torch/csrc/jit/frontend/inline_loop_condition.h 12 torch/csrc/jit/frontend/ir_emitter.cpp 4328 torch/csrc/jit/frontend/ir_emitter.h 16 torch/csrc/jit/frontend/lexer.cpp 95 torch/csrc/jit/frontend/lexer.h 445 torch/csrc/jit/frontend/mini_environment.h 41 torch/csrc/jit/frontend/name_mangler.cpp 26 torch/csrc/jit/frontend/name_mangler.h 13 torch/csrc/jit/frontend/parse_string_literal.h 77 torch/csrc/jit/frontend/parser.cpp 708 torch/csrc/jit/frontend/parser.h 27 torch/csrc/jit/frontend/parser_constants.h 6 torch/csrc/jit/frontend/resolver.h 40 torch/csrc/jit/frontend/schema_matching.cpp 593 torch/csrc/jit/frontend/schema_matching.h 50 torch/csrc/jit/frontend/schema_type_parser.cpp 376 torch/csrc/jit/frontend/schema_type_parser.h 32 torch/csrc/jit/frontend/script_type_parser.cpp 403 torch/csrc/jit/frontend/script_type_parser.h 36 torch/csrc/jit/frontend/source_range.cpp 172 torch/csrc/jit/frontend/source_range.h 216 torch/csrc/jit/frontend/source_ref.h 32 torch/csrc/jit/frontend/strtod.cpp 29 torch/csrc/jit/frontend/strtod.h 8 torch/csrc/jit/frontend/sugared_value.cpp 639 torch/csrc/jit/frontend/sugared_value.h 612 torch/csrc/jit/frontend/tracer.cpp 897 torch/csrc/jit/frontend/tracer.h 314 torch/csrc/jit/frontend/tree.h 186 torch/csrc/jit/frontend/tree_views.cpp 52 torch/csrc/jit/frontend/tree_views.h 1021 torch/csrc/jit/frontend/versioned_symbols.cpp 55 torch/csrc/jit/frontend/versioned_symbols.h 12 torch/csrc/jit/ir/alias_analysis.cpp 1419 torch/csrc/jit/ir/alias_analysis.h 129 torch/csrc/jit/ir/attributes.cpp 17 torch/csrc/jit/ir/attributes.h 159 torch/csrc/jit/ir/constants.cpp 202 torch/csrc/jit/ir/constants.h 34 torch/csrc/jit/ir/graph_node_list.h 145 torch/csrc/jit/ir/ir.cpp 1885 torch/csrc/jit/ir/ir.h 1233 torch/csrc/jit/ir/ir_views.h 137 torch/csrc/jit/ir/irparser.cpp 401 torch/csrc/jit/ir/irparser.h 16 torch/csrc/jit/ir/named_value.h 61 torch/csrc/jit/ir/node_hashing.cpp 236 torch/csrc/jit/ir/node_hashing.h 12 torch/csrc/jit/ir/scope.cpp 168 torch/csrc/jit/ir/scope.h 110 torch/csrc/jit/ir/subgraph_matcher.cpp 277 torch/csrc/jit/ir/subgraph_matcher.h 15 torch/csrc/jit/ir/type_hashing.cpp 37 torch/csrc/jit/ir/type_hashing.h 15 torch/csrc/jit/jit_log.cpp 155 torch/csrc/jit/jit_log.h 63 torch/csrc/jit/jit_opt_limit.cpp 72 torch/csrc/jit/jit_opt_limit.h 10 torch/csrc/jit/mobile/code.h 25 torch/csrc/jit/mobile/compatibility/backport.cpp 88 torch/csrc/jit/mobile/compatibility/backport.h 30 torch/csrc/jit/mobile/compatibility/backport_manager.cpp 374 torch/csrc/jit/mobile/compatibility/backport_manager.h 40 torch/csrc/jit/mobile/compatibility/model_compatibility.cpp 316 torch/csrc/jit/mobile/compatibility/model_compatibility.h 68 torch/csrc/jit/mobile/compatibility/runtime_compatibility.cpp 78 torch/csrc/jit/mobile/compatibility/runtime_compatibility.h 28 torch/csrc/jit/mobile/debug_info.cpp 189 torch/csrc/jit/mobile/debug_info.h 39 torch/csrc/jit/mobile/flatbuffer_loader.cpp 459 torch/csrc/jit/mobile/flatbuffer_loader.h 25 torch/csrc/jit/mobile/frame.h 41 torch/csrc/jit/mobile/function.cpp 216 torch/csrc/jit/mobile/function.h 61 torch/csrc/jit/mobile/import.cpp 510 torch/csrc/jit/mobile/import.h 62 torch/csrc/jit/mobile/import_data.cpp 158 torch/csrc/jit/mobile/import_data.h 21 torch/csrc/jit/mobile/interpreter.cpp 265 torch/csrc/jit/mobile/interpreter.h 23 torch/csrc/jit/mobile/method.h 30 torch/csrc/jit/mobile/model_tracer/BuildFeatureTracer.cpp 11 torch/csrc/jit/mobile/model_tracer/BuildFeatureTracer.h 30 torch/csrc/jit/mobile/model_tracer/CustomClassTracer.cpp 11 torch/csrc/jit/mobile/model_tracer/CustomClassTracer.h 30 torch/csrc/jit/mobile/model_tracer/KernelDTypeTracer.cpp 14 torch/csrc/jit/mobile/model_tracer/KernelDTypeTracer.h 33 torch/csrc/jit/mobile/model_tracer/MobileModelRunner.cpp 200 torch/csrc/jit/mobile/model_tracer/MobileModelRunner.h 62 torch/csrc/jit/mobile/model_tracer/OperatorCallTracer.cpp 8 torch/csrc/jit/mobile/model_tracer/OperatorCallTracer.h 31 torch/csrc/jit/mobile/model_tracer/TensorUtils.cpp 39 torch/csrc/jit/mobile/model_tracer/TensorUtils.h 11 torch/csrc/jit/mobile/model_tracer/TracerRunner.cpp 217 torch/csrc/jit/mobile/model_tracer/TracerRunner.h 24 torch/csrc/jit/mobile/model_tracer/tracer.cpp 115 torch/csrc/jit/mobile/module.cpp 218 torch/csrc/jit/mobile/module.h 104 torch/csrc/jit/mobile/nnc/aot_compiler.cpp 132 torch/csrc/jit/mobile/nnc/aot_compiler.h 18 torch/csrc/jit/mobile/nnc/backend.cpp 43 torch/csrc/jit/mobile/nnc/context.cpp 236 torch/csrc/jit/mobile/nnc/context.h 118 torch/csrc/jit/mobile/nnc/registry.cpp 10 torch/csrc/jit/mobile/nnc/registry.h 35 torch/csrc/jit/mobile/observer.cpp 7 torch/csrc/jit/mobile/observer.h 71 torch/csrc/jit/mobile/parse_bytecode.cpp 158 torch/csrc/jit/mobile/parse_bytecode.h 24 torch/csrc/jit/mobile/parse_operators.cpp 71 torch/csrc/jit/mobile/parse_operators.h 19 torch/csrc/jit/mobile/prim_ops_registery.cpp 28 torch/csrc/jit/mobile/prim_ops_registery.h 24 torch/csrc/jit/mobile/profiler_edge.cpp 113 torch/csrc/jit/mobile/profiler_edge.h 47 torch/csrc/jit/mobile/promoted_prim_ops.cpp 155 torch/csrc/jit/mobile/promoted_prim_ops.h 25 torch/csrc/jit/mobile/train/export_data.cpp 80 torch/csrc/jit/mobile/train/export_data.h 12 torch/csrc/jit/mobile/train/optim/sgd.cpp 116 torch/csrc/jit/mobile/train/optim/sgd.h 103 torch/csrc/jit/mobile/train/random.cpp 42 torch/csrc/jit/mobile/train/random.h 31 torch/csrc/jit/mobile/train/sequential.cpp 40 torch/csrc/jit/mobile/train/sequential.h 30 torch/csrc/jit/mobile/type_parser.cpp 271 torch/csrc/jit/mobile/type_parser.h 37 torch/csrc/jit/mobile/upgrader_mobile.cpp 267 torch/csrc/jit/mobile/upgrader_mobile.h 33 torch/csrc/jit/operator_upgraders/upgraders.cpp 71 torch/csrc/jit/operator_upgraders/upgraders.h 37 torch/csrc/jit/operator_upgraders/upgraders_entry.cpp 92 torch/csrc/jit/operator_upgraders/upgraders_entry.h 18 torch/csrc/jit/operator_upgraders/upgraders_guard.cpp 9 torch/csrc/jit/operator_upgraders/upgraders_guard.h 7 torch/csrc/jit/operator_upgraders/utils.cpp 61 torch/csrc/jit/operator_upgraders/utils.h 24 torch/csrc/jit/operator_upgraders/version_map.cpp 65 torch/csrc/jit/operator_upgraders/version_map.h 21 torch/csrc/jit/passes/annotate_warns.cpp 22 torch/csrc/jit/passes/annotate_warns.h 7 torch/csrc/jit/passes/autocast.cpp 365 torch/csrc/jit/passes/autocast.h 9 torch/csrc/jit/passes/bailout_graph.cpp 274 torch/csrc/jit/passes/bailout_graph.h 18 torch/csrc/jit/passes/batch_mm.cpp 378 torch/csrc/jit/passes/batch_mm.h 7 torch/csrc/jit/passes/canonicalize.cpp 166 torch/csrc/jit/passes/canonicalize.h 15 torch/csrc/jit/passes/canonicalize_graph_fuser_ops.cpp 89 torch/csrc/jit/passes/canonicalize_graph_fuser_ops.h 7 torch/csrc/jit/passes/clear_profiling.cpp 34 torch/csrc/jit/passes/clear_profiling.h 13 torch/csrc/jit/passes/clear_undefinedness.cpp 33 torch/csrc/jit/passes/clear_undefinedness.h 11 torch/csrc/jit/passes/common_expression_hoisting.cpp 112 torch/csrc/jit/passes/common_expression_hoisting.h 7 torch/csrc/jit/passes/common_subexpression_elimination.cpp 97 torch/csrc/jit/passes/common_subexpression_elimination.h 8 torch/csrc/jit/passes/concat_opt.cpp 447 torch/csrc/jit/passes/concat_opt.h 10 torch/csrc/jit/passes/constant_pooling.cpp 54 torch/csrc/jit/passes/constant_pooling.h 7 torch/csrc/jit/passes/constant_propagation.cpp 374 torch/csrc/jit/passes/constant_propagation.h 14 torch/csrc/jit/passes/create_autodiff_subgraphs.cpp 224 torch/csrc/jit/passes/create_autodiff_subgraphs.h 11 torch/csrc/jit/passes/create_functional_graphs.cpp 166 torch/csrc/jit/passes/create_functional_graphs.h 9 torch/csrc/jit/passes/cuda_graph_fuser.h 27 torch/csrc/jit/passes/dead_code_elimination.cpp 333 torch/csrc/jit/passes/dead_code_elimination.h 24 torch/csrc/jit/passes/decompose_ops.cpp 195 torch/csrc/jit/passes/decompose_ops.h 7 torch/csrc/jit/passes/device_type_analysis.cpp 207 torch/csrc/jit/passes/device_type_analysis.h 8 torch/csrc/jit/passes/dtype_analysis.cpp 268 torch/csrc/jit/passes/dtype_analysis.h 10 torch/csrc/jit/passes/eliminate_no_ops.cpp 59 torch/csrc/jit/passes/eliminate_no_ops.h 9 torch/csrc/jit/passes/erase_number_types.cpp 68 torch/csrc/jit/passes/erase_number_types.h 8 torch/csrc/jit/passes/fixup_trace_scope_blocks.cpp 361 torch/csrc/jit/passes/fixup_trace_scope_blocks.h 10 torch/csrc/jit/passes/fold_conv_bn.cpp 319 torch/csrc/jit/passes/fold_conv_bn.h 18 torch/csrc/jit/passes/freeze_module.cpp 694 torch/csrc/jit/passes/freeze_module.h 17 torch/csrc/jit/passes/frozen_concat_linear.cpp 201 torch/csrc/jit/passes/frozen_concat_linear.h 7 torch/csrc/jit/passes/frozen_conv_add_relu_fusion.cpp 26 torch/csrc/jit/passes/frozen_conv_add_relu_fusion.h 10 torch/csrc/jit/passes/frozen_conv_add_relu_fusion_cuda.cpp 96 torch/csrc/jit/passes/frozen_conv_folding.cpp 303 torch/csrc/jit/passes/frozen_conv_folding.h 9 torch/csrc/jit/passes/frozen_graph_optimizations.cpp 29 torch/csrc/jit/passes/frozen_graph_optimizations.h 9 torch/csrc/jit/passes/frozen_linear_transpose.cpp 81 torch/csrc/jit/passes/frozen_linear_transpose.h 7 torch/csrc/jit/passes/frozen_ops_to_mkldnn.cpp 914 torch/csrc/jit/passes/frozen_ops_to_mkldnn.h 7 torch/csrc/jit/passes/fuse_linear.cpp 83 torch/csrc/jit/passes/fuse_linear.h 7 torch/csrc/jit/passes/fuse_relu.cpp 57 torch/csrc/jit/passes/fuse_relu.h 9 torch/csrc/jit/passes/graph_fuser.cpp 918 torch/csrc/jit/passes/graph_fuser.h 16 torch/csrc/jit/passes/graph_rewrite_helper.cpp 228 torch/csrc/jit/passes/graph_rewrite_helper.h 41 torch/csrc/jit/passes/guard_elimination.cpp 362 torch/csrc/jit/passes/guard_elimination.h 14 torch/csrc/jit/passes/hoist_conv_packed_params.cpp 78 torch/csrc/jit/passes/hoist_conv_packed_params.h 8 torch/csrc/jit/passes/inline_autodiff_subgraphs.cpp 66 torch/csrc/jit/passes/inline_autodiff_subgraphs.h 10 torch/csrc/jit/passes/inline_fork_wait.cpp 48 torch/csrc/jit/passes/inline_fork_wait.h 7 torch/csrc/jit/passes/inline_forked_closures.cpp 62 torch/csrc/jit/passes/inline_forked_closures.h 8 torch/csrc/jit/passes/inliner.cpp 71 torch/csrc/jit/passes/inliner.h 7 torch/csrc/jit/passes/inplace_check.cpp 19 torch/csrc/jit/passes/inplace_check.h 7 torch/csrc/jit/passes/insert_guards.cpp 43 torch/csrc/jit/passes/insert_guards.h 15 torch/csrc/jit/passes/integer_value_refinement.cpp 164 torch/csrc/jit/passes/integer_value_refinement.h 7 torch/csrc/jit/passes/lift_closures.cpp 62 torch/csrc/jit/passes/lift_closures.h 8 torch/csrc/jit/passes/liveness.cpp 130 torch/csrc/jit/passes/liveness.h 18 torch/csrc/jit/passes/loop_unrolling.cpp 297 torch/csrc/jit/passes/loop_unrolling.h 23 torch/csrc/jit/passes/lower_grad_of.cpp 31 torch/csrc/jit/passes/lower_grad_of.h 7 torch/csrc/jit/passes/lower_graph.cpp 134 torch/csrc/jit/passes/lower_graph.h 10 torch/csrc/jit/passes/lower_tuples.cpp 270 torch/csrc/jit/passes/lower_tuples.h 9 torch/csrc/jit/passes/metal_rewrite.cpp 270 torch/csrc/jit/passes/metal_rewrite.h 16 torch/csrc/jit/passes/normalize_ops.cpp 147 torch/csrc/jit/passes/normalize_ops.h 8 torch/csrc/jit/passes/onnx.cpp 405 torch/csrc/jit/passes/onnx.h 26 torch/csrc/jit/passes/onnx/cast_all_constant_to_floating.cpp 57 torch/csrc/jit/passes/onnx/cast_all_constant_to_floating.h 8 torch/csrc/jit/passes/onnx/constant_fold.cpp 566 torch/csrc/jit/passes/onnx/constant_fold.h 25 torch/csrc/jit/passes/onnx/constant_map.cpp 276 torch/csrc/jit/passes/onnx/constant_map.h 61 torch/csrc/jit/passes/onnx/eliminate_unused_items.cpp 18 torch/csrc/jit/passes/onnx/eliminate_unused_items.h 9 torch/csrc/jit/passes/onnx/eval_peephole.cpp 126 torch/csrc/jit/passes/onnx/eval_peephole.h 10 torch/csrc/jit/passes/onnx/fixup_onnx_controlflow.cpp 387 torch/csrc/jit/passes/onnx/fixup_onnx_controlflow.h 8 torch/csrc/jit/passes/onnx/function_extraction.cpp 876 torch/csrc/jit/passes/onnx/function_extraction.h 14 torch/csrc/jit/passes/onnx/function_substitution.cpp 77 torch/csrc/jit/passes/onnx/function_substitution.h 7 torch/csrc/jit/passes/onnx/helper.cpp 230 torch/csrc/jit/passes/onnx/helper.h 45 torch/csrc/jit/passes/onnx/list_model_parameters.cpp 154 torch/csrc/jit/passes/onnx/list_model_parameters.h 9 torch/csrc/jit/passes/onnx/pattern_conversion/common.cpp 27 torch/csrc/jit/passes/onnx/pattern_conversion/common.h 12 torch/csrc/jit/passes/onnx/pattern_conversion/pattern_conversion.cpp 268 torch/csrc/jit/passes/onnx/pattern_conversion/pattern_conversion.h 10 torch/csrc/jit/passes/onnx/pattern_conversion/pattern_encapsulation.cpp 56 torch/csrc/jit/passes/onnx/pattern_conversion/pattern_encapsulation.h 7 torch/csrc/jit/passes/onnx/peephole.cpp 728 torch/csrc/jit/passes/onnx/peephole.h 10 torch/csrc/jit/passes/onnx/prepare_division_for_onnx.cpp 37 torch/csrc/jit/passes/onnx/prepare_division_for_onnx.h 7 torch/csrc/jit/passes/onnx/preprocess_for_onnx.cpp 133 torch/csrc/jit/passes/onnx/preprocess_for_onnx.h 7 torch/csrc/jit/passes/onnx/remove_inplace_ops_for_onnx.cpp 608 torch/csrc/jit/passes/onnx/remove_inplace_ops_for_onnx.h 9 torch/csrc/jit/passes/onnx/scalar_type_analysis.cpp 319 torch/csrc/jit/passes/onnx/scalar_type_analysis.h 11 torch/csrc/jit/passes/onnx/shape_type_inference.cpp 2036 torch/csrc/jit/passes/onnx/shape_type_inference.h 36 torch/csrc/jit/passes/onnx/unpack_quantized_weights.cpp 476 torch/csrc/jit/passes/onnx/unpack_quantized_weights.h 14 torch/csrc/jit/passes/pass_manager.cpp 56 torch/csrc/jit/passes/pass_manager.h 59 torch/csrc/jit/passes/peephole.cpp 375 torch/csrc/jit/passes/peephole.h 13 torch/csrc/jit/passes/peephole_alias_sensitive.cpp 140 torch/csrc/jit/passes/peephole_alias_sensitive.h 9 torch/csrc/jit/passes/peephole_dict_idioms.cpp 226 torch/csrc/jit/passes/peephole_dict_idioms.h 7 torch/csrc/jit/passes/peephole_list_idioms.cpp 282 torch/csrc/jit/passes/peephole_list_idioms.h 9 torch/csrc/jit/passes/peephole_non_tensor.cpp 222 torch/csrc/jit/passes/peephole_non_tensor.h 7 torch/csrc/jit/passes/prepack_folding.cpp 63 torch/csrc/jit/passes/prepack_folding.h 12 torch/csrc/jit/passes/quantization/dedup_module_uses.cpp 91 torch/csrc/jit/passes/quantization/dedup_module_uses.h 7 torch/csrc/jit/passes/quantization/finalize.cpp 85 torch/csrc/jit/passes/quantization/finalize.h 19 torch/csrc/jit/passes/quantization/fusion_passes.cpp 59 torch/csrc/jit/passes/quantization/fusion_passes.h 7 torch/csrc/jit/passes/quantization/helper.cpp 631 torch/csrc/jit/passes/quantization/helper.h 104 torch/csrc/jit/passes/quantization/insert_observers.cpp 1236 torch/csrc/jit/passes/quantization/insert_observers.h 23 torch/csrc/jit/passes/quantization/insert_quant_dequant.cpp 1300 torch/csrc/jit/passes/quantization/insert_quant_dequant.h 18 torch/csrc/jit/passes/quantization/quantization_patterns.h 991 torch/csrc/jit/passes/quantization/quantization_type.cpp 18 torch/csrc/jit/passes/quantization/quantization_type.h 8 torch/csrc/jit/passes/remove_dropout.cpp 53 torch/csrc/jit/passes/remove_dropout.h 9 torch/csrc/jit/passes/remove_exceptions.cpp 42 torch/csrc/jit/passes/remove_exceptions.h 7 torch/csrc/jit/passes/remove_expands.cpp 19 torch/csrc/jit/passes/remove_expands.h 7 torch/csrc/jit/passes/remove_inplace_ops.cpp 94 torch/csrc/jit/passes/remove_inplace_ops.h 9 torch/csrc/jit/passes/remove_mutation.cpp 276 torch/csrc/jit/passes/remove_mutation.h 55 torch/csrc/jit/passes/remove_redundant_profiles.cpp 36 torch/csrc/jit/passes/remove_redundant_profiles.h 8 torch/csrc/jit/passes/replacement_of_old_operators.cpp 85 torch/csrc/jit/passes/replacement_of_old_operators.h 8 torch/csrc/jit/passes/requires_grad_analysis.cpp 138 torch/csrc/jit/passes/requires_grad_analysis.h 10 torch/csrc/jit/passes/restore_mutation.cpp 61 torch/csrc/jit/passes/restore_mutation.h 46 torch/csrc/jit/passes/shape_analysis.cpp 1821 torch/csrc/jit/passes/shape_analysis.h 30 torch/csrc/jit/passes/specialize_autogradzero.cpp 379 torch/csrc/jit/passes/specialize_autogradzero.h 9 torch/csrc/jit/passes/subgraph_rewrite.cpp 166 torch/csrc/jit/passes/subgraph_rewrite.h 49 torch/csrc/jit/passes/symbolic_shape_analysis.cpp 696 torch/csrc/jit/passes/symbolic_shape_analysis.h 32 torch/csrc/jit/passes/symbolic_shape_runtime_fusion.cpp 557 torch/csrc/jit/passes/symbolic_shape_runtime_fusion.h 23 torch/csrc/jit/passes/tensorexpr_fuser.cpp 1000 torch/csrc/jit/passes/tensorexpr_fuser.h 34 torch/csrc/jit/passes/update_differentiable_graph_requires_grad.cpp 35 torch/csrc/jit/passes/update_differentiable_graph_requires_grad.h 9 torch/csrc/jit/passes/utils/check_alias_annotation.cpp 234 torch/csrc/jit/passes/utils/check_alias_annotation.h 14 torch/csrc/jit/passes/utils/memory_dag.cpp 177 torch/csrc/jit/passes/utils/memory_dag.h 79 torch/csrc/jit/passes/utils/op_registry.cpp 53 torch/csrc/jit/passes/utils/op_registry.h 10 torch/csrc/jit/passes/utils/optimization_utils.cpp 13 torch/csrc/jit/passes/utils/optimization_utils.h 7 torch/csrc/jit/passes/utils/subgraph_utils.cpp 500 torch/csrc/jit/passes/utils/subgraph_utils.h 35 torch/csrc/jit/passes/value_refinement_utils.cpp 143 torch/csrc/jit/passes/value_refinement_utils.h 60 torch/csrc/jit/passes/variadic_ops.cpp 200 torch/csrc/jit/passes/variadic_ops.h 20 torch/csrc/jit/passes/vulkan_rewrite.cpp 255 torch/csrc/jit/passes/vulkan_rewrite.h 14 torch/csrc/jit/passes/xnnpack_rewrite.cpp 450 torch/csrc/jit/passes/xnnpack_rewrite.h 25 torch/csrc/jit/python/init.cpp 1529 torch/csrc/jit/python/init.h 7 torch/csrc/jit/python/module_python.h 23 torch/csrc/jit/python/pybind.h 179 torch/csrc/jit/python/pybind_utils.cpp 316 torch/csrc/jit/python/pybind_utils.h 922 torch/csrc/jit/python/python_arg_flatten.cpp 173 torch/csrc/jit/python/python_arg_flatten.h 96 torch/csrc/jit/python/python_custom_class.cpp 74 torch/csrc/jit/python/python_custom_class.h 15 torch/csrc/jit/python/python_dict.cpp 153 torch/csrc/jit/python/python_dict.h 85 torch/csrc/jit/python/python_interpreter.cpp 71 torch/csrc/jit/python/python_ir.cpp 977 torch/csrc/jit/python/python_ir.h 32 torch/csrc/jit/python/python_ivalue.h 63 torch/csrc/jit/python/python_list.cpp 275 torch/csrc/jit/python/python_list.h 155 torch/csrc/jit/python/python_sugared_value.cpp 1043 torch/csrc/jit/python/python_sugared_value.h 268 torch/csrc/jit/python/python_tracer.cpp 199 torch/csrc/jit/python/python_tracer.h 29 torch/csrc/jit/python/python_tree_views.cpp 382 torch/csrc/jit/python/python_tree_views.h 7 torch/csrc/jit/python/script_init.cpp 1978 torch/csrc/jit/python/script_init.h 7 torch/csrc/jit/python/update_graph_executor_opt.cpp 12 torch/csrc/jit/python/update_graph_executor_opt.h 8 torch/csrc/jit/resource_guard.h 20 torch/csrc/jit/runtime/argument_spec.cpp 240 torch/csrc/jit/runtime/argument_spec.h 397 torch/csrc/jit/runtime/autodiff.cpp 575 torch/csrc/jit/runtime/autodiff.h 27 torch/csrc/jit/runtime/calculate_necessary_args.h 50 torch/csrc/jit/runtime/custom_operator.h 18 torch/csrc/jit/runtime/exception_message.h 26 torch/csrc/jit/runtime/graph_executor.cpp 793 torch/csrc/jit/runtime/graph_executor.h 86 torch/csrc/jit/runtime/graph_executor_impl.h 78 torch/csrc/jit/runtime/graph_iterator.h 82 torch/csrc/jit/runtime/instruction.cpp 79 torch/csrc/jit/runtime/instruction.h 62 torch/csrc/jit/runtime/interpreter.cpp 872 torch/csrc/jit/runtime/interpreter.h 125 torch/csrc/jit/runtime/interpreter/can_emit_inline.h 58 torch/csrc/jit/runtime/interpreter/code_impl.h 740 torch/csrc/jit/runtime/interpreter/frame.cpp 12 torch/csrc/jit/runtime/interpreter/frame.h 20 torch/csrc/jit/runtime/interpreter/preprocess_graph.cpp 138 torch/csrc/jit/runtime/interpreter/preprocess_graph.h 15 torch/csrc/jit/runtime/jit_exception.cpp 6 torch/csrc/jit/runtime/jit_exception.h 10 torch/csrc/jit/runtime/jit_trace.cpp 225 torch/csrc/jit/runtime/jit_trace.h 9 torch/csrc/jit/runtime/logging.cpp 58 torch/csrc/jit/runtime/logging.h 67 torch/csrc/jit/runtime/operator.cpp 367 torch/csrc/jit/runtime/operator.h 171 torch/csrc/jit/runtime/operator_options.h 7 torch/csrc/jit/runtime/print_handler.cpp 16 torch/csrc/jit/runtime/print_handler.h 12 torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp 560 torch/csrc/jit/runtime/profiling_graph_executor_impl.h 41 torch/csrc/jit/runtime/profiling_record.cpp 264 torch/csrc/jit/runtime/profiling_record.h 88 torch/csrc/jit/runtime/register_c10_ops.cpp 45 torch/csrc/jit/runtime/register_cuda_ops.cpp 144 torch/csrc/jit/runtime/register_distributed_ops.cpp 247 torch/csrc/jit/runtime/register_ops_utils.cpp 399 torch/csrc/jit/runtime/register_ops_utils.h 764 torch/csrc/jit/runtime/register_prim_ops.cpp 2958 torch/csrc/jit/runtime/register_prim_ops_fulljit.cpp 751 torch/csrc/jit/runtime/register_special_ops.cpp 412 torch/csrc/jit/runtime/script_profile.cpp 147 torch/csrc/jit/runtime/script_profile.h 64 torch/csrc/jit/runtime/shape_functions.h 393 torch/csrc/jit/runtime/shape_functions_1.h 8 torch/csrc/jit/runtime/slice_indices_adjust.cpp 47 torch/csrc/jit/runtime/slice_indices_adjust.h 13 torch/csrc/jit/runtime/static/ProcessedNodeInputs.cpp 1 torch/csrc/jit/runtime/static/ProcessedNodeInputs.h 176 torch/csrc/jit/runtime/static/fusion.cpp 276 torch/csrc/jit/runtime/static/fusion.h 12 torch/csrc/jit/runtime/static/impl.cpp 1550 torch/csrc/jit/runtime/static/impl.h 489 torch/csrc/jit/runtime/static/init.cpp 114 torch/csrc/jit/runtime/static/init.h 6 torch/csrc/jit/runtime/static/memory_planner.cpp 343 torch/csrc/jit/runtime/static/memory_planner.h 128 torch/csrc/jit/runtime/static/native_ops.cpp 620 torch/csrc/jit/runtime/static/ops.cpp 2339 torch/csrc/jit/runtime/static/ops.h 146 torch/csrc/jit/runtime/static/passes.cpp 746 torch/csrc/jit/runtime/static/passes.h 28 torch/csrc/jit/runtime/static/static_method.h 42 torch/csrc/jit/runtime/static/te_wrapper.cpp 214 torch/csrc/jit/runtime/static/te_wrapper.h 36 torch/csrc/jit/runtime/symbolic_script.cpp 1296 torch/csrc/jit/runtime/symbolic_script.h 15 torch/csrc/jit/runtime/symbolic_shape_registry.cpp 233 torch/csrc/jit/runtime/symbolic_shape_registry.h 9 torch/csrc/jit/runtime/symbolic_shape_registry_util.cpp 107 torch/csrc/jit/runtime/symbolic_shape_registry_util.h 8 torch/csrc/jit/runtime/vararg_functions.cpp 375 torch/csrc/jit/runtime/vararg_functions.h 26 torch/csrc/jit/runtime/variable_tensor_list.h 14 torch/csrc/jit/serialization/callstack_debug_info_serialization.cpp 212 torch/csrc/jit/serialization/callstack_debug_info_serialization.h 62 torch/csrc/jit/serialization/export.cpp 1114 torch/csrc/jit/serialization/export.h 157 torch/csrc/jit/serialization/export_bytecode.cpp 325 torch/csrc/jit/serialization/export_bytecode.h 36 torch/csrc/jit/serialization/export_module.cpp 658 torch/csrc/jit/serialization/flatbuffer_serializer.cpp 581 torch/csrc/jit/serialization/flatbuffer_serializer.h 20 torch/csrc/jit/serialization/import.cpp 338 torch/csrc/jit/serialization/import.h 69 torch/csrc/jit/serialization/import_export_constants.h 17 torch/csrc/jit/serialization/import_export_functions.h 14 torch/csrc/jit/serialization/import_export_helpers.cpp 45 torch/csrc/jit/serialization/import_export_helpers.h 20 torch/csrc/jit/serialization/import_legacy.cpp 337 torch/csrc/jit/serialization/import_legacy.h 16 torch/csrc/jit/serialization/import_read.cpp 62 torch/csrc/jit/serialization/import_read.h 25 torch/csrc/jit/serialization/import_source.cpp 642 torch/csrc/jit/serialization/import_source.h 80 torch/csrc/jit/serialization/mobile_bytecode_generated.h 2339 torch/csrc/jit/serialization/onnx.cpp 209 torch/csrc/jit/serialization/onnx.h 8 torch/csrc/jit/serialization/pickle.cpp 129 torch/csrc/jit/serialization/pickle.h 33 torch/csrc/jit/serialization/pickler.cpp 590 torch/csrc/jit/serialization/pickler.h 207 torch/csrc/jit/serialization/python_print.cpp 1304 torch/csrc/jit/serialization/python_print.h 41 torch/csrc/jit/serialization/source_range_serialization.cpp 126 torch/csrc/jit/serialization/source_range_serialization.h 43 torch/csrc/jit/serialization/source_range_serialization_impl.h 19 torch/csrc/jit/serialization/storage_context.h 65 torch/csrc/jit/serialization/type_name_uniquer.cpp 24 torch/csrc/jit/serialization/type_name_uniquer.h 20 torch/csrc/jit/serialization/unpickler.cpp 878 torch/csrc/jit/serialization/unpickler.h 111 torch/csrc/jit/tensorexpr/analysis.h 294 torch/csrc/jit/tensorexpr/block_codegen.cpp 315 torch/csrc/jit/tensorexpr/block_codegen.h 118 torch/csrc/jit/tensorexpr/bounds_inference.cpp 319 torch/csrc/jit/tensorexpr/bounds_inference.h 57 torch/csrc/jit/tensorexpr/bounds_overlap.cpp 187 torch/csrc/jit/tensorexpr/bounds_overlap.h 54 torch/csrc/jit/tensorexpr/codegen.cpp 190 torch/csrc/jit/tensorexpr/codegen.h 179 torch/csrc/jit/tensorexpr/codegen_external.py 56 torch/csrc/jit/tensorexpr/cpp_codegen.cpp 341 torch/csrc/jit/tensorexpr/cpp_codegen.h 63 torch/csrc/jit/tensorexpr/cpp_intrinsics.h 29 torch/csrc/jit/tensorexpr/cuda_codegen.cpp 1027 torch/csrc/jit/tensorexpr/cuda_codegen.h 214 torch/csrc/jit/tensorexpr/cuda_random.h 93 torch/csrc/jit/tensorexpr/dim_arg.h 23 torch/csrc/jit/tensorexpr/eval.cpp 1097 torch/csrc/jit/tensorexpr/eval.h 265 torch/csrc/jit/tensorexpr/exceptions.h 77 torch/csrc/jit/tensorexpr/expr.cpp 353 torch/csrc/jit/tensorexpr/expr.h 348 torch/csrc/jit/tensorexpr/external_functions.cpp 944 torch/csrc/jit/tensorexpr/external_functions.h 71 torch/csrc/jit/tensorexpr/external_functions_codegen.cpp 3297 torch/csrc/jit/tensorexpr/external_functions_registry.cpp 11 torch/csrc/jit/tensorexpr/external_functions_registry.h 27 torch/csrc/jit/tensorexpr/fwd_decls.h 113 torch/csrc/jit/tensorexpr/graph_opt.cpp 402 torch/csrc/jit/tensorexpr/graph_opt.h 27 torch/csrc/jit/tensorexpr/half_support.h 161 torch/csrc/jit/tensorexpr/hash_provider.cpp 294 torch/csrc/jit/tensorexpr/hash_provider.h 234 torch/csrc/jit/tensorexpr/intrinsic_symbols.cpp 271 torch/csrc/jit/tensorexpr/intrinsic_symbols.h 16 torch/csrc/jit/tensorexpr/ir.cpp 208 torch/csrc/jit/tensorexpr/ir.h 778 torch/csrc/jit/tensorexpr/ir_cloner.cpp 288 torch/csrc/jit/tensorexpr/ir_cloner.h 56 torch/csrc/jit/tensorexpr/ir_mutator.cpp 481 torch/csrc/jit/tensorexpr/ir_mutator.h 57 torch/csrc/jit/tensorexpr/ir_printer.cpp 565 torch/csrc/jit/tensorexpr/ir_printer.h 103 torch/csrc/jit/tensorexpr/ir_simplifier.cpp 2198 torch/csrc/jit/tensorexpr/ir_simplifier.h 376 torch/csrc/jit/tensorexpr/ir_verifier.cpp 180 torch/csrc/jit/tensorexpr/ir_verifier.h 50 torch/csrc/jit/tensorexpr/ir_visitor.cpp 211 torch/csrc/jit/tensorexpr/ir_visitor.h 56 torch/csrc/jit/tensorexpr/kernel.cpp 1440 torch/csrc/jit/tensorexpr/kernel.h 257 torch/csrc/jit/tensorexpr/llvm_codegen.cpp 1790 torch/csrc/jit/tensorexpr/llvm_codegen.h 109 torch/csrc/jit/tensorexpr/llvm_jit.cpp 264 torch/csrc/jit/tensorexpr/llvm_jit.h 53 torch/csrc/jit/tensorexpr/loopnest.cpp 2502 torch/csrc/jit/tensorexpr/loopnest.h 121 torch/csrc/jit/tensorexpr/loopnest_randomization.cpp 599 torch/csrc/jit/tensorexpr/loopnest_randomization.h 8 torch/csrc/jit/tensorexpr/lowerings.cpp 1456 torch/csrc/jit/tensorexpr/lowerings.h 39 torch/csrc/jit/tensorexpr/mem_dependency_checker.cpp 944 torch/csrc/jit/tensorexpr/mem_dependency_checker.h 254 torch/csrc/jit/tensorexpr/operators/conv2d.cpp 379 torch/csrc/jit/tensorexpr/operators/conv2d.h 79 torch/csrc/jit/tensorexpr/operators/matmul.cpp 65 torch/csrc/jit/tensorexpr/operators/matmul.h 18 torch/csrc/jit/tensorexpr/operators/misc.cpp 542 torch/csrc/jit/tensorexpr/operators/misc.h 79 torch/csrc/jit/tensorexpr/operators/norm.cpp 56 torch/csrc/jit/tensorexpr/operators/norm.h 13 torch/csrc/jit/tensorexpr/operators/operators.h 9 torch/csrc/jit/tensorexpr/operators/pointwise.cpp 174 torch/csrc/jit/tensorexpr/operators/pointwise.h 64 torch/csrc/jit/tensorexpr/operators/quantization.cpp 720 torch/csrc/jit/tensorexpr/operators/quantization.h 108 torch/csrc/jit/tensorexpr/operators/reduction.cpp 155 torch/csrc/jit/tensorexpr/operators/reduction.h 28 torch/csrc/jit/tensorexpr/operators/softmax.cpp 106 torch/csrc/jit/tensorexpr/operators/softmax.h 12 torch/csrc/jit/tensorexpr/reduction.cpp 26 torch/csrc/jit/tensorexpr/reduction.h 181 torch/csrc/jit/tensorexpr/registerizer.cpp 559 torch/csrc/jit/tensorexpr/registerizer.h 226 torch/csrc/jit/tensorexpr/scripts/bisect.py 53 torch/csrc/jit/tensorexpr/stmt.h 731 torch/csrc/jit/tensorexpr/tensor.cpp 149 torch/csrc/jit/tensorexpr/tensor.h 182 torch/csrc/jit/tensorexpr/tensorexpr_init.cpp 878 torch/csrc/jit/tensorexpr/tensorexpr_init.h 8 torch/csrc/jit/tensorexpr/types.cpp 93 torch/csrc/jit/tensorexpr/types.h 129 torch/csrc/jit/tensorexpr/unique_name_manager.cpp 38 torch/csrc/jit/tensorexpr/unique_name_manager.h 25 torch/csrc/jit/tensorexpr/var_substitutor.h 53 torch/csrc/jit/testing/catch_utils.hpp 6 torch/csrc/jit/testing/file_check.cpp 507 torch/csrc/jit/testing/file_check.h 39 torch/csrc/jit/testing/hooks_for_testing.cpp 25 torch/csrc/jit/testing/hooks_for_testing.h 16 torch/csrc/lazy/backend/backend_data.h 39 torch/csrc/lazy/backend/backend_device.cpp 51 torch/csrc/lazy/backend/backend_device.h 49 torch/csrc/lazy/backend/backend_interface.cpp 38 torch/csrc/lazy/backend/backend_interface.h 56 torch/csrc/lazy/backend/lowering_context.cpp 14 torch/csrc/lazy/backend/lowering_context.h 52 torch/csrc/lazy/core/cache.h 85 torch/csrc/lazy/core/config.cpp 31 torch/csrc/lazy/core/config.h 11 torch/csrc/lazy/core/hash.cpp 71 torch/csrc/lazy/core/hash.h 157 torch/csrc/lazy/core/helpers.cpp 122 torch/csrc/lazy/core/helpers.h 41 torch/csrc/lazy/core/internal_ops/ltc_ops.h 46 torch/csrc/lazy/core/ir.cpp 52 torch/csrc/lazy/core/ir.h 146 torch/csrc/lazy/core/ir_dump_util.cpp 245 torch/csrc/lazy/core/ir_dump_util.h 22 torch/csrc/lazy/core/ir_metadata.cpp 94 torch/csrc/lazy/core/ir_metadata.h 34 torch/csrc/lazy/core/ir_util.cpp 62 torch/csrc/lazy/core/ir_util.h 25 torch/csrc/lazy/core/lazy_graph_executor.cpp 859 torch/csrc/lazy/core/lazy_graph_executor.h 167 torch/csrc/lazy/core/lazy_view.cpp 222 torch/csrc/lazy/core/lazy_view.h 118 torch/csrc/lazy/core/metrics.cpp 324 torch/csrc/lazy/core/metrics.h 157 torch/csrc/lazy/core/multi_wait.cpp 63 torch/csrc/lazy/core/multi_wait.h 29 torch/csrc/lazy/core/permutation_util.cpp 23 torch/csrc/lazy/core/permutation_util.h 26 torch/csrc/lazy/core/shape.cpp 28 torch/csrc/lazy/core/shape.h 28 torch/csrc/lazy/core/tensor.cpp 417 torch/csrc/lazy/core/tensor.h 129 torch/csrc/lazy/core/tensor_impl.cpp 145 torch/csrc/lazy/core/tensor_impl.h 39 torch/csrc/lazy/core/tensor_util.cpp 60 torch/csrc/lazy/core/tensor_util.h 41 torch/csrc/lazy/core/thread_pool.cpp 141 torch/csrc/lazy/core/thread_pool.h 21 torch/csrc/lazy/core/unique.h 41 torch/csrc/lazy/core/util.h 76 torch/csrc/lazy/core/view_ops/as_strided.cpp 44 torch/csrc/lazy/core/view_ops/as_strided.h 32 torch/csrc/lazy/core/view_ops/as_strided_view_update.cpp 32 torch/csrc/lazy/core/view_ops/as_strided_view_update.h 30 torch/csrc/lazy/core/view_ops/diagonal.cpp 52 torch/csrc/lazy/core/view_ops/diagonal.h 29 torch/csrc/lazy/core/view_ops/diagonal_view_update.cpp 27 torch/csrc/lazy/core/view_ops/diagonal_view_update.h 29 torch/csrc/lazy/core/view_ops/narrow.cpp 28 torch/csrc/lazy/core/view_ops/narrow.h 23 torch/csrc/lazy/core/view_ops/narrow_view_update.cpp 24 torch/csrc/lazy/core/view_ops/narrow_view_update.h 19 torch/csrc/lazy/core/view_ops/permute.cpp 29 torch/csrc/lazy/core/view_ops/permute.h 19 torch/csrc/lazy/core/view_ops/resize.cpp 23 torch/csrc/lazy/core/view_ops/resize.h 16 torch/csrc/lazy/core/view_ops/select.cpp 50 torch/csrc/lazy/core/view_ops/select.h 40 torch/csrc/lazy/core/view_ops/select_view_update.cpp 31 torch/csrc/lazy/core/view_ops/select_view_update.h 34 torch/csrc/lazy/core/view_ops/view.cpp 28 torch/csrc/lazy/core/view_ops/view.h 17 torch/csrc/lazy/ts_backend/config.cpp 5 torch/csrc/lazy/ts_backend/config.h 3 torch/csrc/lazy/ts_backend/ops/arithmetic_ir_ops.cpp 36 torch/csrc/lazy/ts_backend/ops/arithmetic_ir_ops.h 10 torch/csrc/lazy/ts_backend/ops/cast.cpp 35 torch/csrc/lazy/ts_backend/ops/cast.h 25 torch/csrc/lazy/ts_backend/ops/device_data.cpp 22 torch/csrc/lazy/ts_backend/ops/device_data.h 18 torch/csrc/lazy/ts_backend/ops/expand.cpp 25 torch/csrc/lazy/ts_backend/ops/expand.h 21 torch/csrc/lazy/ts_backend/ops/generic.cpp 30 torch/csrc/lazy/ts_backend/ops/generic.h 38 torch/csrc/lazy/ts_backend/ops/scalar.cpp 31 torch/csrc/lazy/ts_backend/ops/scalar.h 19 torch/csrc/lazy/ts_backend/ts_lowering_context.h 107 torch/csrc/lazy/ts_backend/ts_node.cpp 112 torch/csrc/lazy/ts_backend/ts_node.h 50 torch/csrc/lazy/ts_backend/ts_node_lowering.h 12 torch/csrc/monitor/counters.cpp 58 torch/csrc/monitor/counters.h 209 torch/csrc/monitor/events.cpp 47 torch/csrc/monitor/events.h 28 torch/csrc/monitor/python_init.cpp 288 torch/csrc/monitor/python_init.h 7 torch/csrc/multiprocessing/init.cpp 44 torch/csrc/multiprocessing/init.h 7 torch/csrc/onnx/init.cpp 45 torch/csrc/onnx/init.h 5 torch/csrc/onnx/onnx.h 16 torch/csrc/profiler/api.cpp 115 torch/csrc/profiler/api.h 111 torch/csrc/profiler/cuda.cpp 87 torch/csrc/profiler/kineto_shim.cpp 224 torch/csrc/profiler/kineto_shim.h 103 torch/csrc/profiler/util.cpp 490 torch/csrc/profiler/util.h 81 torch/csrc/python_dimname.cpp 72 torch/csrc/python_dimname.h 6 torch/csrc/python_headers.h 13 torch/csrc/serialization.cpp 126 torch/csrc/serialization.h 9 torch/csrc/stub.c 15 torch/csrc/tensor/python_tensor.cpp 301 torch/csrc/tensor/python_tensor.h 17 torch/csrc/utils.cpp 219 torch/csrc/utils.h 149 torch/csrc/utils/auto_gil.h 24 torch/csrc/utils/byte_order.cpp 281 torch/csrc/utils/byte_order.h 100 torch/csrc/utils/crash_handler.cpp 140 torch/csrc/utils/crash_handler.h 18 torch/csrc/utils/cuda_enabled.h 12 torch/csrc/utils/cuda_lazy_init.cpp 23 torch/csrc/utils/cuda_lazy_init.h 13 torch/csrc/utils/disable_torch_function.cpp 193 torch/csrc/utils/disable_torch_function.h 14 torch/csrc/utils/disallow_copy.h 3 torch/csrc/utils/init.cpp 51 torch/csrc/utils/init.h 10 torch/csrc/utils/invalid_arguments.cpp 347 torch/csrc/utils/invalid_arguments.h 11 torch/csrc/utils/memory.h 27 torch/csrc/utils/numpy_stub.h 14 torch/csrc/utils/object_ptr.cpp 8 torch/csrc/utils/object_ptr.h 22 torch/csrc/utils/out_types.cpp 34 torch/csrc/utils/out_types.h 10 torch/csrc/utils/pybind.h 133 torch/csrc/utils/pycfunction_helpers.h 5 torch/csrc/utils/python_arg_parser.cpp 926 torch/csrc/utils/python_arg_parser.h 576 torch/csrc/utils/python_compat.h 59 torch/csrc/utils/python_dispatch.cpp 158 torch/csrc/utils/python_dispatch.h 6 torch/csrc/utils/python_numbers.h 159 torch/csrc/utils/python_scalars.h 53 torch/csrc/utils/python_strings.h 77 torch/csrc/utils/python_stub.h 3 torch/csrc/utils/python_tuples.h 20 torch/csrc/utils/six.h 31 torch/csrc/utils/structseq.cpp 44 torch/csrc/utils/structseq.h 5 torch/csrc/utils/tensor_apply.cpp 95 torch/csrc/utils/tensor_apply.h 9 torch/csrc/utils/tensor_dtypes.cpp 81 torch/csrc/utils/tensor_dtypes.h 8 torch/csrc/utils/tensor_flatten.cpp 97 torch/csrc/utils/tensor_flatten.h 41 torch/csrc/utils/tensor_layouts.cpp 39 torch/csrc/utils/tensor_layouts.h 4 torch/csrc/utils/tensor_list.cpp 38 torch/csrc/utils/tensor_list.h 8 torch/csrc/utils/tensor_memoryformats.cpp 30 torch/csrc/utils/tensor_memoryformats.h 4 torch/csrc/utils/tensor_new.cpp 966 torch/csrc/utils/tensor_new.h 25 torch/csrc/utils/tensor_numpy.cpp 336 torch/csrc/utils/tensor_numpy.h 14 torch/csrc/utils/tensor_qschemes.cpp 36 torch/csrc/utils/tensor_qschemes.h 6 torch/csrc/utils/tensor_types.cpp 85 torch/csrc/utils/tensor_types.h 11 torch/csrc/utils/throughput_benchmark-inl.h 118 torch/csrc/utils/throughput_benchmark.cpp 116 torch/csrc/utils/throughput_benchmark.h 102 torch/csrc/utils/variadic.cpp 2 torch/csrc/utils/variadic.h 95 torch/cuda/__init__.py 428 torch/cuda/_utils.py 24 torch/cuda/amp/__init__.py 2 torch/cuda/amp/autocast_mode.py 82 torch/cuda/amp/common.py 4 torch/cuda/amp/grad_scaler.py 255 torch/cuda/comm.py 3 torch/cuda/error.py 1 torch/cuda/graphs.py 199 torch/cuda/memory.py 238 torch/cuda/nccl.py 84 torch/cuda/nvtx.py 24 torch/cuda/profiler.py 38 torch/cuda/random.py 87 torch/cuda/sparse.py 1 torch/cuda/streams.py 86 torch/custom_class.h 335 torch/custom_class_detail.h 163 torch/deploy.h 2 torch/distributed/__init__.py 44 torch/distributed/_fsdp/__init__.py 2 torch/distributed/_fsdp/flatten_params_wrapper.py 131 torch/distributed/_fsdp/fully_sharded_data_parallel.py 537 torch/distributed/_fsdp/utils.py 15 torch/distributed/_fsdp/wrap.py 111 torch/distributed/_sharded_optim/__init__.py 22 torch/distributed/_sharded_optim/api.py 37 torch/distributed/_sharded_tensor/__init__.py 227 torch/distributed/_sharded_tensor/api.py 473 torch/distributed/_sharded_tensor/metadata.py 61 torch/distributed/_sharded_tensor/ops/__init__.py 8 torch/distributed/_sharded_tensor/ops/_common.py 204 torch/distributed/_sharded_tensor/ops/binary_cmp.py 47 torch/distributed/_sharded_tensor/ops/embedding.py 165 torch/distributed/_sharded_tensor/ops/embedding_bag.py 482 torch/distributed/_sharded_tensor/ops/init.py 48 torch/distributed/_sharded_tensor/ops/linear.py 111 torch/distributed/_sharded_tensor/shard.py 34 torch/distributed/_sharded_tensor/utils.py 182 torch/distributed/_sharding_spec/__init__.py 8 torch/distributed/_sharding_spec/_internals.py 63 torch/distributed/_sharding_spec/api.py 44 torch/distributed/algorithms/__init__.py 3 torch/distributed/algorithms/_checkpoint/__init__.py 1 torch/distributed/algorithms/_checkpoint/_checkpoint_wrapper.py 43 torch/distributed/algorithms/ddp_comm_hooks/__init__.py 67 torch/distributed/algorithms/ddp_comm_hooks/ddp_zero_hook.py 201 torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.py 7 torch/distributed/algorithms/ddp_comm_hooks/default_hooks.py 74 torch/distributed/algorithms/ddp_comm_hooks/optimizer_overlap_hooks.py 38 torch/distributed/algorithms/ddp_comm_hooks/post_localSGD_hook.py 46 torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.py 399 torch/distributed/algorithms/ddp_comm_hooks/quantization_hooks.py 127 torch/distributed/algorithms/join.py 160 torch/distributed/algorithms/model_averaging/__init__.py 1 torch/distributed/algorithms/model_averaging/averagers.py 44 torch/distributed/algorithms/model_averaging/utils.py 20 torch/distributed/algorithms/quantization/quantization.py 106 torch/distributed/argparse_util.py 18 torch/distributed/autograd/__init__.py 41 torch/distributed/benchmarks/benchmark_ddp_rpc.py 242 torch/distributed/constants.py 2 torch/distributed/distributed_c10d.py 1327 torch/distributed/elastic/__init__.py 1 torch/distributed/elastic/agent/__init__.py 1 torch/distributed/elastic/agent/server/__init__.py 9 torch/distributed/elastic/agent/server/api.py 524 torch/distributed/elastic/agent/server/local_elastic_agent.py 127 torch/distributed/elastic/events/__init__.py 72 torch/distributed/elastic/events/api.py 55 torch/distributed/elastic/events/handlers.py 10 torch/distributed/elastic/metrics/__init__.py 21 torch/distributed/elastic/metrics/api.py 116 torch/distributed/elastic/multiprocessing/__init__.py 91 torch/distributed/elastic/multiprocessing/api.py 469 torch/distributed/elastic/multiprocessing/errors/__init__.py 162 torch/distributed/elastic/multiprocessing/errors/error_handler.py 86 torch/distributed/elastic/multiprocessing/errors/handlers.py 3 torch/distributed/elastic/multiprocessing/redirects.py 42 torch/distributed/elastic/multiprocessing/tail_log.py 80 torch/distributed/elastic/rendezvous/__init__.py 15 torch/distributed/elastic/rendezvous/api.py 116 torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.py 138 torch/distributed/elastic/rendezvous/dynamic_rendezvous.py 719 torch/distributed/elastic/rendezvous/etcd_rendezvous.py 566 torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.py 127 torch/distributed/elastic/rendezvous/etcd_server.py 141 torch/distributed/elastic/rendezvous/etcd_store.py 103 torch/distributed/elastic/rendezvous/registry.py 24 torch/distributed/elastic/rendezvous/static_tcp_rendezvous.py 78 torch/distributed/elastic/rendezvous/utils.py 146 torch/distributed/elastic/timer/__init__.py 2 torch/distributed/elastic/timer/api.py 135 torch/distributed/elastic/timer/local_timer.py 76 torch/distributed/elastic/utils/__init__.py 1 torch/distributed/elastic/utils/api.py 37 torch/distributed/elastic/utils/data/__init__.py 2 torch/distributed/elastic/utils/data/cycling_iterator.py 18 torch/distributed/elastic/utils/data/elastic_distributed_sampler.py 32 torch/distributed/elastic/utils/distributed.py 95 torch/distributed/elastic/utils/log_level.py 2 torch/distributed/elastic/utils/logging.py 30 torch/distributed/elastic/utils/store.py 26 torch/distributed/launch.py 39 torch/distributed/launcher/__init__.py 5 torch/distributed/launcher/api.py 150 torch/distributed/nn/__init__.py 2 torch/distributed/nn/api/__init__.py 1 torch/distributed/nn/api/remote_module.py 343 torch/distributed/nn/functional.py 179 torch/distributed/nn/jit/__init__.py 1 torch/distributed/nn/jit/instantiator.py 122 torch/distributed/nn/jit/templates/__init__.py 1 torch/distributed/nn/jit/templates/remote_module_template.py 9 torch/distributed/optim/__init__.py 24 torch/distributed/optim/functional_adadelta.py 63 torch/distributed/optim/functional_adagrad.py 67 torch/distributed/optim/functional_adam.py 127 torch/distributed/optim/functional_adamax.py 75 torch/distributed/optim/functional_adamw.py 126 torch/distributed/optim/functional_rmsprop.py 79 torch/distributed/optim/functional_rprop.py 61 torch/distributed/optim/functional_sgd.py 104 torch/distributed/optim/optimizer.py 125 torch/distributed/optim/post_localSGD_optimizer.py 33 torch/distributed/optim/zero_redundancy_optimizer.py 745 torch/distributed/optim/zero_redundancy_optimizer.pyi 69 torch/distributed/pipeline/__init__.py 1 torch/distributed/pipeline/sync/__init__.py 4 torch/distributed/pipeline/sync/_balance/__init__.py 34 torch/distributed/pipeline/sync/_balance/blockpartition.py 36 torch/distributed/pipeline/sync/_balance/profile.py 65 torch/distributed/pipeline/sync/batchnorm.py 101 torch/distributed/pipeline/sync/checkpoint.py 203 torch/distributed/pipeline/sync/copy.py 55 torch/distributed/pipeline/sync/dependency.py 30 torch/distributed/pipeline/sync/microbatch.py 139 torch/distributed/pipeline/sync/phony.py 15 torch/distributed/pipeline/sync/pipe.py 193 torch/distributed/pipeline/sync/pipeline.py 137 torch/distributed/pipeline/sync/skip/__init__.py 3 torch/distributed/pipeline/sync/skip/layout.py 41 torch/distributed/pipeline/sync/skip/namespace.py 23 torch/distributed/pipeline/sync/skip/portal.py 93 torch/distributed/pipeline/sync/skip/skippable.py 179 torch/distributed/pipeline/sync/skip/tracker.py 82 torch/distributed/pipeline/sync/stream.py 52 torch/distributed/pipeline/sync/utils.py 18 torch/distributed/pipeline/sync/worker.py 78 torch/distributed/remote_device.py 78 torch/distributed/rendezvous.py 145 torch/distributed/rpc/__init__.py 164 torch/distributed/rpc/_testing/__init__.py 11 torch/distributed/rpc/_testing/faulty_agent_backend_registry.py 52 torch/distributed/rpc/api.py 379 torch/distributed/rpc/backend_registry.py 219 torch/distributed/rpc/constants.py 13 torch/distributed/rpc/functions.py 8 torch/distributed/rpc/internal.py 142 torch/distributed/rpc/options.py 74 torch/distributed/rpc/rref_proxy.py 53 torch/distributed/rpc/server_process_global_profiler.py 57 torch/distributed/run.py 314 torch/distributions/__init__.py 93 torch/distributions/bernoulli.py 79 torch/distributions/beta.py 62 torch/distributions/binomial.py 86 torch/distributions/categorical.py 84 torch/distributions/cauchy.py 49 torch/distributions/chi2.py 13 torch/distributions/constraint_registry.py 111 torch/distributions/constraints.py 328 torch/distributions/continuous_bernoulli.py 147 torch/distributions/dirichlet.py 66 torch/distributions/distribution.py 133 torch/distributions/exp_family.py 21 torch/distributions/exponential.py 54 torch/distributions/fishersnedecor.py 60 torch/distributions/gamma.py 54 torch/distributions/geometric.py 76 torch/distributions/gumbel.py 44 torch/distributions/half_cauchy.py 44 torch/distributions/half_normal.py 41 torch/distributions/independent.py 65 torch/distributions/kl.py 565 torch/distributions/kumaraswamy.py 41 torch/distributions/laplace.py 55 torch/distributions/lkj_cholesky.py 58 torch/distributions/log_normal.py 29 torch/distributions/logistic_normal.py 25 torch/distributions/lowrank_multivariate_normal.py 131 torch/distributions/mixture_same_family.py 122 torch/distributions/multinomial.py 77 torch/distributions/multivariate_normal.py 140 torch/distributions/negative_binomial.py 70 torch/distributions/normal.py 64 torch/distributions/one_hot_categorical.py 69 torch/distributions/pareto.py 31 torch/distributions/poisson.py 43 torch/distributions/relaxed_bernoulli.py 86 torch/distributions/relaxed_categorical.py 77 torch/distributions/studentT.py 60 torch/distributions/transformed_distribution.py 113 torch/distributions/transforms.py 708 torch/distributions/uniform.py 59 torch/distributions/utils.py 85 torch/distributions/von_mises.py 84 torch/distributions/weibull.py 43 torch/distributions/wishart.py 199 torch/extension.h 3 torch/fft/__init__.py 53 torch/for_onnx/__init__.py 1 torch/functional.py 494 torch/futures/__init__.py 44 torch/fx/__init__.py 69 torch/fx/__init__.pyi 7 torch/fx/_compatibility.py 22 torch/fx/_pytree.py 34 torch/fx/_symbolic_trace.py 388 torch/fx/annotate.py 18 torch/fx/experimental/__init__.py 1 torch/fx/experimental/accelerator_partitioner.py 717 torch/fx/experimental/const_fold.py 146 torch/fx/experimental/debug.py 16 torch/fx/experimental/fx2trt/__init__.py 11 torch/fx/experimental/fx2trt/converter_registry.py 24 torch/fx/experimental/fx2trt/converters/__init__.py 15 torch/fx/experimental/fx2trt/converters/acc_ops_converters.py 1769 torch/fx/experimental/fx2trt/converters/activation.py 36 torch/fx/experimental/fx2trt/converters/adaptive_avgpool.py 21 torch/fx/experimental/fx2trt/converters/add.py 43 torch/fx/experimental/fx2trt/converters/batchnorm.py 34 torch/fx/experimental/fx2trt/converters/converter_utils.py 300 torch/fx/experimental/fx2trt/converters/convolution.py 65 torch/fx/experimental/fx2trt/converters/linear.py 43 torch/fx/experimental/fx2trt/converters/maxpool.py 26 torch/fx/experimental/fx2trt/converters/mul.py 28 torch/fx/experimental/fx2trt/converters/quantization.py 38 torch/fx/experimental/fx2trt/converters/transformation.py 32 torch/fx/experimental/fx2trt/example/fx2trt_example.py 43 torch/fx/experimental/fx2trt/example/lower_example.py 96 torch/fx/experimental/fx2trt/fx2trt.py 241 torch/fx/experimental/fx2trt/input_tensor_spec.py 46 torch/fx/experimental/fx2trt/lower.py 205 torch/fx/experimental/fx2trt/passes/__init__.py 1 torch/fx/experimental/fx2trt/passes/fuse_pass.py 158 torch/fx/experimental/fx2trt/passes/remove_duplicate_output_args.py 82 torch/fx/experimental/fx2trt/split.py 113 torch/fx/experimental/fx2trt/tools/__init__.py 1 torch/fx/experimental/fx2trt/tools/engine_layer_visualize.py 165 torch/fx/experimental/fx2trt/tools/graph_util.py 39 torch/fx/experimental/fx2trt/tools/timing_cache_utils.py 28 torch/fx/experimental/fx2trt/tools/trt_minimizer.py 55 torch/fx/experimental/fx2trt/tools/trt_splitter.py 70 torch/fx/experimental/fx2trt/trt_module.py 172 torch/fx/experimental/fx2trt/types.py 20 torch/fx/experimental/fx2trt/utils.py 36 torch/fx/experimental/fx_acc/__init__.py 1 torch/fx/experimental/fx_acc/acc_normalizer.py 271 torch/fx/experimental/fx_acc/acc_op_properties.py 21 torch/fx/experimental/fx_acc/acc_ops.py 1518 torch/fx/experimental/fx_acc/acc_tracer.py 223 torch/fx/experimental/fx_acc/acc_utils.py 130 torch/fx/experimental/graph_gradual_typechecker.py 549 torch/fx/experimental/merge_matmul.py 87 torch/fx/experimental/normalize.py 109 torch/fx/experimental/optimization.py 286 torch/fx/experimental/partitioner_utils.py 220 torch/fx/experimental/refinement_types.py 13 torch/fx/experimental/rewriter.py 70 torch/fx/experimental/schema_type_annotation.py 63 torch/fx/experimental/unification/__init__.py 3 torch/fx/experimental/unification/core.py 54 torch/fx/experimental/unification/dispatch.py 4 torch/fx/experimental/unification/match.py 75 torch/fx/experimental/unification/more.py 43 torch/fx/experimental/unification/multipledispatch/__init__.py 3 torch/fx/experimental/unification/multipledispatch/conflict.py 79 torch/fx/experimental/unification/multipledispatch/core.py 31 torch/fx/experimental/unification/multipledispatch/dispatcher.py 241 torch/fx/experimental/unification/multipledispatch/utils.py 54 torch/fx/experimental/unification/multipledispatch/variadic.py 29 torch/fx/experimental/unification/unification_tools.py 134 torch/fx/experimental/unification/utils.py 52 torch/fx/experimental/unification/variable.py 43 torch/fx/experimental/unify_refinements.py 72 torch/fx/graph.py 612 torch/fx/graph_module.py 357 torch/fx/immutable_collections.py 17 torch/fx/interpreter.py 136 torch/fx/node.py 267 torch/fx/operator_schemas.py 244 torch/fx/passes/__init__.py 10 torch/fx/passes/graph_drawer.py 218 torch/fx/passes/graph_manipulation.py 299 torch/fx/passes/net_min_base.py 318 torch/fx/passes/operator_support.py 98 torch/fx/passes/param_fetch.py 46 torch/fx/passes/shape_prop.py 62 torch/fx/passes/split_module.py 140 torch/fx/passes/split_utils.py 133 torch/fx/passes/splitter_base.py 488 torch/fx/passes/tools_common.py 122 torch/fx/proxy.py 242 torch/fx/subgraph_rewriter.py 180 torch/fx/tensor_type.py 50 torch/hub.py 304 torch/jit/__init__.py 73 torch/jit/_async.py 12 torch/jit/_builtins.py 133 torch/jit/_check.py 101 torch/jit/_freeze.py 43 torch/jit/_fuser.py 75 torch/jit/_logging.py 7 torch/jit/_monkeytype_config.py 109 torch/jit/_passes/__init__.py 1 torch/jit/_passes/_property_propagation.py 21 torch/jit/_pickle.py 13 torch/jit/_recursive.py 551 torch/jit/_script.py 791 torch/jit/_serialization.py 44 torch/jit/_state.py 67 torch/jit/_trace.py 705 torch/jit/annotations.py 312 torch/jit/frontend.py 775 torch/jit/generate_bytecode.py 23 torch/jit/mobile/__init__.py 88 torch/jit/quantized.py 432 torch/jit/supported_ops.py 241 torch/jit/unsupported_tensor_ops.py 40 torch/lib/libshm/alloc_info.h 7 torch/lib/libshm/core.cpp 137 torch/lib/libshm/err.h 15 torch/lib/libshm/libshm.h 32 torch/lib/libshm/manager.cpp 155 torch/lib/libshm/socket.h 148 torch/lib/libshm_windows/core.cpp 21 torch/lib/libshm_windows/libshm.h 28 torch/library.h 469 torch/linalg/__init__.py 104 torch/monitor/__init__.py 1 torch/multiprocessing/__init__.py 26 torch/multiprocessing/_atfork.py 14 torch/multiprocessing/pool.py 26 torch/multiprocessing/queue.py 32 torch/multiprocessing/reductions.py 195 torch/multiprocessing/spawn.py 137 torch/nn/__init__.py 20 torch/nn/_reduction.py 33 torch/nn/backends/__init__.py 1 torch/nn/backends/thnn.py 2 torch/nn/common_types.py 25 torch/nn/cpp.py 48 torch/nn/functional.py 2645 torch/nn/functional.pyi.in 203 torch/nn/grad.py 135 torch/nn/init.py 217 torch/nn/intrinsic/__init__.py 1 torch/nn/intrinsic/modules/__init__.py 28 torch/nn/intrinsic/modules/fused.py 88 torch/nn/intrinsic/qat/__init__.py 1 torch/nn/intrinsic/qat/modules/__init__.py 26 torch/nn/intrinsic/qat/modules/conv_fused.py 427 torch/nn/intrinsic/qat/modules/linear_relu.py 22 torch/nn/intrinsic/quantized/__init__.py 1 torch/nn/intrinsic/quantized/dynamic/__init__.py 1 torch/nn/intrinsic/quantized/dynamic/modules/__init__.py 5 torch/nn/intrinsic/quantized/dynamic/modules/linear_relu.py 23 torch/nn/intrinsic/quantized/modules/__init__.py 11 torch/nn/intrinsic/quantized/modules/bn_relu.py 36 torch/nn/intrinsic/quantized/modules/conv_relu.py 98 torch/nn/intrinsic/quantized/modules/linear_relu.py 16 torch/nn/modules/__init__.py 65 torch/nn/modules/_functions.py 163 torch/nn/modules/activation.py 418 torch/nn/modules/adaptive.py 148 torch/nn/modules/batchnorm.py 332 torch/nn/modules/channelshuffle.py 14 torch/nn/modules/container.py 392 torch/nn/modules/conv.py 658 torch/nn/modules/distance.py 27 torch/nn/modules/dropout.py 36 torch/nn/modules/flatten.py 56 torch/nn/modules/fold.py 60 torch/nn/modules/instancenorm.py 109 torch/nn/modules/lazy.py 89 torch/nn/modules/linear.py 104 torch/nn/modules/loss.py 245 torch/nn/modules/module.py 798 torch/nn/modules/normalization.py 108 torch/nn/modules/padding.py 93 torch/nn/modules/pixelshuffle.py 25 torch/nn/modules/pooling.py 305 torch/nn/modules/rnn.py 580 torch/nn/modules/sparse.py 179 torch/nn/modules/transformer.py 220 torch/nn/modules/upsampling.py 45 torch/nn/modules/utils.py 43 torch/nn/parallel/__init__.py 12 torch/nn/parallel/__init__.pyi 5 torch/nn/parallel/_functions.py 98 torch/nn/parallel/comm.py 100 torch/nn/parallel/common_types.pyi 4 torch/nn/parallel/data_parallel.py 105 torch/nn/parallel/data_parallel.pyi 15 torch/nn/parallel/distributed.py 791 torch/nn/parallel/distributed.pyi 17 torch/nn/parallel/parallel_apply.py 65 torch/nn/parallel/parallel_apply.pyi 5 torch/nn/parallel/replicate.py 117 torch/nn/parallel/replicate.pyi 5 torch/nn/parallel/scatter_gather.py 57 torch/nn/parallel/scatter_gather.pyi 10 torch/nn/parameter.py 98 torch/nn/parameter.pyi 16 torch/nn/qat/__init__.py 1 torch/nn/qat/dynamic/__init__.py 1 torch/nn/qat/dynamic/modules/__init__.py 2 torch/nn/qat/dynamic/modules/linear.py 9 torch/nn/qat/modules/__init__.py 11 torch/nn/qat/modules/conv.py 132 torch/nn/qat/modules/embedding_ops.py 92 torch/nn/qat/modules/linear.py 37 torch/nn/quantizable/__init__.py 1 torch/nn/quantizable/modules/__init__.py 8 torch/nn/quantizable/modules/activation.py 297 torch/nn/quantizable/modules/rnn.py 277 torch/nn/quantized/__init__.py 1 torch/nn/quantized/_reference/__init__.py 1 torch/nn/quantized/_reference/modules/__init__.py 8 torch/nn/quantized/_reference/modules/conv.py 155 torch/nn/quantized/_reference/modules/linear.py 89 torch/nn/quantized/_reference/modules/utils.py 54 torch/nn/quantized/dynamic/__init__.py 1 torch/nn/quantized/dynamic/modules/__init__.py 17 torch/nn/quantized/dynamic/modules/conv.py 183 torch/nn/quantized/dynamic/modules/linear.py 68 torch/nn/quantized/dynamic/modules/rnn.py 613 torch/nn/quantized/functional.py 196 torch/nn/quantized/modules/__init__.py 74 torch/nn/quantized/modules/activation.py 73 torch/nn/quantized/modules/batchnorm.py 57 torch/nn/quantized/modules/conv.py 505 torch/nn/quantized/modules/dropout.py 11 torch/nn/quantized/modules/embedding_ops.py 162 torch/nn/quantized/modules/functional_modules.py 134 torch/nn/quantized/modules/linear.py 176 torch/nn/quantized/modules/normalization.py 127 torch/nn/quantized/modules/utils.py 58 torch/nn/utils/__init__.py 9 torch/nn/utils/_stateless.py 33 torch/nn/utils/clip_grad.py 47 torch/nn/utils/convert_parameters.py 36 torch/nn/utils/fusion.py 38 torch/nn/utils/init.py 11 torch/nn/utils/memory_format.py 9 torch/nn/utils/parametrizations.py 194 torch/nn/utils/parametrize.py 297 torch/nn/utils/prune.py 461 torch/nn/utils/rnn.py 156 torch/nn/utils/rnn.pyi 40 torch/nn/utils/spectral_norm.py 156 torch/nn/utils/weight_norm.py 59 torch/onnx/__init__.py 264 torch/onnx/operators.py 8 torch/onnx/symbolic_caffe2.py 228 torch/onnx/symbolic_helper.py 686 torch/onnx/symbolic_opset10.py 242 torch/onnx/symbolic_opset11.py 612 torch/onnx/symbolic_opset12.py 200 torch/onnx/symbolic_opset13.py 272 torch/onnx/symbolic_opset14.py 33 torch/onnx/symbolic_opset15.py 1 torch/onnx/symbolic_opset7.py 22 torch/onnx/symbolic_opset8.py 184 torch/onnx/symbolic_opset9.py 2371 torch/onnx/symbolic_registry.py 87 torch/onnx/utils.py 915 torch/optim/__init__.py 30 torch/optim/__init__.pyi 16 torch/optim/_functional.py 371 torch/optim/_multi_tensor/__init__.py 22 torch/optim/_multi_tensor/__init__.pyi 11 torch/optim/_multi_tensor/_functional.py 194 torch/optim/_multi_tensor/adadelta.py 52 torch/optim/_multi_tensor/adadelta.pyi 4 torch/optim/_multi_tensor/adagrad.py 62 torch/optim/_multi_tensor/adagrad.pyi 3 torch/optim/_multi_tensor/adam.py 85 torch/optim/_multi_tensor/adam.pyi 4 torch/optim/_multi_tensor/adamax.py 57 torch/optim/_multi_tensor/adamax.pyi 4 torch/optim/_multi_tensor/adamw.py 83 torch/optim/_multi_tensor/adamw.pyi 4 torch/optim/_multi_tensor/asgd.py 45 torch/optim/_multi_tensor/asgd.pyi 4 torch/optim/_multi_tensor/nadam.py 69 torch/optim/_multi_tensor/nadam.pyi 4 torch/optim/_multi_tensor/radam.py 59 torch/optim/_multi_tensor/radam.pyi 4 torch/optim/_multi_tensor/rmsprop.py 74 torch/optim/_multi_tensor/rmsprop.pyi 4 torch/optim/_multi_tensor/rprop.py 54 torch/optim/_multi_tensor/rprop.pyi 4 torch/optim/_multi_tensor/sgd.py 83 torch/optim/_multi_tensor/sgd.pyi 3 torch/optim/adadelta.py 52 torch/optim/adadelta.pyi 4 torch/optim/adagrad.py 59 torch/optim/adagrad.pyi 4 torch/optim/adam.py 72 torch/optim/adam.pyi 4 torch/optim/adamax.py 62 torch/optim/adamax.pyi 4 torch/optim/adamw.py 75 torch/optim/adamw.pyi 4 torch/optim/asgd.py 56 torch/optim/asgd.pyi 4 torch/optim/lbfgs.py 303 torch/optim/lbfgs.pyi 4 torch/optim/lr_scheduler.py 775 torch/optim/lr_scheduler.pyi 37 torch/optim/nadam.py 70 torch/optim/nadam.pyi 4 torch/optim/optimizer.py 186 torch/optim/optimizer.pyi 14 torch/optim/radam.py 59 torch/optim/radam.pyi 4 torch/optim/rmsprop.py 68 torch/optim/rmsprop.pyi 4 torch/optim/rprop.py 50 torch/optim/rprop.pyi 4 torch/optim/sgd.py 61 torch/optim/sgd.pyi 3 torch/optim/sparse_adam.py 69 torch/optim/sparse_adam.pyi 4 torch/optim/swa_utils.py 115 torch/optim/swa_utils.pyi 13 torch/overrides.py 1262 torch/package/__init__.py 12 torch/package/_digraph.py 86 torch/package/_directory_reader.py 33 torch/package/_importlib.py 60 torch/package/_mangling.py 26 torch/package/_mock.py 104 torch/package/_package_pickler.py 75 torch/package/_package_unpickler.py 15 torch/package/_stdlib.py 880 torch/package/analyze/__init__.py 3 torch/package/analyze/is_from_package.py 8 torch/package/analyze/trace_dependencies.py 27 torch/package/file_structure_representation.py 71 torch/package/find_file_dependencies.py 75 torch/package/glob_group.py 40 torch/package/importer.py 114 torch/package/package_exporter.py 631 torch/package/package_importer.py 422 torch/profiler/__init__.py 7 torch/profiler/profiler.py 248 torch/profiler/python_tracer.py 16 torch/quantization/__init__.py 44 torch/quantization/_numeric_suite.py 20 torch/quantization/_numeric_suite_fx.py 18 torch/quantization/fake_quantize.py 24 torch/quantization/fuse_modules.py 11 torch/quantization/fuser_method_mappings.py 8 torch/quantization/fx/__init__.py 4 torch/quantization/fx/_equalize.py 29 torch/quantization/fx/convert.py 4 torch/quantization/fx/fuse.py 4 torch/quantization/fx/fusion_patterns.py 5 torch/quantization/fx/graph_module.py 10 torch/quantization/fx/match_utils.py 7 torch/quantization/fx/pattern_utils.py 10 torch/quantization/fx/prepare.py 4 torch/quantization/fx/quantization_patterns.py 17 torch/quantization/fx/quantization_types.py 5 torch/quantization/fx/utils.py 20 torch/quantization/observer.py 29 torch/quantization/qconfig.py 23 torch/quantization/quant_type.py 3 torch/quantization/quantization_mappings.py 22 torch/quantization/quantize.py 20 torch/quantization/quantize_fx.py 20 torch/quantization/quantize_jit.py 17 torch/quantization/stubs.py 6 torch/quantization/utils.py 19 torch/quasirandom.py 87 torch/random.py 66 torch/return_types.py 8 torch/script.h 11 torch/serialization.py 647 torch/sparse/__init__.py 44 torch/special/__init__.py 128 torch/storage.py 476 torch/testing/__init__.py 4 torch/testing/_comparison.py 658 torch/testing/_core.py 118 torch/testing/_creation.py 80 torch/testing/_deprecated.py 69 torch/testing/_internal/__init__.py 1 torch/testing/_internal/autocast_test_lists.py 300 torch/testing/_internal/check_kernel_launches.py 83 torch/testing/_internal/codegen/__init__.py 1 torch/testing/_internal/common_cuda.py 108 torch/testing/_internal/common_device_type.py 587 torch/testing/_internal/common_distributed.py 554 torch/testing/_internal/common_dtype.py 24 torch/testing/_internal/common_fsdp.py 396 torch/testing/_internal/common_fx2trt.py 191 torch/testing/_internal/common_jit.py 247 torch/testing/_internal/common_methods_invocations.py 12259 torch/testing/_internal/common_modules.py 964 torch/testing/_internal/common_nn.py 5816 torch/testing/_internal/common_quantization.py 1664 torch/testing/_internal/common_quantized.py 155 torch/testing/_internal/common_utils.py 2066 torch/testing/_internal/composite_compliance.py 147 torch/testing/_internal/data/__init__.py 1 torch/testing/_internal/data/network1.py 5 torch/testing/_internal/data/network2.py 6 torch/testing/_internal/dist_utils.py 117 torch/testing/_internal/distributed/__init__.py 1 torch/testing/_internal/distributed/_sharded_tensor/__init__.py 75 torch/testing/_internal/distributed/_sharded_tensor/_test_ops_common.py 51 torch/testing/_internal/distributed/nn/__init__.py 1 torch/testing/_internal/distributed/nn/api/__init__.py 1 torch/testing/_internal/distributed/pipeline/__init__.py 1 torch/testing/_internal/distributed/rpc/__init__.py 1 torch/testing/_internal/distributed/rpc/faulty_rpc_agent_test_fixture.py 47 torch/testing/_internal/distributed/rpc/jit/__init__.py 1 torch/testing/_internal/distributed/rpc/jit/rpc_test_faulty.py 157 torch/testing/_internal/distributed/rpc/rpc_agent_test_fixture.py 37 torch/testing/_internal/distributed/rpc/tensorpipe_rpc_agent_test_fixture.py 21 torch/testing/_internal/distributed/rpc_utils.py 126 torch/testing/_internal/hypothesis_utils.py 210 torch/testing/_internal/jit_metaprogramming_utils.py 495 torch/testing/_internal/jit_utils.py 563 torch/testing/_internal/logging_tensor.py 74 torch/testing/_internal/opinfo_helper.py 93 torch/testing/_internal/quantization_torch_package_models.py 25 torch/testing/_legacy.py 106 torch/torch_version.py 34 torch/types.py 42 torch/utils/__init__.py 12 torch/utils/_cpp_extension_versioner.py 42 torch/utils/_crash_handler.py 17 torch/utils/_freeze.py 155 torch/utils/_python_dispatch.py 16 torch/utils/_pytree.py 126 torch/utils/_zip.py 44 torch/utils/backcompat/__init__.py 15 torch/utils/benchmark/__init__.py 6 torch/utils/benchmark/op_fuzzers/__init__.py 1 torch/utils/benchmark/op_fuzzers/binary.py 84 torch/utils/benchmark/op_fuzzers/sparse_binary.py 89 torch/utils/benchmark/op_fuzzers/sparse_unary.py 69 torch/utils/benchmark/op_fuzzers/spectral.py 71 torch/utils/benchmark/op_fuzzers/unary.py 62 torch/utils/benchmark/utils/__init__.py 1 torch/utils/benchmark/utils/_stubs.py 29 torch/utils/benchmark/utils/common.py 212 torch/utils/benchmark/utils/compare.py 266 torch/utils/benchmark/utils/cpp_jit.py 98 torch/utils/benchmark/utils/fuzzer.py 253 torch/utils/benchmark/utils/sparse_fuzzer.py 71 torch/utils/benchmark/utils/timeit_template.cpp 18 torch/utils/benchmark/utils/timer.py 249 torch/utils/benchmark/utils/valgrind_wrapper/__init__.py 1 torch/utils/benchmark/utils/valgrind_wrapper/compat_bindings.cpp 29 torch/utils/benchmark/utils/valgrind_wrapper/timer_callgrind_template.cpp 31 torch/utils/benchmark/utils/valgrind_wrapper/timer_interface.py 498 torch/utils/bottleneck/__init__.py 1 torch/utils/bottleneck/__main__.py 147 torch/utils/bundled_inputs.py 221 torch/utils/checkpoint.py 172 torch/utils/collect_env.py 329 torch/utils/cpp_extension.py 1589 torch/utils/data/__init__.py 71 torch/utils/data/_decorator.py 122 torch/utils/data/_typing.py 288 torch/utils/data/_utils/__init__.py 19 torch/utils/data/_utils/collate.py 156 torch/utils/data/_utils/fetch.py 3 torch/utils/data/_utils/pin_memory.py 50 torch/utils/data/_utils/signal_handling.py 54 torch/utils/data/_utils/worker.py 87 torch/utils/data/backward_compatibility.py 6 torch/utils/data/communication/__init__.py 6 torch/utils/data/communication/eventloop.py 34 torch/utils/data/communication/iter.py 132 torch/utils/data/communication/map.py 117 torch/utils/data/communication/messages.py 39 torch/utils/data/communication/protocol.py 157 torch/utils/data/communication/queue.py 39 torch/utils/data/dataloader.py 515 torch/utils/data/dataloader_experimental.py 143 torch/utils/data/datapipes/__init__.py 3 torch/utils/data/datapipes/dataframe/__init__.py 8 torch/utils/data/datapipes/dataframe/dataframe_wrapper.py 79 torch/utils/data/datapipes/dataframe/dataframes.py 209 torch/utils/data/datapipes/dataframe/datapipes.py 102 torch/utils/data/datapipes/dataframe/structures.py 13 torch/utils/data/datapipes/iter/__init__.py 61 torch/utils/data/datapipes/iter/callable.py 102 torch/utils/data/datapipes/iter/combinatorics.py 74 torch/utils/data/datapipes/iter/combining.py 286 torch/utils/data/datapipes/iter/filelister.py 25 torch/utils/data/datapipes/iter/fileopener.py 31 torch/utils/data/datapipes/iter/grouping.py 188 torch/utils/data/datapipes/iter/routeddecoder.py 34 torch/utils/data/datapipes/iter/selecting.py 71 torch/utils/data/datapipes/iter/streamreader.py 14 torch/utils/data/datapipes/iter/utils.py 22 torch/utils/data/datapipes/map/__init__.py 12 torch/utils/data/datapipes/map/callable.py 47 torch/utils/data/datapipes/map/combinatorics.py 27 torch/utils/data/datapipes/map/combining.py 55 torch/utils/data/datapipes/map/grouping.py 45 torch/utils/data/datapipes/map/utils.py 21 torch/utils/data/datapipes/utils/__init__.py 1 torch/utils/data/datapipes/utils/common.py 89 torch/utils/data/datapipes/utils/decoder.py 189 torch/utils/data/dataset.py 204 torch/utils/data/dataset.pyi 88 torch/utils/data/dataset.pyi.in 75 torch/utils/data/distributed.py 61 torch/utils/data/gen_pyi.py 141 torch/utils/data/graph.py 47 torch/utils/data/graph_settings.py 29 torch/utils/data/sampler.py 120 torch/utils/dlpack.py 30 torch/utils/ffi/__init__.py 1 torch/utils/file_baton.py 39 torch/utils/hipify/__init__.py 1 torch/utils/hipify/constants.py 50 torch/utils/hipify/cuda_to_hip_mappings.py 8181 torch/utils/hipify/hipify_python.py 638 torch/utils/hipify/version.py 1 torch/utils/hooks.py 135 torch/utils/mkldnn.py 164 torch/utils/mobile_optimizer.py 84 torch/utils/model_dump/__init__.py 252 torch/utils/model_dump/__main__.py 3 torch/utils/model_dump/code.js 612 torch/utils/model_dump/skeleton.html 20 torch/utils/model_zoo.py 1 torch/utils/show_pickle.py 116 torch/utils/tensorboard/__init__.py 8 torch/utils/tensorboard/_caffe2_graph.py 642 torch/utils/tensorboard/_convert_np.py 20 torch/utils/tensorboard/_embedding.py 51 torch/utils/tensorboard/_onnx_graph.py 45 torch/utils/tensorboard/_proto_graph.py 31 torch/utils/tensorboard/_pytorch_graph.py 198 torch/utils/tensorboard/_utils.py 78 torch/utils/tensorboard/summary.py 487 torch/utils/tensorboard/writer.py 347 torch/utils/throughput_benchmark.py 121