Path Lines of Code android/pytorch_android/src/main/cpp/pytorch_jni_lite.cpp 192 aten/src/ATen/CPUGeneratorImpl.cpp 170 aten/src/ATen/Context.cpp 246 aten/src/ATen/Dispatch.h 669 aten/src/ATen/FunctionalTensorWrapper.cpp 227 aten/src/ATen/ParallelNative.cpp 244 aten/src/ATen/ParallelNativeTBB.cpp 91 aten/src/ATen/ScalarOps.h 49 aten/src/ATen/SparseTensorUtils.cpp 58 aten/src/ATen/SparseTensorUtils.h 35 aten/src/ATen/TensorIndexing.cpp 78 aten/src/ATen/TensorIndexing.h 411 aten/src/ATen/TensorIterator.cpp 1255 aten/src/ATen/TensorUtils.h 123 aten/src/ATen/TracerMode.h 23 aten/src/ATen/Utils.h 104 aten/src/ATen/Version.cpp 177 aten/src/ATen/ZeroTensorFallback.cpp 80 aten/src/ATen/autocast_mode.cpp 590 aten/src/ATen/core/DeprecatedTypePropertiesRegistry.cpp 24 aten/src/ATen/core/Generator.h 82 aten/src/ATen/core/LegacyTypeDispatch.h 35 aten/src/ATen/core/List.h 223 aten/src/ATen/core/List_inl.h 279 aten/src/ATen/core/TensorBase.h 499 aten/src/ATen/core/TransformationHelper.h 96 aten/src/ATen/core/VariableFallbackKernel.cpp 38 aten/src/ATen/core/blob.h 122 aten/src/ATen/core/boxing/KernelFunction.h 57 aten/src/ATen/core/boxing/impl/boxing.h 218 aten/src/ATen/core/boxing/impl/make_boxed_from_unboxed_functor.h 351 aten/src/ATen/core/class_type.h 297 aten/src/ATen/core/dispatch/DispatchKeyExtractor.h 134 aten/src/ATen/core/dispatch/Dispatcher.cpp 281 aten/src/ATen/core/dispatch/Dispatcher.h 304 aten/src/ATen/core/dispatch/OperatorEntry.cpp 345 aten/src/ATen/core/dynamic_type.h 158 aten/src/ATen/core/function_schema.h 364 aten/src/ATen/core/interned_strings.cpp 117 aten/src/ATen/core/ivalue.cpp 980 aten/src/ATen/core/ivalue.h 843 aten/src/ATen/core/ivalue_inl.h 1803 aten/src/ATen/core/jit_type.h 1616 aten/src/ATen/core/library.cpp 203 aten/src/ATen/core/op_registration/adaption.h 37 aten/src/ATen/core/op_registration/op_allowlist.h 104 aten/src/ATen/core/op_registration/op_registration.h 257 aten/src/ATen/core/operator_name.h 68 aten/src/ATen/core/stack.h 146 aten/src/ATen/core/symbol.h 64 aten/src/ATen/core/tensor_type.cpp 339 aten/src/ATen/cpp_custom_type_hack.h 40 aten/src/ATen/cpu/vec/functional_base.h 245 aten/src/ATen/cpu/vec/vec256/vec256.h 133 aten/src/ATen/cpu/vec/vec256/vec256_float_neon.h 645 aten/src/ATen/cpu/vec/vec512/vec512_complex_double.h 460 aten/src/ATen/cpu/vec/vec512/vec512_complex_float.h 965 aten/src/ATen/cuda/ApplyGridUtils.cuh 37 aten/src/ATen/cuda/CUDAApplyUtils.cuh 473 aten/src/ATen/cuda/CUDAGeneratorImpl.cpp 176 aten/src/ATen/cuda/PinnedMemoryAllocator.cpp 28 aten/src/ATen/cuda/ScanUtils.cuh 131 aten/src/ATen/cuda/detail/CUDAHooks.cpp 312 aten/src/ATen/cuda/detail/CUDAHooks.h 38 aten/src/ATen/cuda/detail/IntegerDivider.cuh 108 aten/src/ATen/cudnn/Descriptors.h 250 aten/src/ATen/detail/CUDAHooksInterface.h 134 aten/src/ATen/jit_macros.h 12 aten/src/ATen/native/Activation.cpp 800 aten/src/ATen/native/BatchLinearAlgebra.cpp 2819 aten/src/ATen/native/BinaryOps.cpp 908 aten/src/ATen/native/CPUFallback.h 34 aten/src/ATen/native/ChanelShuffle.cpp 57 aten/src/ATen/native/ConvUtils.h 221 aten/src/ATen/native/Convolution.cpp 1499 aten/src/ATen/native/ConvolutionMM3d.cpp 700 aten/src/ATen/native/Copy.cpp 215 aten/src/ATen/native/DispatchStub.h 191 aten/src/ATen/native/Distance.cpp 202 aten/src/ATen/native/DistributionTemplates.h 296 aten/src/ATen/native/Distributions.cpp 446 aten/src/ATen/native/Embedding.cpp 144 aten/src/ATen/native/EmbeddingBag.cpp 1015 aten/src/ATen/native/Integration.cpp 110 aten/src/ATen/native/LinearAlgebra.cpp 2281 aten/src/ATen/native/Loss.cpp 449 aten/src/ATen/native/LossNLL.cpp 578 aten/src/ATen/native/NaiveConvolutionTranspose3d.cpp 828 aten/src/ATen/native/NaiveDilatedConvolution.cpp 466 aten/src/ATen/native/Normalization.cpp 569 aten/src/ATen/native/RNN.cpp 1712 aten/src/ATen/native/ReduceOps.cpp 1528 aten/src/ATen/native/Resize.cpp 93 aten/src/ATen/native/Resize.h 122 aten/src/ATen/native/SegmentReduce.cpp 288 aten/src/ATen/native/SharedReduceOps.h 417 aten/src/ATen/native/SpectralOps.cpp 964 aten/src/ATen/native/TensorAdvancedIndexing.cpp 1443 aten/src/ATen/native/TensorCompare.cpp 557 aten/src/ATen/native/TensorFactories.cpp 1206 aten/src/ATen/native/TensorShape.cpp 2064 aten/src/ATen/native/UnaryOps.cpp 667 aten/src/ATen/native/UpSample.h 318 aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear.cpp 180 aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_unpack.cpp 57 aten/src/ATen/native/ao_sparse/quantized/cpu/qnnpack_utils.h 59 aten/src/ATen/native/attention.cpp 181 aten/src/ATen/native/cpu/AdaptiveMaxPoolKernel.cpp 392 aten/src/ATen/native/cpu/BinaryOpsKernel.cpp 1090 aten/src/ATen/native/cpu/CopyKernel.cpp 84 aten/src/ATen/native/cpu/DepthwiseConvKernel.cpp 263 aten/src/ATen/native/cpu/DistanceOpsKernel.cpp 333 aten/src/ATen/native/cpu/IndexKernel.cpp 510 aten/src/ATen/native/cpu/MaxPoolKernel.cpp 414 aten/src/ATen/native/cpu/ReduceOpsKernel.cpp 361 aten/src/ATen/native/cpu/UnaryOpsKernel.cpp 580 aten/src/ATen/native/cuda/AdaptiveAveragePooling.cu 686 aten/src/ATen/native/cuda/BatchLinearAlgebra.cpp 2525 aten/src/ATen/native/cuda/Blas.cpp 407 aten/src/ATen/native/cuda/CUDALoops.cuh 406 aten/src/ATen/native/cuda/DepthwiseConv2d.cu 558 aten/src/ATen/native/cuda/DistributionTemplates.h 538 aten/src/ATen/native/cuda/Dropout.cu 369 aten/src/ATen/native/cuda/IGammaKernel.cu 496 aten/src/ATen/native/cuda/Loops.cuh 389 aten/src/ATen/native/cuda/Math.cuh 1509 aten/src/ATen/native/cuda/NaiveDilatedConvolution.cu 573 aten/src/ATen/native/cuda/Nonzero.cu 114 aten/src/ATen/native/cuda/Normalization.cu 632 aten/src/ATen/native/cuda/ROCmLoops.cuh 317 aten/src/ATen/native/cuda/ScanKernels.cu 460 aten/src/ATen/native/cuda/SegmentReduce.cu 369 aten/src/ATen/native/cuda/Shape.cu 539 aten/src/ATen/native/cuda/Sort.cpp 141 aten/src/ATen/native/cuda/Sort.cu 309 aten/src/ATen/native/cuda/SpectralOps.cpp 376 aten/src/ATen/native/cuda/SpectralOps.cu 115 aten/src/ATen/native/cuda/UnaryGammaKernels.cu 100 aten/src/ATen/native/cuda/UpSample.cuh 231 aten/src/ATen/native/cuda/UpSampleNearest2d.cu 403 aten/src/ATen/native/cuda/UpSampleNearest3d.cu 319 aten/src/ATen/native/cuda/attention.cu 218 aten/src/ATen/native/cuda/group_norm_kernel.cu 947 aten/src/ATen/native/cuda/im2col.cuh 195 aten/src/ATen/native/cuda/jit_utils.cpp 808 aten/src/ATen/native/cuda/vol2col.cuh 255 aten/src/ATen/native/cudnn/BatchNorm.cpp 294 aten/src/ATen/native/cudnn/ConvShared.cpp 429 aten/src/ATen/native/cudnn/ConvShared.h 74 aten/src/ATen/native/cudnn/Conv_v7.cpp 713 aten/src/ATen/native/cudnn/Conv_v8.cpp 143 aten/src/ATen/native/cudnn/GridSampler.cpp 111 aten/src/ATen/native/cudnn/RNN.cpp 1255 aten/src/ATen/native/metal/ops/MetalChunk.mm 62 aten/src/ATen/native/metal/ops/MetalReduce.mm 76 aten/src/ATen/native/metal/ops/MetalSoftmax.mm 63 aten/src/ATen/native/metal/ops/MetalTranspose.mm 92 aten/src/ATen/native/miopen/BatchNorm_miopen.cpp 180 aten/src/ATen/native/miopen/Conv_miopen.cpp 948 aten/src/ATen/native/mkldnn/MKLDNNCommon.cpp 89 aten/src/ATen/native/mkldnn/MKLDNNConversions.cpp 141 aten/src/ATen/native/mkldnn/Normalization.cpp 167 aten/src/ATen/native/mkldnn/TensorFactories.cpp 18 aten/src/ATen/native/native_functions.yaml 8466 aten/src/ATen/native/quantized/QTensor.cpp 304 aten/src/ATen/native/quantized/TensorCompare.cpp 38 aten/src/ATen/native/quantized/affine_quantizer_base.h 35 aten/src/ATen/native/quantized/cpu/fbgemm_utils.h 325 aten/src/ATen/native/quantized/cpu/kernels/QuantizedOpKernels.cpp 2972 aten/src/ATen/native/quantized/cpu/q_adaavgpool.cpp 275 aten/src/ATen/native/quantized/cpu/qbatch_norm.cpp 350 aten/src/ATen/native/quantized/cpu/qconv.cpp 824 aten/src/ATen/native/quantized/cpu/qconv_dynamic.cpp 132 aten/src/ATen/native/quantized/cpu/qconv_prepack.cpp 414 aten/src/ATen/native/quantized/cpu/qconv_unpack.cpp 265 aten/src/ATen/native/quantized/cpu/qembeddingbag_prepack.cpp 341 aten/src/ATen/native/quantized/cpu/qembeddingbag_unpack.cpp 181 aten/src/ATen/native/quantized/cpu/qlinear.cpp 347 aten/src/ATen/native/quantized/cpu/qlinear_dynamic.cpp 374 aten/src/ATen/native/quantized/cpu/qlinear_prepack.cpp 295 aten/src/ATen/native/quantized/cpu/qmul.cpp 175 aten/src/ATen/native/quantized/cpu/qnnpack/src/convolution.c 738 aten/src/ATen/native/quantized/cpu/qnormalization.cpp 141 aten/src/ATen/native/quantized/cpu/qrelu.cpp 156 aten/src/ATen/native/quantized/cpu/tensor_operators.cpp 66 aten/src/ATen/native/quantized/library.cpp 190 aten/src/ATen/native/sparse/SparseCsrTensor.cpp 310 aten/src/ATen/native/sparse/SparseCsrTensorMath.cpp 486 aten/src/ATen/native/sparse/SparseTensor.cpp 664 aten/src/ATen/native/sparse/SparseTensorMath.cpp 1140 aten/src/ATen/native/sparse/cuda/SparseBlas.cpp 185 aten/src/ATen/native/sparse/cuda/SparseBlasImpl.cpp 1082 aten/src/ATen/native/sparse/cuda/SparseCUDAApplyUtils.cuh 328 aten/src/ATen/native/sparse/cuda/SparseCUDABlas.cpp 430 aten/src/ATen/native/sparse/cuda/SparseCUDATensor.cpp 53 aten/src/ATen/native/sparse/cuda/SparseCUDATensorMath.cu 820 aten/src/ATen/native/utils/Factory.h 16 aten/src/ATen/native/vulkan/VulkanConvolution.cpp 93 aten/src/ATen/native/vulkan/VulkanConvolution.h 115 aten/src/ATen/native/vulkan/api/vk_mem_alloc.h 12820 aten/src/ATen/native/vulkan/ops/Convolution.cpp 731 aten/src/ATen/native/xnnpack/Convolution.cpp 407 aten/src/ATen/native/xnnpack/Linear.cpp 181 aten/src/ATen/nnapi/codegen.py 73 aten/src/ATen/nnapi/nnapi_bind.cpp 153 aten/src/ATen/nnapi/nnapi_model_loader.cpp 197 aten/src/ATen/nnapi/nnapi_wrapper.cpp 373 aten/src/ATen/quantized/QTensorImpl.h 76 aten/src/ATen/templates/Functions.cpp 81 aten/src/ATen/templates/RegisterBackendSelect.cpp 33 aten/src/ATen/templates/TensorBody.h 380 aten/src/THC/THCAtomics.cuh 3 aten/src/THC/THCDeviceUtils.cuh 3 benchmarks/cpp/nvfuser/bert.cpp 557 benchmarks/cpp/nvfuser/lstm_cell.cpp 163 benchmarks/distributed/rpc/parameter_server/launcher.py 438 benchmarks/fastrnns/bench.py 210 benchmarks/fastrnns/factory.py 350 benchmarks/functional_autograd_benchmark/torchvision_models.py 466 benchmarks/instruction_counts/applications/ci.py 59 benchmarks/instruction_counts/definitions/standard.py 78 benchmarks/operator_benchmark/benchmark_core.py 242 benchmarks/operator_benchmark/benchmark_pytorch.py 101 benchmarks/operator_benchmark/benchmark_utils.py 158 benchmarks/operator_benchmark/operator_benchmark.py 4 benchmarks/tensorexpr/__main__.py 288 benchmarks/tensorexpr/broadcast.py 240 binaries/aot_model_compiler.cc 222 binaries/lite_interpreter_model_load.cc 26 c10/core/Backend.h 273 c10/core/CPUAllocator.cpp 256 c10/core/CPUAllocator.h 39 c10/core/DeviceGuard.h 66 c10/core/DispatchKey.h 142 c10/core/DispatchKeySet.h 211 c10/core/GeneratorImpl.cpp 60 c10/core/GeneratorImpl.h 49 c10/core/MemoryFormat.h 145 c10/core/Scalar.h 161 c10/core/ScalarType.h 353 c10/core/ScalarTypeToTypeMeta.h 30 c10/core/Storage.h 128 c10/core/StorageImpl.h 135 c10/core/TensorImpl.cpp 482 c10/core/TensorImpl.h 1402 c10/core/TensorOptions.h 517 c10/core/impl/InlineDeviceGuard.h 187 c10/core/impl/InlineStreamGuard.h 147 c10/core/impl/SizesAndStrides.h 240 c10/core/thread_pool.h 90 c10/cuda/CUDACachingAllocator.h 94 c10/macros/Macros.h 313 c10/util/ArrayRef.h 222 c10/util/Backtrace.cpp 275 c10/util/C++17.h 291 c10/util/Deprecated.h 52 c10/util/Exception.cpp 163 c10/util/Exception.h 341 c10/util/IdWrapper.h 44 c10/util/MathConstants.h 27 c10/util/Metaprogramming.h 318 c10/util/SmallVector.h 946 c10/util/TypeCast.h 114 c10/util/TypeIndex.h 139 c10/util/TypeList.h 350 c10/util/complex_utils.h 29 c10/util/flags_use_no_gflags.cpp 169 c10/util/hash.h 235 c10/util/int128.h 264 c10/util/intrusive_ptr.h 620 c10/util/llvmMathExtras.h 568 c10/util/math_compat.h 209 c10/util/signal_handler.cpp 317 c10/util/string_view.h 545 c10/util/typeid.h 391 caffe2/contrib/fakelowp/layernorm_fp16_fake_op.h 159 caffe2/contrib/fakelowp/spatial_batch_norm_fp16_fake_op.h 333 caffe2/contrib/nccl/cuda_nccl_op_gpu.cc 231 caffe2/contrib/opencl/context.h 81 caffe2/contrib/playground/resnetdemo/explicit_resnet_forward.py 260 caffe2/contrib/pytorch/script_module_op.cc 132 caffe2/core/blob.h 106 caffe2/core/blob_serialization.cc 984 caffe2/core/blob_serialization.h 175 caffe2/core/common_cudnn.h 261 caffe2/core/common_gpu.h 322 caffe2/core/context.h 167 caffe2/core/context_base.cc 12 caffe2/core/context_base.h 123 caffe2/core/context_gpu.cu 603 caffe2/core/context_gpu.h 235 caffe2/core/export_c10_op_to_caffe2.h 207 caffe2/core/export_caffe2_op_to_c10.h 159 caffe2/core/init.cc 80 caffe2/core/net_async_task.cc 85 caffe2/core/net_async_task_graph.cc 110 caffe2/core/net_async_task_graph.h 44 caffe2/core/net_dag_utils.cc 402 caffe2/core/net_parallel.cc 163 caffe2/core/nomnigraph/Representations/NeuralNet.cc 317 caffe2/core/nomnigraph/include/nomnigraph/Graph/Algorithms.h 154 caffe2/core/nomnigraph/include/nomnigraph/Graph/TarjansImpl.h 122 caffe2/core/nomnigraph/include/nomnigraph/Graph/TopoSort.h 59 caffe2/core/nomnigraph/include/nomnigraph/Representations/ControlFlow.h 144 caffe2/core/nomnigraph/include/nomnigraph/Representations/NeuralNet.h 381 caffe2/core/nomnigraph/include/nomnigraph/Support/Common.h 76 caffe2/core/nomnigraph/include/nomnigraph/Transformations/Match.h 60 caffe2/core/nomnigraph/include/nomnigraph/Transformations/SubgraphMatcher.h 317 caffe2/core/qtensor.h 161 caffe2/core/tensor.cc 253 caffe2/core/tensor.h 373 caffe2/core/workspace.cc 278 caffe2/cuda_rtc/common_rtc.h 109 caffe2/experiments/operators/sparse_funhash_op.cc 70 caffe2/experiments/operators/sparse_matrix_reshape_op.cc 41 caffe2/experiments/python/SparseTransformer.py 131 caffe2/experiments/python/convnet_benchmarks.py 567 caffe2/experiments/python/net_construct_bench.py 97 caffe2/ideep/operators/adam_op.cc 166 caffe2/ideep/operators/momentum_sgd_op.cc 108 caffe2/image/image_input_op.h 1110 caffe2/mobile/contrib/ios/mpscnn/MPSCNN.metal 1032 caffe2/mobile/contrib/ios/mpscnn/mpscnn_graph.mm 313 caffe2/mobile/contrib/ios/mpscnn/mpscnn_kernels.h 1034 caffe2/mobile/contrib/ulp2/ulp.cc 344 caffe2/mobile/contrib/ulp2/ulp_neon.cc 451 caffe2/onnx/backend.cc 1595 caffe2/onnx/onnx_exporter.cc 1260 caffe2/operators/accumulate_op.h 32 caffe2/operators/batch_sparse_to_dense_op.cc 206 caffe2/operators/boolean_mask_ops.cc 515 caffe2/operators/ceil_op.cc 49 caffe2/operators/conv_op_impl.h 848 caffe2/operators/conv_transpose_op_mobile_impl.h 523 caffe2/operators/dataset_ops.h 149 caffe2/operators/distance_op.cc 636 caffe2/operators/distance_op.h 248 caffe2/operators/elementwise_ops.h 509 caffe2/operators/expand_op.h 104 caffe2/operators/floor_op.cc 51 caffe2/operators/gather_op.h 163 caffe2/operators/generate_proposals_op.cu 620 caffe2/operators/generate_proposals_op.h 116 caffe2/operators/generate_proposals_op_util_nms.h 549 caffe2/operators/length_split_op.cc 29 caffe2/operators/lengths_reducer_ops.h 597 caffe2/operators/load_save_op.cc 438 caffe2/operators/lp_pool_op.cc 264 caffe2/operators/lp_pool_op.cu 323 caffe2/operators/one_hot_ops.cc 310 caffe2/operators/order_switch_ops_cudnn.cc 147 caffe2/operators/prefetch_op.h 103 caffe2/operators/prelu_op.cu 257 caffe2/operators/reduction_ops.cc 297 caffe2/operators/rnn/recurrent_op_cudnn.cc 524 caffe2/operators/roi_pool_op.cc 163 caffe2/operators/segment_reduction_op.cc 495 caffe2/operators/segment_reduction_op.h 1612 caffe2/operators/sequence_ops.cc 481 caffe2/operators/sparse_to_dense_mask_op.cc 120 caffe2/operators/sparse_to_dense_mask_op.h 257 caffe2/operators/spatial_batch_norm_op_cudnn.cu 447 caffe2/operators/spatial_softmax_with_loss_op.cc 216 caffe2/operators/stylizer_ops.cc 430 caffe2/operators/top_k_heap_selection.cuh 251 caffe2/operators/tt_linear_op.h 151 caffe2/operators/utility_ops.cc 907 caffe2/operators/utility_ops.h 1250 caffe2/opt/bound_shape_inferencer.cc 1021 caffe2/opt/converter.cc 511 caffe2/opt/nql/graphmatcher.cc 222 caffe2/opt/nql/graphmatcher.h 85 caffe2/opt/onnxifi_transformer.cc 1264 caffe2/opt/tvm_transformer.cc 265 caffe2/perfkernels/embedding_lookup.h 21 caffe2/perfkernels/embedding_lookup_idx.h 21 caffe2/predictor/emulator/data_filler.cc 217 caffe2/predictor/predictor_config.h 29 caffe2/python/__init__.py 77 caffe2/python/core.py 1443 caffe2/python/data_parallel_model.py 1765 caffe2/python/dataset.py 179 caffe2/python/layer_model_helper.py 570 caffe2/python/layers/feature_sparse_to_dense.py 259 caffe2/python/layers/homotopy_weight.py 103 caffe2/python/layers/sparse_lookup.py 455 caffe2/python/memonger.py 790 caffe2/python/model_helper.py 473 caffe2/python/models/seq2seq/beam_search.py 424 caffe2/python/onnx/backend.py 720 caffe2/python/pybind_state.cc 1776 caffe2/python/pybind_state.h 404 caffe2/python/pybind_state_dlpack.h 112 caffe2/python/pybind_state_ideep.cc 175 caffe2/python/regularizer.py 367 caffe2/python/rnn_cell.py 1660 caffe2/python/schema.py 800 caffe2/python/utils.py 268 caffe2/quantization/server/conv_dnnlowp_op.cc 1498 caffe2/quantization/server/conv_pool_dnnlowp_op_base.h 193 caffe2/quantization/server/dnnlowp_op.h 214 caffe2/quantization/server/dnnlowp_test_utils.py 326 caffe2/quantization/server/fbgemm_pack_matrix_cache.cc 62 caffe2/quantization/server/fbgemm_pack_op.cc 805 caffe2/quantization/server/fully_connected_dnnlowp_acc16_op.cc 242 caffe2/quantization/server/op_wrapper.h 70 caffe2/quantization/server/transpose.h 5 caffe2/quantization/server/utility_dnnlowp_ops.h 84 caffe2/queue/rebatching_queue.h 44 caffe2/serialize/versions.h 22 caffe2/sgd/adagrad_fused_op_gpu.cu 1530 caffe2/sgd/adagrad_fused_op_gpu.cuh 224 caffe2/sgd/math_lp.cc 36 caffe2/share/contrib/depthwise/depthwise3x3_conv_op.cc 470 caffe2/utils/GpuScanUtils.cuh 103 caffe2/utils/cpuid.h 116 caffe2/utils/filler.h 112 caffe2/utils/math.h 369 caffe2/utils/math_cpu.cc 2441 caffe2/utils/math_gpu.cu 2801 caffe2/video/video_input_op.h 882 cmake/Dependencies.cmake 1814 cmake/External/nccl.cmake 79 cmake/MiscCheck.cmake 297 cmake/Modules/FindMKL.cmake 385 cmake/Modules/FindNumPy.cmake 50 cmake/Modules/FindOpenMP.cmake 588 cmake/TorchConfig.cmake.in 171 cmake/public/LoadHIP.cmake 221 cmake/public/cuda.cmake 476 modules/detectron/upsample_nearest_op.cu 200 scripts/model_zoo/update-caffe2-models.py 127 scripts/model_zoo/update-models-from-caffe2.py 249 setup.py 691 tools/actions_local_runner.py 326 tools/amd_build/build_amd.py 147 tools/autograd/gen_autograd_functions.py 290 tools/autograd/gen_inplace_or_view_type.py 309 tools/autograd/gen_python_functions.py 581 tools/autograd/gen_trace_type.py 252 tools/autograd/gen_variable_factories.py 63 tools/autograd/gen_variable_type.py 643 tools/autograd/load_derivatives.py 430 tools/autograd/templates/python_variable_methods.cpp 1049 tools/autograd/templates/variable_factories.h 75 tools/build_variables.bzl 1266 tools/code_analyzer/gen_op_registration_allowlist.py 56 tools/code_coverage/package/oss/utils.py 68 tools/codegen/api/autograd.py 169 tools/codegen/api/cpp.py 238 tools/codegen/api/lazy.py 113 tools/codegen/api/native.py 85 tools/codegen/api/python.py 625 tools/codegen/api/structured.py 82 tools/codegen/api/translate.py 128 tools/codegen/api/types.py 412 tools/codegen/dest/lazy_ir.py 160 tools/codegen/dest/lazy_ts_lowering.py 24 tools/codegen/dest/register_dispatch_key.py 434 tools/codegen/gen.py 1190 tools/codegen/gen_backend_stubs.py 261 tools/codegen/model.py 1023 tools/codegen/operator_versions/gen_mobile_upgraders.py 227 tools/codegen/utils.py 162 tools/coverage_plugins_package/src/coverage_plugins/jit_plugin.py 37 tools/pyi/gen_pyi.py 486 tools/setup_helpers/generate_code.py 140 tools/shared/cwrap_common.py 163 tools/stats/print_test_stats.py 896 tools/stats/upload_binary_size_to_scuba.py 154 torch/_C/__init__.pyi.in 981 torch/__config__.py 8 torch/_deploy.py 75 torch/_jit_internal.py 674 torch/_masked/__init__.py 528 torch/_tensor.py 665 torch/_tensor_str.py 309 torch/_utils.py 266 torch/ao/nn/sparse/quantized/dynamic/linear.py 96 torch/ao/nn/sparse/quantized/linear.py 150 torch/ao/ns/fx/graph_passes.py 539 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/quantization_state.py 556 torch/ao/quantization/_dbr/utils.py 439 torch/ao/quantization/_quantize_dbr.py 74 torch/ao/quantization/fx/_convert_do_not_use.py 210 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/tensorrt.py 190 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/match_utils.py 119 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/utils.py 364 torch/ao/quantization/observer.py 877 torch/ao/quantization/qconfig.py 213 torch/ao/quantization/quantization_mappings.py 228 torch/ao/quantization/quantize.py 322 torch/ao/quantization/quantize_fx.py 288 torch/ao/quantization/utils.py 158 torch/ao/sparsity/experimental/pruner/base_pruner.py 166 torch/ao/sparsity/sparsifier/base_sparsifier.py 153 torch/ao/sparsity/sparsifier/weight_norm_sparsifier.py 72 torch/autocast_mode.py 87 torch/autograd/_functions/tensor.py 42 torch/autograd/gradcheck.py 946 torch/autograd/profiler.py 346 torch/backends/_nnapi/prepare.py 109 torch/backends/_nnapi/serializer.py 1641 torch/contrib/_tensorboard_vis.py 86 torch/cpu/amp/autocast_mode.py 23 torch/csrc/CudaIPCTypes.cpp 214 torch/csrc/CudaIPCTypes.h 113 torch/csrc/DataLoader.cpp 161 torch/csrc/DynamicTypes.cpp 160 torch/csrc/api/include/torch/detail/TensorDataContainer.h 222 torch/csrc/api/include/torch/nn/parallel/data_parallel.h 175 torch/csrc/api/include/torch/types.h 29 torch/csrc/api/src/nn/modules/transformer.cpp 317 torch/csrc/autograd/FunctionsManual.cpp 3181 torch/csrc/autograd/TraceTypeManual.cpp 253 torch/csrc/autograd/VariableTypeManual.cpp 297 torch/csrc/autograd/VariableTypeUtils.h 274 torch/csrc/autograd/autograd.cpp 147 torch/csrc/autograd/autograd_meta.cpp 127 torch/csrc/autograd/autograd_not_implemented_fallback.cpp 255 torch/csrc/autograd/custom_function.cpp 327 torch/csrc/autograd/custom_function.h 234 torch/csrc/autograd/engine.cpp 832 torch/csrc/autograd/function.h 298 torch/csrc/autograd/functions/basic_ops.cpp 44 torch/csrc/autograd/functions/tensor.cpp 88 torch/csrc/autograd/input_buffer.cpp 141 torch/csrc/autograd/profiler_kineto.cpp 638 torch/csrc/autograd/profiler_legacy.cpp 442 torch/csrc/autograd/profiler_python.cpp 471 torch/csrc/autograd/python_engine.cpp 328 torch/csrc/autograd/python_function.cpp 809 torch/csrc/autograd/python_variable.cpp 1321 torch/csrc/autograd/python_variable_indexing.cpp 370 torch/csrc/autograd/variable.cpp 510 torch/csrc/autograd/variable.h 258 torch/csrc/deploy/elf_file.h 51 torch/csrc/deploy/interpreter/interpreter_impl.cpp 330 torch/csrc/distributed/autograd/autograd.cpp 23 torch/csrc/distributed/autograd/context/context.cpp 232 torch/csrc/distributed/autograd/engine/dist_engine.h 73 torch/csrc/distributed/c10d/ProcessGroupGloo.cpp 2263 torch/csrc/distributed/c10d/ProcessGroupGloo.hpp 213 torch/csrc/distributed/c10d/ProcessGroupWrapper.cpp 288 torch/csrc/distributed/c10d/TCPStore.cpp 873 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/quantization/quantization.cpp 85 torch/csrc/distributed/c10d/quantization/quantization_gpu.cu 134 torch/csrc/distributed/c10d/reducer.hpp 282 torch/csrc/distributed/c10d/socket.cpp 607 torch/csrc/distributed/rpc/python_functions.cpp 359 torch/csrc/distributed/rpc/request_callback_impl.cpp 287 torch/csrc/distributed/rpc/request_callback_no_python.cpp 471 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.h 117 torch/csrc/distributed/rpc/script_call.cpp 119 torch/csrc/distributed/rpc/tensorpipe_agent.cpp 1019 torch/csrc/distributed/rpc/tensorpipe_agent.h 281 torch/csrc/distributed/rpc/tensorpipe_utils.cpp 259 torch/csrc/generic/Storage.cpp 311 torch/csrc/generic/StorageSharing.cpp 486 torch/csrc/jit/api/compilation_unit.h 251 torch/csrc/jit/api/function_impl.cpp 111 torch/csrc/jit/backends/backend_detail.cpp 263 torch/csrc/jit/backends/nnapi/nnapi_backend_lib.cpp 92 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/compute_at_map.h 46 torch/csrc/jit/codegen/cuda/executor.cpp 781 torch/csrc/jit/codegen/cuda/executor.h 141 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/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_reference_replay.cpp 277 torch/csrc/jit/codegen/cuda/interface.cpp 228 torch/csrc/jit/codegen/cuda/ir_all_nodes.h 5 torch/csrc/jit/codegen/cuda/ir_base_nodes.h 205 torch/csrc/jit/codegen/cuda/ir_internal_nodes.h 602 torch/csrc/jit/codegen/cuda/ir_nodes.cpp 1269 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_cache.cpp 468 torch/csrc/jit/codegen/cuda/kernel_cache.h 201 torch/csrc/jit/codegen/cuda/kernel_ir.h 1221 torch/csrc/jit/codegen/cuda/lower2device.h 109 torch/csrc/jit/codegen/cuda/lower_alias_memory.cpp 798 torch/csrc/jit/codegen/cuda/lower_allocation.cpp 482 torch/csrc/jit/codegen/cuda/lower_expr_sort.cpp 889 torch/csrc/jit/codegen/cuda/lower_insert_syncs.cpp 451 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_warp_reduce.cpp 394 torch/csrc/jit/codegen/cuda/manager.cpp 159 torch/csrc/jit/codegen/cuda/manager.h 16 torch/csrc/jit/codegen/cuda/parser.cpp 2466 torch/csrc/jit/codegen/cuda/parser.h 28 torch/csrc/jit/codegen/cuda/partition.cpp 242 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/registry.cpp 967 torch/csrc/jit/codegen/cuda/tensor_view.cpp 654 torch/csrc/jit/codegen/cuda/transform_replay.cpp 540 torch/csrc/jit/codegen/cuda/transform_rfactor.h 19 torch/csrc/jit/codegen/cuda/type.h 207 torch/csrc/jit/codegen/cuda/type_inference.cpp 524 torch/csrc/jit/codegen/cuda/type_promotion.cpp 178 torch/csrc/jit/codegen/fuser/codegen.cpp 576 torch/csrc/jit/codegen/fuser/compiler.cpp 229 torch/csrc/jit/codegen/fuser/executor.cpp 317 torch/csrc/jit/codegen/fuser/kernel_cache.cpp 67 torch/csrc/jit/codegen/fuser/kernel_spec.h 114 torch/csrc/jit/frontend/ir_emitter.cpp 4328 torch/csrc/jit/frontend/schema_matching.cpp 593 torch/csrc/jit/frontend/schema_type_parser.cpp 376 torch/csrc/jit/frontend/source_range.h 216 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/tree_views.h 1021 torch/csrc/jit/ir/alias_analysis.cpp 1419 torch/csrc/jit/ir/ir.cpp 1885 torch/csrc/jit/ir/ir.h 1233 torch/csrc/jit/ir/irparser.cpp 401 torch/csrc/jit/ir/scope.cpp 168 torch/csrc/jit/ir/subgraph_matcher.cpp 277 torch/csrc/jit/ir/subgraph_matcher.h 15 torch/csrc/jit/mobile/code.h 25 torch/csrc/jit/mobile/compatibility/model_compatibility.cpp 316 torch/csrc/jit/mobile/function.cpp 216 torch/csrc/jit/mobile/function.h 61 torch/csrc/jit/mobile/model_tracer/TracerRunner.cpp 217 torch/csrc/jit/mobile/module.cpp 218 torch/csrc/jit/mobile/nnc/aot_compiler.cpp 132 torch/csrc/jit/mobile/nnc/context.cpp 236 torch/csrc/jit/mobile/nnc/context.h 118 torch/csrc/jit/mobile/observer.h 71 torch/csrc/jit/mobile/promoted_prim_ops.cpp 155 torch/csrc/jit/mobile/type_parser.cpp 271 torch/csrc/jit/passes/autocast.cpp 365 torch/csrc/jit/passes/batch_mm.cpp 378 torch/csrc/jit/passes/concat_opt.cpp 447 torch/csrc/jit/passes/constant_propagation.cpp 374 torch/csrc/jit/passes/create_functional_graphs.cpp 166 torch/csrc/jit/passes/freeze_module.cpp 694 torch/csrc/jit/passes/frozen_ops_to_mkldnn.cpp 914 torch/csrc/jit/passes/graph_fuser.cpp 918 torch/csrc/jit/passes/graph_rewrite_helper.cpp 228 torch/csrc/jit/passes/onnx.cpp 405 torch/csrc/jit/passes/onnx/function_extraction.cpp 876 torch/csrc/jit/passes/onnx/list_model_parameters.cpp 154 torch/csrc/jit/passes/onnx/peephole.cpp 728 torch/csrc/jit/passes/onnx/remove_inplace_ops_for_onnx.cpp 608 torch/csrc/jit/passes/onnx/scalar_type_analysis.cpp 319 torch/csrc/jit/passes/onnx/shape_type_inference.cpp 2036 torch/csrc/jit/passes/peephole.cpp 375 torch/csrc/jit/passes/peephole_non_tensor.cpp 222 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_quant_dequant.cpp 1300 torch/csrc/jit/passes/quantization/quantization_patterns.h 991 torch/csrc/jit/passes/remove_mutation.cpp 276 torch/csrc/jit/passes/restore_mutation.cpp 61 torch/csrc/jit/passes/specialize_autogradzero.cpp 379 torch/csrc/jit/passes/subgraph_rewrite.cpp 166 torch/csrc/jit/passes/symbolic_shape_analysis.cpp 696 torch/csrc/jit/passes/symbolic_shape_runtime_fusion.cpp 557 torch/csrc/jit/passes/tensorexpr_fuser.cpp 1000 torch/csrc/jit/passes/utils/subgraph_utils.cpp 500 torch/csrc/jit/passes/value_refinement_utils.cpp 143 torch/csrc/jit/passes/value_refinement_utils.h 60 torch/csrc/jit/passes/xnnpack_rewrite.cpp 450 torch/csrc/jit/python/pybind.h 179 torch/csrc/jit/python/pybind_utils.h 922 torch/csrc/jit/python/python_ir.cpp 977 torch/csrc/jit/python/python_list.cpp 275 torch/csrc/jit/python/python_list.h 155 torch/csrc/jit/python/script_init.cpp 1978 torch/csrc/jit/runtime/argument_spec.cpp 240 torch/csrc/jit/runtime/autodiff.cpp 575 torch/csrc/jit/runtime/graph_executor.cpp 793 torch/csrc/jit/runtime/instruction.cpp 79 torch/csrc/jit/runtime/instruction.h 62 torch/csrc/jit/runtime/logging.cpp 58 torch/csrc/jit/runtime/operator.h 171 torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp 560 torch/csrc/jit/runtime/profiling_record.cpp 264 torch/csrc/jit/runtime/register_c10_ops.cpp 45 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/shape_functions.h 393 torch/csrc/jit/runtime/shape_functions_1.h 8 torch/csrc/jit/runtime/static/fusion.cpp 276 torch/csrc/jit/runtime/static/impl.cpp 1550 torch/csrc/jit/runtime/static/memory_planner.cpp 343 torch/csrc/jit/runtime/static/memory_planner.h 128 torch/csrc/jit/runtime/static/ops.cpp 2339 torch/csrc/jit/runtime/static/passes.cpp 746 torch/csrc/jit/runtime/symbolic_script.cpp 1296 torch/csrc/jit/runtime/symbolic_shape_registry.cpp 233 torch/csrc/jit/runtime/symbolic_shape_registry_util.cpp 107 torch/csrc/jit/runtime/vararg_functions.cpp 375 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_module.cpp 658 torch/csrc/jit/serialization/flatbuffer_serializer.cpp 581 torch/csrc/jit/serialization/import.cpp 338 torch/csrc/jit/serialization/import_legacy.cpp 337 torch/csrc/jit/serialization/import_source.cpp 642 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/unpickler.cpp 878 torch/csrc/jit/tensorexpr/block_codegen.cpp 315 torch/csrc/jit/tensorexpr/codegen.cpp 190 torch/csrc/jit/tensorexpr/codegen.h 179 torch/csrc/jit/tensorexpr/cpp_codegen.cpp 341 torch/csrc/jit/tensorexpr/cuda_codegen.cpp 1027 torch/csrc/jit/tensorexpr/eval.cpp 1097 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_registry.h 27 torch/csrc/jit/tensorexpr/graph_opt.cpp 402 torch/csrc/jit/tensorexpr/graph_opt.h 27 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_printer.cpp 565 torch/csrc/jit/tensorexpr/ir_simplifier.cpp 2198 torch/csrc/jit/tensorexpr/ir_verifier.cpp 180 torch/csrc/jit/tensorexpr/kernel.cpp 1440 torch/csrc/jit/tensorexpr/llvm_codegen.cpp 1790 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/lowerings.cpp 1456 torch/csrc/jit/tensorexpr/mem_dependency_checker.cpp 944 torch/csrc/jit/tensorexpr/mem_dependency_checker.h 254 torch/csrc/jit/tensorexpr/operators/matmul.cpp 65 torch/csrc/jit/tensorexpr/operators/misc.cpp 542 torch/csrc/jit/tensorexpr/operators/quantization.cpp 720 torch/csrc/jit/tensorexpr/reduction.h 181 torch/csrc/jit/tensorexpr/registerizer.cpp 559 torch/csrc/jit/tensorexpr/stmt.h 731 torch/csrc/lazy/core/config.cpp 31 torch/csrc/lazy/core/helpers.h 41 torch/csrc/lazy/core/ir_dump_util.cpp 245 torch/csrc/lazy/core/lazy_graph_executor.cpp 859 torch/csrc/lazy/core/lazy_graph_executor.h 167 torch/csrc/lazy/core/tensor.cpp 417 torch/csrc/lazy/core/tensor_util.cpp 60 torch/csrc/serialization.cpp 126 torch/csrc/tensor/python_tensor.cpp 301 torch/csrc/tensor/python_tensor.h 17 torch/csrc/utils/auto_gil.h 24 torch/csrc/utils/python_arg_parser.cpp 926 torch/csrc/utils/python_arg_parser.h 576 torch/csrc/utils/python_dispatch.cpp 158 torch/csrc/utils/tensor_new.cpp 966 torch/csrc/utils/tensor_numpy.cpp 336 torch/csrc/utils/throughput_benchmark-inl.h 118 torch/csrc/utils/throughput_benchmark.cpp 116 torch/cuda/amp/autocast_mode.py 82 torch/cuda/amp/grad_scaler.py 255 torch/custom_class.h 335 torch/custom_class_detail.h 163 torch/distributed/_fsdp/fully_sharded_data_parallel.py 537 torch/distributed/_sharded_optim/api.py 37 torch/distributed/_sharded_tensor/api.py 473 torch/distributed/_sharding_spec/_internals.py 63 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/powerSGD_hook.py 399 torch/distributed/distributed_c10d.py 1327 torch/distributed/elastic/agent/server/api.py 524 torch/distributed/elastic/multiprocessing/api.py 469 torch/distributed/elastic/rendezvous/etcd_rendezvous.py 566 torch/distributed/elastic/utils/distributed.py 95 torch/distributed/nn/api/remote_module.py 343 torch/distributed/nn/jit/templates/remote_module_template.py 9 torch/distributed/optim/functional_adagrad.py 67 torch/distributed/optim/functional_sgd.py 104 torch/distributed/optim/optimizer.py 125 torch/distributed/optim/zero_redundancy_optimizer.py 745 torch/distributed/rpc/backend_registry.py 219 torch/distributions/constraint_registry.py 111 torch/distributions/kl.py 565 torch/distributions/laplace.py 55 torch/distributions/mixture_same_family.py 122 torch/distributions/uniform.py 59 torch/functional.py 494 torch/fx/_symbolic_trace.py 388 torch/fx/experimental/accelerator_partitioner.py 717 torch/fx/experimental/fx2trt/converters/acc_ops_converters.py 1769 torch/fx/experimental/fx2trt/converters/linear.py 43 torch/fx/experimental/fx2trt/lower.py 205 torch/fx/experimental/fx2trt/passes/fuse_pass.py 158 torch/fx/experimental/fx2trt/split.py 113 torch/fx/experimental/fx_acc/acc_tracer.py 223 torch/fx/experimental/merge_matmul.py 87 torch/fx/experimental/optimization.py 286 torch/fx/experimental/schema_type_annotation.py 63 torch/fx/operator_schemas.py 244 torch/fx/passes/split_module.py 140 torch/fx/passes/splitter_base.py 488 torch/jit/_builtins.py 133 torch/jit/_check.py 101 torch/jit/_monkeytype_config.py 109 torch/jit/_recursive.py 551 torch/jit/_script.py 791 torch/jit/_serialization.py 44 torch/jit/_trace.py 705 torch/jit/annotations.py 312 torch/jit/frontend.py 775 torch/jit/quantized.py 432 torch/lib/libshm/core.cpp 137 torch/lib/libshm/manager.cpp 155 torch/library.h 469 torch/multiprocessing/reductions.py 195 torch/nn/_reduction.py 33 torch/nn/functional.py 2645 torch/nn/intrinsic/quantized/dynamic/modules/linear_relu.py 23 torch/nn/intrinsic/quantized/modules/bn_relu.py 36 torch/nn/modules/activation.py 418 torch/nn/modules/batchnorm.py 332 torch/nn/modules/conv.py 658 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/pooling.py 305 torch/nn/modules/rnn.py 580 torch/nn/parallel/comm.py 100 torch/nn/parallel/data_parallel.py 105 torch/nn/parallel/distributed.py 791 torch/nn/parallel/distributed.pyi 17 torch/nn/parallel/scatter_gather.pyi 10 torch/nn/quantizable/modules/activation.py 297 torch/nn/quantized/dynamic/modules/rnn.py 613 torch/nn/quantized/modules/batchnorm.py 57 torch/nn/quantized/modules/conv.py 505 torch/nn/utils/_stateless.py 33 torch/nn/utils/memory_format.py 9 torch/nn/utils/prune.py 461 torch/nn/utils/rnn.py 156 torch/nn/utils/weight_norm.py 59 torch/onnx/symbolic_helper.py 686 torch/onnx/symbolic_opset13.py 272 torch/onnx/symbolic_opset15.py 1 torch/onnx/symbolic_opset9.py 2371 torch/onnx/utils.py 915 torch/optim/_functional.py 371 torch/package/importer.py 114 torch/package/package_exporter.py 631 torch/package/package_importer.py 422 torch/quantization/__init__.py 44 torch/quantization/fuse_modules.py 11 torch/serialization.py 647 torch/testing/_comparison.py 658 torch/testing/_core.py 118 torch/testing/_deprecated.py 69 torch/testing/_internal/common_device_type.py 587 torch/testing/_internal/common_distributed.py 554 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/distributed/rpc/tensorpipe_rpc_agent_test_fixture.py 21 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/utils/_python_dispatch.py 16 torch/utils/benchmark/utils/cpp_jit.py 98 torch/utils/benchmark/utils/valgrind_wrapper/timer_interface.py 498 torch/utils/bundled_inputs.py 221 torch/utils/data/_decorator.py 122 torch/utils/data/_typing.py 288 torch/utils/data/_utils/worker.py 87 torch/utils/data/communication/protocol.py 157 torch/utils/data/dataloader.py 515 torch/utils/data/datapipes/iter/fileopener.py 31 torch/utils/data/datapipes/utils/decoder.py 189 torch/utils/data/gen_pyi.py 141 torch/utils/dlpack.py 30 torch/utils/hipify/cuda_to_hip_mappings.py 8181 torch/utils/hipify/hipify_python.py 638 torch/utils/mkldnn.py 164 torch/utils/model_dump/__init__.py 252 torch/utils/model_dump/code.js 612 torch/utils/tensorboard/_pytorch_graph.py 198 torch/utils/tensorboard/summary.py 487 torch/utils/tensorboard/writer.py 347