Path Lines of Code MANIFEST.in 28 android/pytorch_android/generate_test_asset.cpp 16 android/pytorch_android/generate_test_torchscripts.py 108 android/pytorch_android/src/main/cpp/cmake_macros.h 1 android/pytorch_android/src/main/cpp/cmake_macros.h.in 2 android/pytorch_android/src/main/cpp/pytorch_jni_common.cpp 639 android/pytorch_android/src/main/cpp/pytorch_jni_common.h 107 android/pytorch_android/src/main/cpp/pytorch_jni_jit.cpp 228 android/pytorch_android/src/main/cpp/pytorch_jni_lite.cpp 192 android/pytorch_android/src/main/java/org/pytorch/DType.java 14 android/pytorch_android/src/main/java/org/pytorch/Device.java 10 android/pytorch_android/src/main/java/org/pytorch/INativePeer.java 6 android/pytorch_android/src/main/java/org/pytorch/IValue.java 269 android/pytorch_android/src/main/java/org/pytorch/LiteModuleLoader.java 20 android/pytorch_android/src/main/java/org/pytorch/LiteNativePeer.java 35 android/pytorch_android/src/main/java/org/pytorch/MemoryFormat.java 11 android/pytorch_android/src/main/java/org/pytorch/Module.java 29 android/pytorch_android/src/main/java/org/pytorch/NativePeer.java 36 android/pytorch_android/src/main/java/org/pytorch/PyTorchAndroid.java 30 android/pytorch_android/src/main/java/org/pytorch/PyTorchCodegenLoader.java 11 android/pytorch_android/src/main/java/org/pytorch/Tensor.java 472 android/pytorch_android_torchvision/src/main/cpp/pytorch_vision_jni.cpp 168 android/pytorch_android_torchvision/src/main/java/org/pytorch/torchvision/TensorImageUtils.java 301 aten.bzl 76 aten/conda/meta.yaml 27 aten/src/ATen/ATen.h 32 aten/src/ATen/ATenConfig.cmake.in 8 aten/src/ATen/AccumulateType.cpp 15 aten/src/ATen/AccumulateType.h 48 aten/src/ATen/ArrayRef.h 2 aten/src/ATen/Backend.h 2 aten/src/ATen/Backtrace.h 2 aten/src/ATen/BatchedFallback.cpp 245 aten/src/ATen/BatchedFallback.h 7 aten/src/ATen/BatchedTensorImpl.cpp 119 aten/src/ATen/BatchedTensorImpl.h 75 aten/src/ATen/BatchingRegistrations.cpp 833 aten/src/ATen/CPUApplyUtils.h 264 aten/src/ATen/CPUFixedAllocator.h 18 aten/src/ATen/CPUGeneratorImpl.cpp 170 aten/src/ATen/CPUGeneratorImpl.h 35 aten/src/ATen/CollapseDims.h 59 aten/src/ATen/Config.h.in 15 aten/src/ATen/ConjugateFallback.cpp 51 aten/src/ATen/Context.cpp 246 aten/src/ATen/Context.h 255 aten/src/ATen/DLConvertor.cpp 239 aten/src/ATen/DLConvertor.h 11 aten/src/ATen/Device.h 2 aten/src/ATen/DeviceGuard.h 23 aten/src/ATen/DimVector.h 2 aten/src/ATen/Dimname.h 1 aten/src/ATen/Dispatch.cpp 10 aten/src/ATen/Dispatch.h 669 aten/src/ATen/DynamicLibrary.cpp 86 aten/src/ATen/DynamicLibrary.h 13 aten/src/ATen/EmptyTensor.cpp 225 aten/src/ATen/EmptyTensor.h 78 aten/src/ATen/ExpandUtils.cpp 152 aten/src/ATen/ExpandUtils.h 269 aten/src/ATen/Formatting.h 1 aten/src/ATen/FuncTorchTLS.cpp 22 aten/src/ATen/FuncTorchTLS.h 12 aten/src/ATen/FunctionalInverses.cpp 155 aten/src/ATen/FunctionalStorageImpl.cpp 69 aten/src/ATen/FunctionalStorageImpl.h 46 aten/src/ATen/FunctionalTensorWrapper.cpp 227 aten/src/ATen/FunctionalTensorWrapper.h 62 aten/src/ATen/FunctionalizeFallbackKernel.cpp 50 aten/src/ATen/Generator.h 2 aten/src/ATen/InferSize.h 47 aten/src/ATen/InitialTensorOptions.h 8 aten/src/ATen/Layout.h 2 aten/src/ATen/LinalgBackend.h 24 aten/src/ATen/MapAllocator.cpp 500 aten/src/ATen/MapAllocator.h 73 aten/src/ATen/MatrixRef.h 46 aten/src/ATen/MemoryOverlap.cpp 78 aten/src/ATen/MemoryOverlap.h 20 aten/src/ATen/NamedTensor.h 1 aten/src/ATen/NamedTensorUtils.cpp 439 aten/src/ATen/NamedTensorUtils.h 96 aten/src/ATen/NumericUtils.h 103 aten/src/ATen/OpMathType.h 10 aten/src/ATen/OpaqueTensorImpl.h 122 aten/src/ATen/PTThreadPool.h 15 aten/src/ATen/Parallel-inl.h 73 aten/src/ATen/Parallel.h 67 aten/src/ATen/ParallelCommon.cpp 86 aten/src/ATen/ParallelFuture.h 8 aten/src/ATen/ParallelNative.cpp 244 aten/src/ATen/ParallelNative.h 15 aten/src/ATen/ParallelNativeTBB.cpp 91 aten/src/ATen/ParallelNativeTBB.h 42 aten/src/ATen/ParallelOpenMP.cpp 86 aten/src/ATen/ParallelOpenMP.h 42 aten/src/ATen/ParallelThreadPoolNative.cpp 69 aten/src/ATen/SavedTensorHooks.cpp 33 aten/src/ATen/SavedTensorHooks.h 15 aten/src/ATen/Scalar.h 2 aten/src/ATen/ScalarOps.cpp 31 aten/src/ATen/ScalarOps.h 49 aten/src/ATen/ScalarType.h 4 aten/src/ATen/SequenceNumber.cpp 14 aten/src/ATen/SequenceNumber.h 9 aten/src/ATen/SmallVector.h 2 aten/src/ATen/SparseCsrTensorImpl.cpp 106 aten/src/ATen/SparseCsrTensorImpl.h 31 aten/src/ATen/SparseCsrTensorUtils.h 16 aten/src/ATen/SparseTensorImpl.cpp 94 aten/src/ATen/SparseTensorImpl.h 167 aten/src/ATen/SparseTensorUtils.cpp 58 aten/src/ATen/SparseTensorUtils.h 35 aten/src/ATen/Storage.h 2 aten/src/ATen/Tensor.h 2 aten/src/ATen/TensorAccessor.h 2 aten/src/ATen/TensorGeometry.cpp 11 aten/src/ATen/TensorGeometry.h 51 aten/src/ATen/TensorIndexing.cpp 78 aten/src/ATen/TensorIndexing.h 411 aten/src/ATen/TensorIterator.cpp 1255 aten/src/ATen/TensorIterator.h 439 aten/src/ATen/TensorIteratorInternal.h 55 aten/src/ATen/TensorMeta.cpp 3 aten/src/ATen/TensorMeta.h 33 aten/src/ATen/TensorNames.cpp 96 aten/src/ATen/TensorNames.h 33 aten/src/ATen/TensorOperators.h 40 aten/src/ATen/TensorOptions.h 2 aten/src/ATen/TensorSubclassLikeUtils.h 22 aten/src/ATen/TensorUtils.cpp 348 aten/src/ATen/TensorUtils.h 123 aten/src/ATen/ThreadLocalState.cpp 31 aten/src/ATen/ThreadLocalState.h 55 aten/src/ATen/TracerMode.h 23 aten/src/ATen/TypeDefault.h 20 aten/src/ATen/Utils.cpp 71 aten/src/ATen/Utils.h 104 aten/src/ATen/Version.cpp 177 aten/src/ATen/Version.h 8 aten/src/ATen/VmapMode.cpp 23 aten/src/ATen/VmapMode.h 11 aten/src/ATen/VmapModeRegistrations.cpp 73 aten/src/ATen/VmapTransforms.cpp 204 aten/src/ATen/VmapTransforms.h 40 aten/src/ATen/WrapDimUtils.h 68 aten/src/ATen/WrapDimUtilsMulti.h 19 aten/src/ATen/ZeroTensorFallback.cpp 80 aten/src/ATen/autocast_mode.cpp 590 aten/src/ATen/autocast_mode.h 159 aten/src/ATen/benchmarks/quantize_per_channel.cpp 70 aten/src/ATen/benchmarks/stateful_conv1d.cpp 76 aten/src/ATen/benchmarks/tensor_add.cpp 24 aten/src/ATen/ceil_div.h 12 aten/src/ATen/code_template.h 196 aten/src/ATen/core/ATenGeneral.cpp 1 aten/src/ATen/core/ATenGeneral.h 2 aten/src/ATen/core/ATenOpList.h 8 aten/src/ATen/core/ATen_pch.h 145 aten/src/ATen/core/Array.h 30 aten/src/ATen/core/BackendSelectFallbackKernel.cpp 4 aten/src/ATen/core/Backtrace.h 2 aten/src/ATen/core/CheckMemoryFormat.h 22 aten/src/ATen/core/DeprecatedTypeProperties.cpp 18 aten/src/ATen/core/DeprecatedTypeProperties.h 100 aten/src/ATen/core/DeprecatedTypePropertiesRegistry.cpp 24 aten/src/ATen/core/DeprecatedTypePropertiesRegistry.h 19 aten/src/ATen/core/Dict.cpp 24 aten/src/ATen/core/Dict.h 193 aten/src/ATen/core/Dict_inl.h 169 aten/src/ATen/core/DimVector.h 7 aten/src/ATen/core/Dimname.cpp 62 aten/src/ATen/core/Dimname.h 34 aten/src/ATen/core/DistributionsHelper.h 242 aten/src/ATen/core/Formatting.cpp 302 aten/src/ATen/core/Formatting.h 33 aten/src/ATen/core/Generator.cpp 12 aten/src/ATen/core/Generator.h 82 aten/src/ATen/core/LegacyTypeDispatch.h 35 aten/src/ATen/core/List.cpp 14 aten/src/ATen/core/List.h 223 aten/src/ATen/core/List_inl.h 279 aten/src/ATen/core/MT19937RNGEngine.h 91 aten/src/ATen/core/Macros.h 2 aten/src/ATen/core/NamedRegistrations.cpp 525 aten/src/ATen/core/NamedTensor.cpp 123 aten/src/ATen/core/NamedTensor.h 77 aten/src/ATen/core/PhiloxRNGEngine.h 130 aten/src/ATen/core/PythonFallbackKernel.cpp 35 aten/src/ATen/core/PythonModeTLS.cpp 20 aten/src/ATen/core/PythonModeTLS.h 13 aten/src/ATen/core/QuantizerBase.h 28 aten/src/ATen/core/Range.cpp 8 aten/src/ATen/core/Range.h 17 aten/src/ATen/core/Reduction.h 11 aten/src/ATen/core/Scalar.h 1 aten/src/ATen/core/ScalarType.h 1 aten/src/ATen/core/Tensor.cpp 113 aten/src/ATen/core/Tensor.h 57 aten/src/ATen/core/TensorAccessor.h 206 aten/src/ATen/core/TensorBase.h 499 aten/src/ATen/core/TransformationHelper.h 96 aten/src/ATen/core/UndefinedTensorImpl.h 1 aten/src/ATen/core/UnsafeFromTH.h 17 aten/src/ATen/core/VariableFallbackKernel.cpp 38 aten/src/ATen/core/VariableHooksInterface.cpp 13 aten/src/ATen/core/VariableHooksInterface.h 38 aten/src/ATen/core/Variadic.h 52 aten/src/ATen/core/Vitals.cpp 81 aten/src/ATen/core/Vitals.h 68 aten/src/ATen/core/adaption.cpp 11 aten/src/ATen/core/alias_info.h 88 aten/src/ATen/core/blob.cpp 1 aten/src/ATen/core/blob.h 122 aten/src/ATen/core/boxing/KernelFunction.cpp 48 aten/src/ATen/core/boxing/KernelFunction.h 57 aten/src/ATen/core/boxing/KernelFunction_impl.h 164 aten/src/ATen/core/boxing/impl/WrapFunctionIntoFunctor.h 25 aten/src/ATen/core/boxing/impl/WrapFunctionIntoRuntimeFunctor.h 27 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/builtin_function.h 67 aten/src/ATen/core/class_type.cpp 591 aten/src/ATen/core/class_type.h 297 aten/src/ATen/core/custom_class.cpp 113 aten/src/ATen/core/custom_class.h 38 aten/src/ATen/core/dispatch/CppSignature.h 40 aten/src/ATen/core/dispatch/DispatchKeyExtractor.cpp 26 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/ObservedOperators.cpp 20 aten/src/ATen/core/dispatch/ObservedOperators.h 11 aten/src/ATen/core/dispatch/OperatorEntry.cpp 345 aten/src/ATen/core/dispatch/OperatorEntry.h 156 aten/src/ATen/core/dispatch/OperatorOptions.h 24 aten/src/ATen/core/dispatch/RegistrationHandleRAII.h 27 aten/src/ATen/core/dynamic_type.cpp 328 aten/src/ATen/core/dynamic_type.h 158 aten/src/ATen/core/enum_type.h 82 aten/src/ATen/core/function.h 66 aten/src/ATen/core/function_schema.cpp 7 aten/src/ATen/core/function_schema.h 364 aten/src/ATen/core/function_schema_inl.h 376 aten/src/ATen/core/functional.h 36 aten/src/ATen/core/grad_mode.h 8 aten/src/ATen/core/interned_strings.cpp 117 aten/src/ATen/core/interned_strings.h 318 aten/src/ATen/core/interned_strings_class.h 27 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/ivalue_to.h 26 aten/src/ATen/core/jit_type.h 1616 aten/src/ATen/core/jit_type_base.h 534 aten/src/ATen/core/library.cpp 203 aten/src/ATen/core/op_registration/adaption.h 37 aten/src/ATen/core/op_registration/infer_schema.cpp 61 aten/src/ATen/core/op_registration/infer_schema.h 100 aten/src/ATen/core/op_registration/op_allowlist.h 104 aten/src/ATen/core/op_registration/op_registration.cpp 86 aten/src/ATen/core/op_registration/op_registration.h 257 aten/src/ATen/core/operator_name.cpp 16 aten/src/ATen/core/operator_name.h 68 aten/src/ATen/core/qualified_name.h 120 aten/src/ATen/core/register_symbols.cpp 38 aten/src/ATen/core/rref_interface.h 20 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/core/type.cpp 836 aten/src/ATen/core/type_factory.cpp 47 aten/src/ATen/core/type_factory.h 68 aten/src/ATen/core/type_ptr.h 37 aten/src/ATen/core/typeid.h 1 aten/src/ATen/core/union_type.cpp 369 aten/src/ATen/cpp_custom_type_hack.h 40 aten/src/ATen/cpu/FlushDenormal.cpp 27 aten/src/ATen/cpu/FlushDenormal.h 3 aten/src/ATen/cpu/tbb/extra/version_string.ver.in 10 aten/src/ATen/cpu/vec/functional.h 3 aten/src/ATen/cpu/vec/functional_base.h 245 aten/src/ATen/cpu/vec/functional_bfloat16.h 459 aten/src/ATen/cpu/vec/intrinsics.h 32 aten/src/ATen/cpu/vec/vec.h 5 aten/src/ATen/cpu/vec/vec256/missing_vld1_neon.h 393 aten/src/ATen/cpu/vec/vec256/missing_vst1_neon.h 6 aten/src/ATen/cpu/vec/vec256/vec256.h 133 aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h 719 aten/src/ATen/cpu/vec/vec256/vec256_complex_double.h 379 aten/src/ATen/cpu/vec/vec256/vec256_complex_float.h 419 aten/src/ATen/cpu/vec/vec256/vec256_double.h 364 aten/src/ATen/cpu/vec/vec256/vec256_float.h 371 aten/src/ATen/cpu/vec/vec256/vec256_float_neon.h 645 aten/src/ATen/cpu/vec/vec256/vec256_int.h 1008 aten/src/ATen/cpu/vec/vec256/vec256_qint.h 931 aten/src/ATen/cpu/vec/vec256/vsx/vec256_bfloat16_vsx.h 47 aten/src/ATen/cpu/vec/vec256/vsx/vec256_common_vsx.h 163 aten/src/ATen/cpu/vec/vec256/vsx/vec256_complex_double_vsx.h 482 aten/src/ATen/cpu/vec/vec256/vsx/vec256_complex_float_vsx.h 530 aten/src/ATen/cpu/vec/vec256/vsx/vec256_double_vsx.h 361 aten/src/ATen/cpu/vec/vec256/vsx/vec256_float_vsx.h 508 aten/src/ATen/cpu/vec/vec256/vsx/vec256_int16_vsx.h 319 aten/src/ATen/cpu/vec/vec256/vsx/vec256_int32_vsx.h 250 aten/src/ATen/cpu/vec/vec256/vsx/vec256_int64_vsx.h 208 aten/src/ATen/cpu/vec/vec256/vsx/vec256_qint32_vsx.h 180 aten/src/ATen/cpu/vec/vec256/vsx/vec256_qint8_vsx.h 300 aten/src/ATen/cpu/vec/vec256/vsx/vec256_quint8_vsx.h 314 aten/src/ATen/cpu/vec/vec256/vsx/vsx_helpers.h 378 aten/src/ATen/cpu/vec/vec256/zarch/vec256_zarch.h 2070 aten/src/ATen/cpu/vec/vec512/vec512.h 127 aten/src/ATen/cpu/vec/vec512/vec512_bfloat16.h 820 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/cpu/vec/vec512/vec512_double.h 399 aten/src/ATen/cpu/vec/vec512/vec512_float.h 418 aten/src/ATen/cpu/vec/vec512/vec512_int.h 1056 aten/src/ATen/cpu/vec/vec512/vec512_qint.h 943 aten/src/ATen/cpu/vec/vec_base.h 802 aten/src/ATen/cpu/vml.h 150 aten/src/ATen/cuda/ATenCUDAGeneral.h 5 aten/src/ATen/cuda/ApplyGridUtils.cuh 37 aten/src/ATen/cuda/AsmUtils.cuh 131 aten/src/ATen/cuda/Atomic.cuh 296 aten/src/ATen/cuda/CUDAApplyUtils.cuh 473 aten/src/ATen/cuda/CUDABlas.cpp 1036 aten/src/ATen/cuda/CUDABlas.h 239 aten/src/ATen/cuda/CUDAConfig.h.in 8 aten/src/ATen/cuda/CUDAContext.cpp 51 aten/src/ATen/cuda/CUDAContext.h 35 aten/src/ATen/cuda/CUDADataType.h 86 aten/src/ATen/cuda/CUDADevice.h 12 aten/src/ATen/cuda/CUDAEvent.h 133 aten/src/ATen/cuda/CUDAGeneratorImpl.cpp 176 aten/src/ATen/cuda/CUDAGeneratorImpl.h 40 aten/src/ATen/cuda/CUDAGraph.cpp 133 aten/src/ATen/cuda/CUDAGraph.h 34 aten/src/ATen/cuda/CUDAGraphsUtils.cuh 51 aten/src/ATen/cuda/CUDASolver.cpp 1648 aten/src/ATen/cuda/CUDASolver.h 545 aten/src/ATen/cuda/CUDASparse.h 29 aten/src/ATen/cuda/CUDASparseBlas.cpp 845 aten/src/ATen/cuda/CUDASparseBlas.h 276 aten/src/ATen/cuda/CUDASparseDescriptors.cpp 132 aten/src/ATen/cuda/CUDASparseDescriptors.h 178 aten/src/ATen/cuda/CUDATensorMethods.cuh 12 aten/src/ATen/cuda/CUDAUtils.h 14 aten/src/ATen/cuda/CachingHostAllocator.cpp 213 aten/src/ATen/cuda/CachingHostAllocator.h 13 aten/src/ATen/cuda/CuSparseHandlePool.cpp 26 aten/src/ATen/cuda/CublasHandlePool.cpp 45 aten/src/ATen/cuda/CusolverDnHandlePool.cpp 29 aten/src/ATen/cuda/DeviceUtils.cuh 104 aten/src/ATen/cuda/EmptyTensor.cpp 81 aten/src/ATen/cuda/EmptyTensor.h 36 aten/src/ATen/cuda/Exceptions.h 114 aten/src/ATen/cuda/NumericLimits.cuh 103 aten/src/ATen/cuda/PeerToPeerAccess.cpp 46 aten/src/ATen/cuda/PeerToPeerAccess.h 9 aten/src/ATen/cuda/PinnedMemoryAllocator.cpp 28 aten/src/ATen/cuda/PinnedMemoryAllocator.h 8 aten/src/ATen/cuda/ScanUtils.cuh 131 aten/src/ATen/cuda/Sleep.cu 23 aten/src/ATen/cuda/Sleep.h 7 aten/src/ATen/cuda/ThrustAllocator.h 17 aten/src/ATen/cuda/cub.cu 126 aten/src/ATen/cuda/cub.cuh 280 aten/src/ATen/cuda/cub.h 65 aten/src/ATen/cuda/cub_definitions.cuh 32 aten/src/ATen/cuda/detail/CUDAHooks.cpp 312 aten/src/ATen/cuda/detail/CUDAHooks.h 38 aten/src/ATen/cuda/detail/DeviceThreadHandles.h 70 aten/src/ATen/cuda/detail/IndexUtils.cu 63 aten/src/ATen/cuda/detail/IndexUtils.cuh 25 aten/src/ATen/cuda/detail/IntegerDivider.cuh 108 aten/src/ATen/cuda/detail/KernelUtils.h 16 aten/src/ATen/cuda/detail/LazyNVRTC.cpp 214 aten/src/ATen/cuda/detail/LazyNVRTC.h 8 aten/src/ATen/cuda/detail/OffsetCalculator.cuh 105 aten/src/ATen/cuda/detail/PhiloxCudaStateRaw.cuh 38 aten/src/ATen/cuda/detail/TensorInfo.cuh 98 aten/src/ATen/cuda/detail/UnpackRaw.cuh 29 aten/src/ATen/cuda/nvrtc_stub/ATenNVRTC.cpp 10 aten/src/ATen/cuda/nvrtc_stub/ATenNVRTC.h 73 aten/src/ATen/cudnn/AutocastRNN.cpp 100 aten/src/ATen/cudnn/Descriptors.cpp 169 aten/src/ATen/cudnn/Descriptors.h 250 aten/src/ATen/cudnn/Exceptions.h 1 aten/src/ATen/cudnn/Handle.cpp 23 aten/src/ATen/cudnn/Handle.h 6 aten/src/ATen/cudnn/Handles.h 2 aten/src/ATen/cudnn/Types.cpp 22 aten/src/ATen/cudnn/Types.h 9 aten/src/ATen/cudnn/Utils.h 13 aten/src/ATen/cudnn/cudnn-wrapper.h 11 aten/src/ATen/detail/CPUGuardImpl.cpp 6 aten/src/ATen/detail/CUDAHooksInterface.cpp 27 aten/src/ATen/detail/CUDAHooksInterface.h 134 aten/src/ATen/detail/FunctionTraits.h 44 aten/src/ATen/detail/HIPHooksInterface.cpp 29 aten/src/ATen/detail/HIPHooksInterface.h 44 aten/src/ATen/detail/MetaGuardImpl.cpp 6 aten/src/ATen/detail/ORTHooksInterface.cpp 22 aten/src/ATen/detail/ORTHooksInterface.h 22 aten/src/ATen/div_rtn.h 8 aten/src/ATen/dlpack.h 70 aten/src/ATen/function_wrapper.py 1 aten/src/ATen/gen_vulkan_glsl.py 89 aten/src/ATen/gen_vulkan_spv.py 123 aten/src/ATen/hip/impl/HIPAllocatorMasqueradingAsCUDA.h 19 aten/src/ATen/hip/impl/HIPCachingAllocatorMasqueradingAsCUDA.cpp 13 aten/src/ATen/hip/impl/HIPCachingAllocatorMasqueradingAsCUDA.h 13 aten/src/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.cpp 2 aten/src/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.h 248 aten/src/ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h 83 aten/src/ATen/jit_macros.h 12 aten/src/ATen/metal/Context.cpp 24 aten/src/ATen/metal/Context.h 24 aten/src/ATen/miopen/Descriptors.cpp 107 aten/src/ATen/miopen/Descriptors.h 108 aten/src/ATen/miopen/Exceptions.h 35 aten/src/ATen/miopen/Handle.cpp 24 aten/src/ATen/miopen/Handle.h 5 aten/src/ATen/miopen/Types.cpp 20 aten/src/ATen/miopen/Types.h 7 aten/src/ATen/miopen/Utils.h 12 aten/src/ATen/miopen/miopen-wrapper.h 2 aten/src/ATen/mkl/Descriptors.h 36 aten/src/ATen/mkl/Exceptions.h 55 aten/src/ATen/mkl/Limits.h 5 aten/src/ATen/mkl/Sparse.h 7 aten/src/ATen/mkl/SparseBlas.cpp 309 aten/src/ATen/mkl/SparseBlas.h 170 aten/src/ATen/mkl/SparseDescriptors.h 112 aten/src/ATen/mkl/Utils.h 15 aten/src/ATen/native/Activation.cpp 800 aten/src/ATen/native/Activation.h 58 aten/src/ATen/native/AdaptiveAveragePooling.cpp 128 aten/src/ATen/native/AdaptiveAveragePooling3d.cpp 281 aten/src/ATen/native/AdaptiveMaxPooling2d.cpp 67 aten/src/ATen/native/AdaptiveMaxPooling3d.cpp 368 aten/src/ATen/native/AdaptivePooling.h 19 aten/src/ATen/native/AffineGridGenerator.cpp 119 aten/src/ATen/native/AutogradComposite.cpp 49 aten/src/ATen/native/AveragePool2d.cpp 172 aten/src/ATen/native/AveragePool3d.cpp 432 aten/src/ATen/native/BatchLinearAlgebra.cpp 2819 aten/src/ATen/native/BatchLinearAlgebra.h 197 aten/src/ATen/native/BatchLinearAlgebraKernel.cpp 683 aten/src/ATen/native/Batching.cpp 71 aten/src/ATen/native/BinaryOps.cpp 908 aten/src/ATen/native/BinaryOps.h 92 aten/src/ATen/native/Blas.cpp 156 aten/src/ATen/native/BlasKernel.cpp 312 aten/src/ATen/native/Bucketization.cpp 163 aten/src/ATen/native/BucketizationUtils.h 131 aten/src/ATen/native/CPUBlas.cpp 563 aten/src/ATen/native/CPUBlas.h 143 aten/src/ATen/native/CPUFallback.cpp 121 aten/src/ATen/native/CPUFallback.h 34 aten/src/ATen/native/CanUse32BitIndexMath.h 9 aten/src/ATen/native/ChanelShuffle.cpp 57 aten/src/ATen/native/Col2Im.cpp 166 aten/src/ATen/native/ComplexHelper.h 65 aten/src/ATen/native/CompositeRandomAccessor.h 26 aten/src/ATen/native/CompositeRandomAccessorCommon.h 185 aten/src/ATen/native/ConstantPadNd.cpp 70 aten/src/ATen/native/ConvUtils.h 221 aten/src/ATen/native/Convolution.cpp 1499 aten/src/ATen/native/ConvolutionMM2d.cpp 579 aten/src/ATen/native/ConvolutionMM3d.cpp 700 aten/src/ATen/native/ConvolutionMM3d.h 12 aten/src/ATen/native/ConvolutionTBC.cpp 84 aten/src/ATen/native/Copy.cpp 215 aten/src/ATen/native/Copy.h 11 aten/src/ATen/native/Correlation.cpp 107 aten/src/ATen/native/Cross.cpp 43 aten/src/ATen/native/Cross.h 7 aten/src/ATen/native/DilatedConvolutionUtils.h 189 aten/src/ATen/native/DilatedMaxPool2d.cpp 167 aten/src/ATen/native/DilatedMaxPool3d.cpp 495 aten/src/ATen/native/DispatchStub.cpp 159 aten/src/ATen/native/DispatchStub.h 191 aten/src/ATen/native/Distance.cpp 202 aten/src/ATen/native/Distance.h 13 aten/src/ATen/native/DistributionTemplates.h 296 aten/src/ATen/native/Distributions.cpp 446 aten/src/ATen/native/Distributions.h 392 aten/src/ATen/native/Dropout.cpp 128 aten/src/ATen/native/Embedding.cpp 144 aten/src/ATen/native/EmbeddingBag.cpp 1015 aten/src/ATen/native/EmbeddingBag.h 43 aten/src/ATen/native/Fill.cpp 104 aten/src/ATen/native/Fill.h 8 aten/src/ATen/native/ForeachOpsKernels.cpp 214 aten/src/ATen/native/ForeachUtils.h 100 aten/src/ATen/native/FractionalMaxPool2d.cpp 330 aten/src/ATen/native/FractionalMaxPool3d.cpp 373 aten/src/ATen/native/FunctionOfAMatrixUtils.cpp 65 aten/src/ATen/native/FunctionOfAMatrixUtils.h 14 aten/src/ATen/native/GatedLinearUnit.cpp 55 aten/src/ATen/native/GridSampler.cpp 813 aten/src/ATen/native/GridSampler.h 221 aten/src/ATen/native/Histogram.cpp 272 aten/src/ATen/native/Histogram.h 10 aten/src/ATen/native/Im2Col.cpp 183 aten/src/ATen/native/IndexKernel.h 33 aten/src/ATen/native/IndexingUtils.cpp 24 aten/src/ATen/native/IndexingUtils.h 128 aten/src/ATen/native/Integration.cpp 110 aten/src/ATen/native/Itertools.cpp 49 aten/src/ATen/native/LegacyBridge.cpp 1 aten/src/ATen/native/Lerp.cpp 36 aten/src/ATen/native/Lerp.h 15 aten/src/ATen/native/Linear.cpp 556 aten/src/ATen/native/LinearAlgebra.cpp 2281 aten/src/ATen/native/LinearAlgebra.h 20 aten/src/ATen/native/LinearAlgebraUtils.h 454 aten/src/ATen/native/Loss.cpp 449 aten/src/ATen/native/LossCTC.cpp 318 aten/src/ATen/native/LossMulti.h 65 aten/src/ATen/native/LossMultiLabelMargin.cpp 276 aten/src/ATen/native/LossMultiMargin.cpp 290 aten/src/ATen/native/LossNLL.cpp 578 aten/src/ATen/native/LossNLL2d.cpp 397 aten/src/ATen/native/Math.h 1620 aten/src/ATen/native/MathBitFallThroughLists.h 67 aten/src/ATen/native/MathBitsFallback.h 96 aten/src/ATen/native/MaxPooling.cpp 97 aten/src/ATen/native/MaxPooling.h 31 aten/src/ATen/native/MaxUnpooling.cpp 271 aten/src/ATen/native/Memory.cpp 18 aten/src/ATen/native/MetaTensor.cpp 35 aten/src/ATen/native/NNPACK.cpp 238 aten/src/ATen/native/NaiveConvolutionTranspose2d.cpp 729 aten/src/ATen/native/NaiveConvolutionTranspose3d.cpp 828 aten/src/ATen/native/NaiveDilatedConvolution.cpp 466 aten/src/ATen/native/NamedTensor.cpp 296 aten/src/ATen/native/NegateFallback.cpp 35 aten/src/ATen/native/NonEmptyUtils.h 21 aten/src/ATen/native/Normalization.cpp 569 aten/src/ATen/native/Normalization.h 8 aten/src/ATen/native/Onehot.cpp 32 aten/src/ATen/native/PackedSequence.cpp 164 aten/src/ATen/native/PixelShuffle.cpp 80 aten/src/ATen/native/PointwiseOps.cpp 57 aten/src/ATen/native/PointwiseOps.h 19 aten/src/ATen/native/Pool.h 257 aten/src/ATen/native/Pooling.cpp 132 aten/src/ATen/native/Pow.cpp 100 aten/src/ATen/native/Pow.h 53 aten/src/ATen/native/QuantizedLinear.cpp 374 aten/src/ATen/native/RNN.cpp 1712 aten/src/ATen/native/RNN.h 44 aten/src/ATen/native/RangeFactories.cpp 185 aten/src/ATen/native/RangeFactories.h 8 aten/src/ATen/native/ReduceAllOps.cpp 27 aten/src/ATen/native/ReduceAllOps.h 11 aten/src/ATen/native/ReduceOps.cpp 1528 aten/src/ATen/native/ReduceOps.h 41 aten/src/ATen/native/ReduceOpsUtils.h 327 aten/src/ATen/native/ReflectionPad.cpp 990 aten/src/ATen/native/Repeat.cpp 72 aten/src/ATen/native/Repeat.h 37 aten/src/ATen/native/ReplicationPadding.cpp 949 aten/src/ATen/native/Resize.cpp 93 aten/src/ATen/native/Resize.h 122 aten/src/ATen/native/ResizeCommon.h 42 aten/src/ATen/native/RowwisePrune.cpp 67 aten/src/ATen/native/Scalar.cpp 27 aten/src/ATen/native/ScatterGatherChecks.h 98 aten/src/ATen/native/SegmentReduce.cpp 288 aten/src/ATen/native/SegmentReduce.h 24 aten/src/ATen/native/SharedReduceOps.h 417 aten/src/ATen/native/SobolEngineOps.cpp 122 aten/src/ATen/native/SobolEngineOpsUtils.cpp 42413 aten/src/ATen/native/SobolEngineOpsUtils.h 32 aten/src/ATen/native/SoftMax.cpp 409 aten/src/ATen/native/Sorting.cpp 751 aten/src/ATen/native/Sorting.h 18 aten/src/ATen/native/SortingUtils.h 156 aten/src/ATen/native/SpectralOps.cpp 964 aten/src/ATen/native/SpectralOpsUtils.h 37 aten/src/ATen/native/StridedRandomAccessor.h 223 aten/src/ATen/native/SummaryOps.cpp 64 aten/src/ATen/native/TensorAdvancedIndexing.cpp 1443 aten/src/ATen/native/TensorAdvancedIndexing.h 26 aten/src/ATen/native/TensorCompare.cpp 557 aten/src/ATen/native/TensorCompare.h 37 aten/src/ATen/native/TensorConversions.cpp 301 aten/src/ATen/native/TensorConversions.h 18 aten/src/ATen/native/TensorDimApply.h 49 aten/src/ATen/native/TensorFactories.cpp 1206 aten/src/ATen/native/TensorFactories.h 55 aten/src/ATen/native/TensorIterator.h 2 aten/src/ATen/native/TensorIteratorDynamicCasting.h 34 aten/src/ATen/native/TensorIteratorReduce.cpp 133 aten/src/ATen/native/TensorProperties.cpp 65 aten/src/ATen/native/TensorShape.cpp 2064 aten/src/ATen/native/TensorShape.h 26 aten/src/ATen/native/TensorTransformations.cpp 162 aten/src/ATen/native/TensorTransformations.h 21 aten/src/ATen/native/TestOps.cpp 57 aten/src/ATen/native/TriangularOps.cpp 155 aten/src/ATen/native/TriangularOpsUtils.h 40 aten/src/ATen/native/TypeProperties.cpp 145 aten/src/ATen/native/TypeProperties.h 12 aten/src/ATen/native/UnaryOps.cpp 667 aten/src/ATen/native/UnaryOps.h 92 aten/src/ATen/native/Unfold2d.cpp 5 aten/src/ATen/native/Unfold2d.h 23 aten/src/ATen/native/Unfold3d.cpp 499 aten/src/ATen/native/Unfold3d.h 46 aten/src/ATen/native/UnfoldBackward.cpp 20 aten/src/ATen/native/UnfoldBackward.h 113 aten/src/ATen/native/Unique.cpp 253 aten/src/ATen/native/UpSample.cpp 29 aten/src/ATen/native/UpSample.h 318 aten/src/ATen/native/UpSampleBicubic2d.cpp 265 aten/src/ATen/native/UpSampleBilinear2d.cpp 164 aten/src/ATen/native/UpSampleLinear1d.cpp 82 aten/src/ATen/native/UpSampleNearest1d.cpp 123 aten/src/ATen/native/UpSampleNearest2d.cpp 153 aten/src/ATen/native/UpSampleNearest3d.cpp 170 aten/src/ATen/native/UpSampleTrilinear3d.cpp 98 aten/src/ATen/native/VariableMethodStubs.cpp 43 aten/src/ATen/native/WeightNorm.cpp 75 aten/src/ATen/native/ao_sparse/library.cpp 22 aten/src/ATen/native/ao_sparse/quantized/cpu/fbgemm_utils.cpp 68 aten/src/ATen/native/ao_sparse/quantized/cpu/fbgemm_utils.h 74 aten/src/ATen/native/ao_sparse/quantized/cpu/packed_params.h 35 aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear.cpp 180 aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_dynamic.cpp 154 aten/src/ATen/native/ao_sparse/quantized/cpu/qlinear_prepack.cpp 207 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/batch_norm.h 27 aten/src/ATen/native/cpu/Activation.cpp 601 aten/src/ATen/native/cpu/AdaptiveAvgPoolKernel.cpp 337 aten/src/ATen/native/cpu/AdaptiveMaxPoolKernel.cpp 392 aten/src/ATen/native/cpu/AtomicAddFloat.h 26 aten/src/ATen/native/cpu/AvgPoolKernel.cpp 456 aten/src/ATen/native/cpu/BinaryOpsKernel.cpp 1090 aten/src/ATen/native/cpu/BlasKernel.cpp 195 aten/src/ATen/native/cpu/CatKernel.cpp 56 aten/src/ATen/native/cpu/CatKernel.h 7 aten/src/ATen/native/cpu/ChannelShuffleKernel.cpp 89 aten/src/ATen/native/cpu/ChannelShuffleKernel.h 8 aten/src/ATen/native/cpu/ComplexKernel.cpp 26 aten/src/ATen/native/cpu/CopyKernel.cpp 84 aten/src/ATen/native/cpu/CrossKernel.cpp 67 aten/src/ATen/native/cpu/DepthwiseConvKernel.cpp 263 aten/src/ATen/native/cpu/DepthwiseConvKernel.h 10 aten/src/ATen/native/cpu/DistanceOpsKernel.cpp 333 aten/src/ATen/native/cpu/DistributionKernels.cpp 129 aten/src/ATen/native/cpu/DistributionTemplates.h 307 aten/src/ATen/native/cpu/FillKernel.cpp 50 aten/src/ATen/native/cpu/FunctionOfAMatrixUtilsKernel.cpp 44 aten/src/ATen/native/cpu/GridSamplerKernel.cpp 880 aten/src/ATen/native/cpu/GridSamplerKernel.h 13 aten/src/ATen/native/cpu/HistogramKernel.cpp 152 aten/src/ATen/native/cpu/IndexKernel.cpp 510 aten/src/ATen/native/cpu/Intrinsics.h 23 aten/src/ATen/native/cpu/IsContiguous.h 45 aten/src/ATen/native/cpu/LerpKernel.cpp 38 aten/src/ATen/native/cpu/LinearAlgebraKernel.cpp 138 aten/src/ATen/native/cpu/Loops.h 293 aten/src/ATen/native/cpu/MaxPoolKernel.cpp 414 aten/src/ATen/native/cpu/MaxPooling.cpp 49 aten/src/ATen/native/cpu/MaxUnpoolKernel.cpp 293 aten/src/ATen/native/cpu/MaxUnpoolKernel.h 11 aten/src/ATen/native/cpu/MultinomialKernel.cpp 91 aten/src/ATen/native/cpu/PointwiseOpsKernel.cpp 179 aten/src/ATen/native/cpu/PowKernel.cpp 127 aten/src/ATen/native/cpu/RangeFactoriesKernel.cpp 75 aten/src/ATen/native/cpu/Reduce.h 228 aten/src/ATen/native/cpu/ReduceAllOpsKernel.cpp 199 aten/src/ATen/native/cpu/ReduceOpsKernel.cpp 361 aten/src/ATen/native/cpu/RenormKernel.cpp 31 aten/src/ATen/native/cpu/ScatterGatherKernel.cpp 270 aten/src/ATen/native/cpu/SerialStackImpl.h 98 aten/src/ATen/native/cpu/SoftMaxKernel.cpp 433 aten/src/ATen/native/cpu/SoftmaxKernel.h 15 aten/src/ATen/native/cpu/SortingKernel.cpp 169 aten/src/ATen/native/cpu/StackKernel.cpp 17 aten/src/ATen/native/cpu/StackKernel.h 7 aten/src/ATen/native/cpu/SumKernel.cpp 489 aten/src/ATen/native/cpu/TensorCompareKernel.cpp 365 aten/src/ATen/native/cpu/UnaryOpsKernel.cpp 580 aten/src/ATen/native/cpu/Unfold2d.cpp 281 aten/src/ATen/native/cpu/UnfoldBackwardKernel.cpp 115 aten/src/ATen/native/cpu/UpSampleKernel.cpp 1349 aten/src/ATen/native/cpu/UpSampleMoreKernel.cpp 165 aten/src/ATen/native/cpu/avx_mathfun.h 312 aten/src/ATen/native/cpu/batch_norm_kernel.cpp 546 aten/src/ATen/native/cpu/group_norm_kernel.cpp 439 aten/src/ATen/native/cpu/layer_norm_kernel.cpp 255 aten/src/ATen/native/cpu/moments_utils.h 136 aten/src/ATen/native/cpu/utils.h 57 aten/src/ATen/native/cpu/zmath.h 196 aten/src/ATen/native/cuda/AbsKernel.cu 20 aten/src/ATen/native/cuda/Activation.cpp 122 aten/src/ATen/native/cuda/Activation.cu 548 aten/src/ATen/native/cuda/Activation.h 21 aten/src/ATen/native/cuda/AdaptiveAveragePooling.cu 686 aten/src/ATen/native/cuda/AdaptiveAveragePooling3d.cu 455 aten/src/ATen/native/cuda/AdaptiveMaxPooling2d.cu 395 aten/src/ATen/native/cuda/AdaptiveMaxPooling3d.cu 418 aten/src/ATen/native/cuda/AmpKernels.cu 221 aten/src/ATen/native/cuda/AveragePool2d.cu 404 aten/src/ATen/native/cuda/AveragePool3d.cu 526 aten/src/ATen/native/cuda/BatchLinearAlgebra.cpp 2525 aten/src/ATen/native/cuda/BatchLinearAlgebraLib.cpp 1096 aten/src/ATen/native/cuda/BatchLinearAlgebraLib.h 41 aten/src/ATen/native/cuda/BinaryAddSubKernel.cu 30 aten/src/ATen/native/cuda/BinaryBitwiseOpsKernels.cu 67 aten/src/ATen/native/cuda/BinaryGeometricKernels.cu 32 aten/src/ATen/native/cuda/BinaryLogicalOpsKernels.cu 37 aten/src/ATen/native/cuda/BinaryMiscBackwardOpsKernels.cu 78 aten/src/ATen/native/cuda/BinaryMiscOpsKernels.cu 71 aten/src/ATen/native/cuda/BinaryMulDivKernel.cu 168 aten/src/ATen/native/cuda/BinaryRemainderKernel.cu 54 aten/src/ATen/native/cuda/BinaryShiftOpsKernels.cu 54 aten/src/ATen/native/cuda/Blas.cpp 407 aten/src/ATen/native/cuda/Bucketization.cu 191 aten/src/ATen/native/cuda/CUDALoops.cuh 406 aten/src/ATen/native/cuda/CUDAScalar.cu 17 aten/src/ATen/native/cuda/Col2Im.cu 175 aten/src/ATen/native/cuda/CompareEQKernel.cu 36 aten/src/ATen/native/cuda/CompareKernels.cu 52 aten/src/ATen/native/cuda/ComplexKernel.cu 30 aten/src/ATen/native/cuda/CompositeRandomAccessor.h 27 aten/src/ATen/native/cuda/ConvolutionMM2d.cu 424 aten/src/ATen/native/cuda/Copy.cu 207 aten/src/ATen/native/cuda/CopysignKernel.cu 27 aten/src/ATen/native/cuda/CrossKernel.cu 75 aten/src/ATen/native/cuda/CuFFTPlanCache.h 379 aten/src/ATen/native/cuda/CuFFTUtils.h 65 aten/src/ATen/native/cuda/DepthwiseConv2d.cu 558 aten/src/ATen/native/cuda/DepthwiseConv3d.cu 606 aten/src/ATen/native/cuda/DeviceSqrt.cuh 22 aten/src/ATen/native/cuda/DilatedMaxPool2d.cu 489 aten/src/ATen/native/cuda/DilatedMaxPool3d.cu 455 aten/src/ATen/native/cuda/DistanceKernel.cu 339 aten/src/ATen/native/cuda/DistributionBernoulli.cu 33 aten/src/ATen/native/cuda/DistributionCauchyKernel.cu 27 aten/src/ATen/native/cuda/DistributionExponentialKernel.cu 27 aten/src/ATen/native/cuda/DistributionGeometricKernel.cu 27 aten/src/ATen/native/cuda/DistributionLogNormalKernel.cu 27 aten/src/ATen/native/cuda/DistributionNormal.cu 27 aten/src/ATen/native/cuda/DistributionRandomKernel.cu 37 aten/src/ATen/native/cuda/DistributionTemplates.h 538 aten/src/ATen/native/cuda/DistributionUniform.cu 13 aten/src/ATen/native/cuda/Distributions.cu 234 aten/src/ATen/native/cuda/Dropout.cu 369 aten/src/ATen/native/cuda/Embedding.cu 289 aten/src/ATen/native/cuda/EmbeddingBackwardKernel.cu 274 aten/src/ATen/native/cuda/EmbeddingBackwardKernel.cuh 20 aten/src/ATen/native/cuda/EmbeddingBag.cu 444 aten/src/ATen/native/cuda/Equal.cpp 21 aten/src/ATen/native/cuda/FillKernel.cu 23 aten/src/ATen/native/cuda/ForeachBinaryOpList.cu 84 aten/src/ATen/native/cuda/ForeachBinaryOpScalar.cu 83 aten/src/ATen/native/cuda/ForeachBinaryOpScalarList.cu 86 aten/src/ATen/native/cuda/ForeachFunctors.cuh 381 aten/src/ATen/native/cuda/ForeachPointwiseOp.cu 172 aten/src/ATen/native/cuda/ForeachReduceOp.cu 163 aten/src/ATen/native/cuda/ForeachUnaryOp.cu 246 aten/src/ATen/native/cuda/FractionalMaxPool2d.cu 222 aten/src/ATen/native/cuda/FractionalMaxPool3d.cu 310 aten/src/ATen/native/cuda/FunctionOfAMatrixUtilsKernel.cu 93 aten/src/ATen/native/cuda/GcdLcmKernel.cu 51 aten/src/ATen/native/cuda/GridSampler.cu 835 aten/src/ATen/native/cuda/GridSampler.cuh 294 aten/src/ATen/native/cuda/IGammaKernel.cu 496 aten/src/ATen/native/cuda/Im2Col.cu 191 aten/src/ATen/native/cuda/IndexKernel.cpp 57 aten/src/ATen/native/cuda/IndexKernel.cu 388 aten/src/ATen/native/cuda/IndexKernel.h 13 aten/src/ATen/native/cuda/Indexing.cu 859 aten/src/ATen/native/cuda/KernelUtils.cuh 84 aten/src/ATen/native/cuda/LaunchUtils.h 14 aten/src/ATen/native/cuda/LegacyThrustHelpers.cu 87 aten/src/ATen/native/cuda/Lerp.cu 47 aten/src/ATen/native/cuda/LinearAlgebra.cu 131 aten/src/ATen/native/cuda/LogAddExpKernel.cu 47 aten/src/ATen/native/cuda/Loops.cuh 389 aten/src/ATen/native/cuda/Loss.cu 567 aten/src/ATen/native/cuda/LossCTC.cu 702 aten/src/ATen/native/cuda/Math.cuh 1509 aten/src/ATen/native/cuda/MaxMinElementwiseKernel.cu 87 aten/src/ATen/native/cuda/MaxUnpooling.cu 525 aten/src/ATen/native/cuda/MemoryAccess.cuh 319 aten/src/ATen/native/cuda/MiscUtils.h 84 aten/src/ATen/native/cuda/MultiLabelMarginCriterion.cu 394 aten/src/ATen/native/cuda/MultiMarginLoss.cu 343 aten/src/ATen/native/cuda/MultiTensorApply.cuh 164 aten/src/ATen/native/cuda/MultinomialKernel.cu 376 aten/src/ATen/native/cuda/NLLLoss2d.cu 468 aten/src/ATen/native/cuda/NaiveConvolutionTranspose2d.cu 724 aten/src/ATen/native/cuda/NaiveConvolutionTranspose3d.cu 897 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/Normalization.cuh 1556 aten/src/ATen/native/cuda/PersistentSoftmax.cuh 345 aten/src/ATen/native/cuda/PointwiseOpsKernel.cu 77 aten/src/ATen/native/cuda/PowKernel.cu 148 aten/src/ATen/native/cuda/RNN.cu 533 aten/src/ATen/native/cuda/ROCmLoops.cuh 317 aten/src/ATen/native/cuda/Randperm.cu 103 aten/src/ATen/native/cuda/Randperm.cuh 50 aten/src/ATen/native/cuda/RangeFactories.cu 235 aten/src/ATen/native/cuda/RecordStream.cu 7 aten/src/ATen/native/cuda/Reduce.cu 49 aten/src/ATen/native/cuda/Reduce.cuh 984 aten/src/ATen/native/cuda/ReduceLogicKernel.cu 33 aten/src/ATen/native/cuda/ReduceMinMaxKernel.cu 146 aten/src/ATen/native/cuda/ReduceMomentKernel.cu 66 aten/src/ATen/native/cuda/ReduceNormKernel.cu 47 aten/src/ATen/native/cuda/ReduceOps.cpp 72 aten/src/ATen/native/cuda/ReduceOps.h 15 aten/src/ATen/native/cuda/ReduceSumProdKernel.cu 101 aten/src/ATen/native/cuda/ReflectionPad.cu 554 aten/src/ATen/native/cuda/RenormKernel.cu 23 aten/src/ATen/native/cuda/Repeat.cu 55 aten/src/ATen/native/cuda/ReplicationPadding.cu 638 aten/src/ATen/native/cuda/Resize.cpp 55 aten/src/ATen/native/cuda/Resize.h 44 aten/src/ATen/native/cuda/RreluWithNoise.cu 157 aten/src/ATen/native/cuda/ScanKernels.cpp 83 aten/src/ATen/native/cuda/ScanKernels.cu 460 aten/src/ATen/native/cuda/ScanKernels.h 12 aten/src/ATen/native/cuda/ScatterGatherKernel.cu 370 aten/src/ATen/native/cuda/SegmentReduce.cu 369 aten/src/ATen/native/cuda/Shape.cu 539 aten/src/ATen/native/cuda/SoftMax.cu 880 aten/src/ATen/native/cuda/Sort.cpp 141 aten/src/ATen/native/cuda/Sort.cu 309 aten/src/ATen/native/cuda/Sort.h 14 aten/src/ATen/native/cuda/SortImpl.cu 32 aten/src/ATen/native/cuda/SortUtils.cuh 136 aten/src/ATen/native/cuda/Sorting.cpp 145 aten/src/ATen/native/cuda/Sorting.cu 247 aten/src/ATen/native/cuda/Sorting.h 14 aten/src/ATen/native/cuda/SortingCommon.cuh 166 aten/src/ATen/native/cuda/SortingRadixSelect.cuh 362 aten/src/ATen/native/cuda/SparseMM.cu 14 aten/src/ATen/native/cuda/SpectralOps.cpp 376 aten/src/ATen/native/cuda/SpectralOps.cu 115 aten/src/ATen/native/cuda/StepKernel.cu 27 aten/src/ATen/native/cuda/SummaryOps.cu 398 aten/src/ATen/native/cuda/TensorCompare.cpp 14 aten/src/ATen/native/cuda/TensorCompare.cu 142 aten/src/ATen/native/cuda/TensorFactories.cu 317 aten/src/ATen/native/cuda/TensorModeKernel.cpp 66 aten/src/ATen/native/cuda/TensorModeKernel.cu 200 aten/src/ATen/native/cuda/TensorModeKernel.cuh 383 aten/src/ATen/native/cuda/TensorModeKernel.h 14 aten/src/ATen/native/cuda/TensorShapeCUDA.cpp 27 aten/src/ATen/native/cuda/TensorTopK.cpp 55 aten/src/ATen/native/cuda/TensorTopK.cu 258 aten/src/ATen/native/cuda/TensorTopK.h 12 aten/src/ATen/native/cuda/TensorTransformations.cu 131 aten/src/ATen/native/cuda/TriangularOps.cu 207 aten/src/ATen/native/cuda/UnaryComplexKernels.cu 83 aten/src/ATen/native/cuda/UnaryFractionKernels.cu 153 aten/src/ATen/native/cuda/UnaryGammaKernels.cu 100 aten/src/ATen/native/cuda/UnaryGeometricKernels.cu 143 aten/src/ATen/native/cuda/UnaryLogKernels.cu 43 aten/src/ATen/native/cuda/UnaryOpsKernel.cu 162 aten/src/ATen/native/cuda/UnarySignKernels.cu 66 aten/src/ATen/native/cuda/UnarySpecialOpsKernel.cu 293 aten/src/ATen/native/cuda/UnfoldBackwardKernel.cu 158 aten/src/ATen/native/cuda/Unique.cu 195 aten/src/ATen/native/cuda/UniqueCub.cu 167 aten/src/ATen/native/cuda/UniqueCub.cuh 13 aten/src/ATen/native/cuda/UpSample.cuh 231 aten/src/ATen/native/cuda/UpSampleBicubic2d.cu 250 aten/src/ATen/native/cuda/UpSampleBilinear2d.cu 430 aten/src/ATen/native/cuda/UpSampleLinear1d.cu 196 aten/src/ATen/native/cuda/UpSampleNearest1d.cu 188 aten/src/ATen/native/cuda/UpSampleNearest2d.cu 403 aten/src/ATen/native/cuda/UpSampleNearest3d.cu 319 aten/src/ATen/native/cuda/UpSampleTrilinear3d.cu 360 aten/src/ATen/native/cuda/WeightNorm.cu 435 aten/src/ATen/native/cuda/ZetaKernel.cu 33 aten/src/ATen/native/cuda/attention.cu 218 aten/src/ATen/native/cuda/block_reduce.cuh 77 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/jit_utils.h 53 aten/src/ATen/native/cuda/layer_norm_kernel.cu 884 aten/src/ATen/native/cuda/vol2col.cuh 255 aten/src/ATen/native/cudnn/AffineGridGenerator.cpp 75 aten/src/ATen/native/cudnn/BatchNorm.cpp 294 aten/src/ATen/native/cudnn/ConvPlaceholders.cpp 97 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/LossCTC.cpp 108 aten/src/ATen/native/cudnn/Macros.h 7 aten/src/ATen/native/cudnn/RNN.cpp 1255 aten/src/ATen/native/cudnn/RNNUtils.h 26 aten/src/ATen/native/group_norm.cpp 197 aten/src/ATen/native/group_norm.h 34 aten/src/ATen/native/im2col.h 82 aten/src/ATen/native/im2col_shape_check.h 203 aten/src/ATen/native/layer_norm.cpp 199 aten/src/ATen/native/layer_norm.h 90 aten/src/ATen/native/metal/MetalAten.mm 120 aten/src/ATen/native/metal/MetalCommandBuffer.h 17 aten/src/ATen/native/metal/MetalCommandBuffer.mm 89 aten/src/ATen/native/metal/MetalContext.h 17 aten/src/ATen/native/metal/MetalContext.mm 156 aten/src/ATen/native/metal/MetalConvParams.h 43 aten/src/ATen/native/metal/MetalConvParams.mm 33 aten/src/ATen/native/metal/MetalDevice.h 42 aten/src/ATen/native/metal/MetalGuardImpl.cpp 50 aten/src/ATen/native/metal/MetalNeuronType.h 54 aten/src/ATen/native/metal/MetalPrepackOpContext.h 161 aten/src/ATen/native/metal/MetalPrepackOpRegister.cpp 122 aten/src/ATen/native/metal/MetalShaders.h 773 aten/src/ATen/native/metal/MetalTensorImpl.h 44 aten/src/ATen/native/metal/MetalTensorImplStorage.h 38 aten/src/ATen/native/metal/MetalTensorImplStorage.mm 105 aten/src/ATen/native/metal/MetalTensorUtils.h 65 aten/src/ATen/native/metal/MetalTensorUtils.mm 39 aten/src/ATen/native/metal/mpscnn/MPSCNNClampOp.h 3 aten/src/ATen/native/metal/mpscnn/MPSCNNClampOp.mm 40 aten/src/ATen/native/metal/mpscnn/MPSCNNConvOp.h 20 aten/src/ATen/native/metal/mpscnn/MPSCNNConvOp.mm 140 aten/src/ATen/native/metal/mpscnn/MPSCNNFullyConnectedOp.h 12 aten/src/ATen/native/metal/mpscnn/MPSCNNFullyConnectedOp.mm 52 aten/src/ATen/native/metal/mpscnn/MPSCNNNeuronOp.h 7 aten/src/ATen/native/metal/mpscnn/MPSCNNNeuronOp.mm 45 aten/src/ATen/native/metal/mpscnn/MPSCNNOp.h 13 aten/src/ATen/native/metal/mpscnn/MPSCNNUtils.h 42 aten/src/ATen/native/metal/mpscnn/MPSCNNUtils.mm 36 aten/src/ATen/native/metal/mpscnn/MPSImage+Tensor.h 9 aten/src/ATen/native/metal/mpscnn/MPSImage+Tensor.mm 36 aten/src/ATen/native/metal/mpscnn/MPSImageUtils.h 55 aten/src/ATen/native/metal/mpscnn/MPSImageUtils.mm 216 aten/src/ATen/native/metal/mpscnn/MPSImageWrapper.h 37 aten/src/ATen/native/metal/mpscnn/MPSImageWrapper.mm 121 aten/src/ATen/native/metal/ops/MetalAddmm.mm 127 aten/src/ATen/native/metal/ops/MetalBinaryElementwise.mm 291 aten/src/ATen/native/metal/ops/MetalChunk.mm 62 aten/src/ATen/native/metal/ops/MetalClamp.mm 65 aten/src/ATen/native/metal/ops/MetalConcat.mm 189 aten/src/ATen/native/metal/ops/MetalConvolution.h 20 aten/src/ATen/native/metal/ops/MetalConvolution.mm 106 aten/src/ATen/native/metal/ops/MetalCopy.h 11 aten/src/ATen/native/metal/ops/MetalCopy.mm 51 aten/src/ATen/native/metal/ops/MetalHardswish.mm 78 aten/src/ATen/native/metal/ops/MetalNeurons.mm 84 aten/src/ATen/native/metal/ops/MetalPadding.mm 80 aten/src/ATen/native/metal/ops/MetalPooling.mm 105 aten/src/ATen/native/metal/ops/MetalReduce.mm 76 aten/src/ATen/native/metal/ops/MetalReshape.mm 105 aten/src/ATen/native/metal/ops/MetalSoftmax.mm 63 aten/src/ATen/native/metal/ops/MetalTranspose.mm 92 aten/src/ATen/native/metal/ops/MetalUpsamplingNearest.mm 89 aten/src/ATen/native/miopen/BatchNorm_miopen.cpp 180 aten/src/ATen/native/miopen/Conv_miopen.cpp 948 aten/src/ATen/native/miopen/RNN_miopen.cpp 764 aten/src/ATen/native/mkl/LinearAlgebra.cpp 91 aten/src/ATen/native/mkl/LinearAlgebra.h 24 aten/src/ATen/native/mkl/SparseBlasImpl.cpp 363 aten/src/ATen/native/mkl/SparseBlasImpl.h 36 aten/src/ATen/native/mkl/SparseCsrLinearAlgebra.cpp 240 aten/src/ATen/native/mkl/SparseCsrLinearAlgebra.h 13 aten/src/ATen/native/mkl/SpectralOps.cpp 466 aten/src/ATen/native/mkldnn/BinaryOps.cpp 123 aten/src/ATen/native/mkldnn/Conv.cpp 193 aten/src/ATen/native/mkldnn/Copy.cpp 33 aten/src/ATen/native/mkldnn/Gelu.cpp 40 aten/src/ATen/native/mkldnn/IDeepRegistration.cpp 20 aten/src/ATen/native/mkldnn/Linear.cpp 133 aten/src/ATen/native/mkldnn/MKLDNNCommon.cpp 89 aten/src/ATen/native/mkldnn/MKLDNNCommon.h 13 aten/src/ATen/native/mkldnn/MKLDNNConversions.cpp 141 aten/src/ATen/native/mkldnn/Matmul.cpp 105 aten/src/ATen/native/mkldnn/Matmul.h 16 aten/src/ATen/native/mkldnn/MkldnnTensorMath.cpp 35 aten/src/ATen/native/mkldnn/Normalization.cpp 167 aten/src/ATen/native/mkldnn/Pooling.cpp 581 aten/src/ATen/native/mkldnn/Relu.cpp 53 aten/src/ATen/native/mkldnn/SoftMax.cpp 35 aten/src/ATen/native/mkldnn/TensorFactories.cpp 18 aten/src/ATen/native/mkldnn/TensorShape.cpp 70 aten/src/ATen/native/mkldnn/TensorShape.h 8 aten/src/ATen/native/mkldnn/UnaryOps.cpp 55 aten/src/ATen/native/mkldnn/Utils.cpp 29 aten/src/ATen/native/mkldnn/Utils.h 25 aten/src/ATen/native/native_functions.yaml 8466 aten/src/ATen/native/prim_native_functions.cpp 23 aten/src/ATen/native/quantized/Copy.cpp 31 aten/src/ATen/native/quantized/Copy.h 7 aten/src/ATen/native/quantized/QTensor.cpp 304 aten/src/ATen/native/quantized/TensorCompare.cpp 38 aten/src/ATen/native/quantized/TensorFactories.cpp 119 aten/src/ATen/native/quantized/affine_quantizer.cpp 263 aten/src/ATen/native/quantized/affine_quantizer.h 110 aten/src/ATen/native/quantized/affine_quantizer_base.cpp 210 aten/src/ATen/native/quantized/affine_quantizer_base.h 35 aten/src/ATen/native/quantized/cpu/conv_packed_params.h 24 aten/src/ATen/native/quantized/cpu/conv_serialization.h 258 aten/src/ATen/native/quantized/cpu/embedding_packed_params.h 24 aten/src/ATen/native/quantized/cpu/fbgemm_utils.cpp 484 aten/src/ATen/native/quantized/cpu/fbgemm_utils.h 325 aten/src/ATen/native/quantized/cpu/fused_obs_fake_quant.cpp 226 aten/src/ATen/native/quantized/cpu/init_qnnpack.cpp 19 aten/src/ATen/native/quantized/cpu/init_qnnpack.h 8 aten/src/ATen/native/quantized/cpu/int_repr_quant.cpp 38 aten/src/ATen/native/quantized/cpu/kernels/QuantizedOpKernels.cpp 2972 aten/src/ATen/native/quantized/cpu/make_per_tensor_quantized_tensor.cpp 30 aten/src/ATen/native/quantized/cpu/packed_params.h 63 aten/src/ATen/native/quantized/cpu/q_adaavgpool.cpp 275 aten/src/ATen/native/quantized/cpu/q_avgpool.cpp 346 aten/src/ATen/native/quantized/cpu/q_avgpool3d.cpp 164 aten/src/ATen/native/quantized/cpu/qadd.cpp 248 aten/src/ATen/native/quantized/cpu/qadd.h 7 aten/src/ATen/native/quantized/cpu/qbatch_norm.cpp 350 aten/src/ATen/native/quantized/cpu/qchannel_shuffle.cpp 102 aten/src/ATen/native/quantized/cpu/qclamp.cpp 137 aten/src/ATen/native/quantized/cpu/qconcat.cpp 140 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/qdropout.cpp 16 aten/src/ATen/native/quantized/cpu/qelu.cpp 25 aten/src/ATen/native/quantized/cpu/qembeddingbag.cpp 939 aten/src/ATen/native/quantized/cpu/qembeddingbag.h 27 aten/src/ATen/native/quantized/cpu/qembeddingbag_prepack.cpp 341 aten/src/ATen/native/quantized/cpu/qembeddingbag_prepack.h 7 aten/src/ATen/native/quantized/cpu/qembeddingbag_unpack.cpp 181 aten/src/ATen/native/quantized/cpu/qgelu.cpp 19 aten/src/ATen/native/quantized/cpu/qhardsigmoid.cpp 85 aten/src/ATen/native/quantized/cpu/qhardswish.cpp 83 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/qlinear_unpack.cpp 124 aten/src/ATen/native/quantized/cpu/qmul.cpp 175 aten/src/ATen/native/quantized/cpu/qnnpack/bench/add.cc 138 aten/src/ATen/native/quantized/cpu/qnnpack/bench/average-pooling.cc 150 aten/src/ATen/native/quantized/cpu/qnnpack/bench/channel-shuffle.cc 147 aten/src/ATen/native/quantized/cpu/qnnpack/bench/convolution.cc 644 aten/src/ATen/native/quantized/cpu/qnnpack/bench/global-average-pooling.cc 74 aten/src/ATen/native/quantized/cpu/qnnpack/bench/hardsigmoid.cc 75 aten/src/ATen/native/quantized/cpu/qnnpack/bench/hardswish.cc 75 aten/src/ATen/native/quantized/cpu/qnnpack/bench/hgemm.cc 308 aten/src/ATen/native/quantized/cpu/qnnpack/bench/max-pooling.cc 119 aten/src/ATen/native/quantized/cpu/qnnpack/bench/q8gemm.cc 936 aten/src/ATen/native/quantized/cpu/qnnpack/bench/q8gemm_sparse.cc 576 aten/src/ATen/native/quantized/cpu/qnnpack/bench/requantization.cc 335 aten/src/ATen/native/quantized/cpu/qnnpack/bench/sgemm.cc 485 aten/src/ATen/native/quantized/cpu/qnnpack/bench/sigmoid.cc 75 aten/src/ATen/native/quantized/cpu/qnnpack/bench/softargmax.cc 72 aten/src/ATen/native/quantized/cpu/qnnpack/bench/tanh.cc 75 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadCpuinfo.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadFP16.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadFXdiv.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadGoogleBenchmark.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadGoogleTest.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadPSimd.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/cmake/DownloadPThreadPool.cmake 18 aten/src/ATen/native/quantized/cpu/qnnpack/configure.py 253 aten/src/ATen/native/quantized/cpu/qnnpack/confu.yaml 18 aten/src/ATen/native/quantized/cpu/qnnpack/generate-wrapper.py 116 aten/src/ATen/native/quantized/cpu/qnnpack/include/pack_block_sparse.h 33 aten/src/ATen/native/quantized/cpu/qnnpack/include/pytorch_qnnpack.h 392 aten/src/ATen/native/quantized/cpu/qnnpack/include/qnnpack_func.h 136 aten/src/ATen/native/quantized/cpu/qnnpack/src/add.c 130 aten/src/ATen/native/quantized/cpu/qnnpack/src/average-pooling.c 245 aten/src/ATen/native/quantized/cpu/qnnpack/src/channel-shuffle.c 77 aten/src/ATen/native/quantized/cpu/qnnpack/src/clamp.c 78 aten/src/ATen/native/quantized/cpu/qnnpack/src/conv-prepack.cc 270 aten/src/ATen/native/quantized/cpu/qnnpack/src/conv-run.cc 639 aten/src/ATen/native/quantized/cpu/qnnpack/src/convolution.c 738 aten/src/ATen/native/quantized/cpu/qnnpack/src/deconv-run.cc 189 aten/src/ATen/native/quantized/cpu/qnnpack/src/deconvolution.c 259 aten/src/ATen/native/quantized/cpu/qnnpack/src/fc-dynamic-run.cc 112 aten/src/ATen/native/quantized/cpu/qnnpack/src/fc-prepack.cc 52 aten/src/ATen/native/quantized/cpu/qnnpack/src/fc-run.cc 111 aten/src/ATen/native/quantized/cpu/qnnpack/src/fully-connected-sparse.c 113 aten/src/ATen/native/quantized/cpu/qnnpack/src/fully-connected.c 130 aten/src/ATen/native/quantized/cpu/qnnpack/src/global-average-pooling.c 128 aten/src/ATen/native/quantized/cpu/qnnpack/src/hardsigmoid.c 132 aten/src/ATen/native/quantized/cpu/qnnpack/src/hardswish.c 118 aten/src/ATen/native/quantized/cpu/qnnpack/src/hgemm/8x8-aarch32-neonfp16arith.S 426 aten/src/ATen/native/quantized/cpu/qnnpack/src/hgemm/8x8-neonfp16arith.c 339 aten/src/ATen/native/quantized/cpu/qnnpack/src/indirection.c 379 aten/src/ATen/native/quantized/cpu/qnnpack/src/init.c 354 aten/src/ATen/native/quantized/cpu/qnnpack/src/leaky-relu.c 137 aten/src/ATen/native/quantized/cpu/qnnpack/src/max-pooling.c 188 aten/src/ATen/native/quantized/cpu/qnnpack/src/operator-delete.c 16 aten/src/ATen/native/quantized/cpu/qnnpack/src/operator-run.c 1471 aten/src/ATen/native/quantized/cpu/qnnpack/src/pack_block_sparse.cc 90 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8avgpool/mp8x9p8q-neon.c 394 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8avgpool/mp8x9p8q-sse2.c 411 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8avgpool/up8x9-neon.c 210 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8avgpool/up8x9-sse2.c 204 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8avgpool/up8xm-neon.c 103 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8avgpool/up8xm-sse2.c 83 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/4x4c2-sse2.c 328 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/4x8-aarch32-neon.S 671 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/4x8-neon.c 663 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/8x8-aarch64-neon.S 663 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/8x8-neon.c 1175 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/mp8x25-neon-per-channel.c 859 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/mp8x25-neon.c 835 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/mp8x25-sse2-per-channel.c 1021 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/mp8x25-sse2.c 901 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/mp8x27-neon.c 935 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/mp8x27-sse2.c 95 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/up8x9-aarch32-neon-per-channel.S 343 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/up8x9-aarch32-neon.S 298 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/up8x9-neon-per-channel.c 894 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/up8x9-neon.c 874 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/up8x9-sse2-per-channel.c 398 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/up8x9-sse2.c 354 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gavgpool/mp8x7p7q-neon.c 272 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gavgpool/mp8x7p7q-sse2.c 284 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gavgpool/up8x7-neon.c 181 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gavgpool/up8x7-sse2.c 176 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gavgpool/up8xm-neon.c 100 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gavgpool/up8xm-sse2.c 80 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/2x4c8-sse2.c 196 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x-sumrows-neon.c 130 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x4c2-dq-sse2.c 289 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x4c2-sse2.c 315 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x8-aarch32-neon.S 687 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x8-dq-aarch32-neon.S 643 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x8-dq-neon.c 538 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x8-neon.c 651 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x8c2-xzp-aarch32-neon.S 499 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/4x8c2-xzp-neon.c 489 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/6x4-neon.c 524 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/8x8-aarch64-neon.S 673 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/8x8-dq-aarch64-neon.S 640 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/8x8-neon.c 1175 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/4x4-packA-aarch32-neon.S 204 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/4x8c1x4-dq-packedA-aarch32-neon.S 393 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/4x8c8x1-dq-packedA-aarch32-neon.S 306 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4-packA-aarch32-neon.S 244 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4-packA-aarch64-neon.S 216 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4-packA-sse2.c 26 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-dq-packedA-sse2.c 334 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-packed-sse2.h 4 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x8c1x4-dq-packedA-aarch64-neon.S 427 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x8c8x1-dq-packedA-aarch64-neon.S 353 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8vadd/neon.c 311 aten/src/ATen/native/quantized/cpu/qnnpack/src/q8vadd/sse2.c 176 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/AlignedAllocator.h 81 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/assembly.h 22 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/common.h 65 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/hgemm.h 23 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/indirection.h 27 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/isa-checks.h 20 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/log.h 23 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/math.h 21 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/operator.h 119 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/pack.h 705 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/params.h 563 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8avgpool.h 42 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8conv.h 28 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8dwconv.h 64 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8gavgpool.h 38 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8gemm.h 71 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8gemm_sparse.h 87 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/q8vadd.h 20 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/requantization-stubs.h 55 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/requantization.h 471 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/scalar-utils.h 80 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/sconv.h 23 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/sdwconv.h 33 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/sgemm.h 24 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/u8clamp.h 19 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/u8lut32norm.h 15 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/u8maxpool.h 25 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/u8rmax.h 15 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/x8lut.h 15 aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/x8zip.h 24 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/fp32-neon.c 78 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/fp32-psimd.c 55 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/fp32-scalar.c 99 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/fp32-sse2.c 44 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-neon.c 71 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-scalar.c 59 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-scalar.h 17 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse.h 75 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse2.c 53 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse4.c 53 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-ssse3.c 53 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/precise-neon.c 144 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/precise-psimd.c 137 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/precise-scalar.c 206 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/precise-sse2.c 122 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/precise-sse4.c 94 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/precise-ssse3.c 110 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/q31-neon.c 74 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/q31-scalar.c 83 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/q31-sse2.c 182 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/q31-sse4.c 114 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/q31-ssse3.c 178 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/runtime-assembly.h 17 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/runtime-neon.h 10 aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/runtime-sse2.h 10 aten/src/ATen/native/quantized/cpu/qnnpack/src/sconv/6x8-psimd.c 185 aten/src/ATen/native/quantized/cpu/qnnpack/src/sdwconv/up4x9-psimd.c 119 aten/src/ATen/native/quantized/cpu/qnnpack/src/sgemm/5x8-neon.c 247 aten/src/ATen/native/quantized/cpu/qnnpack/src/sgemm/6x8-neon.c 286 aten/src/ATen/native/quantized/cpu/qnnpack/src/sgemm/6x8-psimd.c 197 aten/src/ATen/native/quantized/cpu/qnnpack/src/sigmoid.c 127 aten/src/ATen/native/quantized/cpu/qnnpack/src/softargmax.c 109 aten/src/ATen/native/quantized/cpu/qnnpack/src/tanh.c 127 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8clamp/neon.c 69 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8clamp/sse2.c 65 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8lut32norm/scalar.c 34 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8maxpool/16x9p8q-neon.c 220 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8maxpool/16x9p8q-sse2.c 222 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8maxpool/sub16-neon.c 75 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8maxpool/sub16-sse2.c 71 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8rmax/neon.c 36 aten/src/ATen/native/quantized/cpu/qnnpack/src/u8rmax/sse2.c 35 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8lut/scalar.c 32 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/x2-neon.c 35 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/x2-sse2.c 41 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/x3-neon.c 40 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/x3-sse2.c 124 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/x4-neon.c 46 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/x4-sse2.c 71 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/xm-neon.c 146 aten/src/ATen/native/quantized/cpu/qnnpack/src/x8zip/xm-sse2.c 188 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/dummy.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/hgemm/8x8-aarch32-neonfp16arith.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8avgpool/mp8x9p8q-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8avgpool/mp8x9p8q-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8avgpool/up8x9-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8avgpool/up8x9-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8avgpool/up8xm-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8avgpool/up8xm-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8conv/4x4c2-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8conv/4x8-aarch32-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8conv/4x8-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8conv/8x8-aarch64-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8conv/8x8-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/mp8x25-neon-per-channel.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/mp8x25-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/mp8x25-sse2-per-channel.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/mp8x25-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/mp8x27-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/mp8x27-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/up8x9-aarch32-neon-per-channel.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/up8x9-aarch32-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/up8x9-neon-per-channel.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/up8x9-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/up8x9-sse2-per-channel.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8dwconv/up8x9-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gavgpool/mp8x7p7q-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gavgpool/mp8x7p7q-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gavgpool/up8x7-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gavgpool/up8x7-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gavgpool/up8xm-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gavgpool/up8xm-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/2x4c8-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x-sumrows-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x4c2-dq-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x4c2-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x8-aarch32-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x8-dq-aarch32-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x8-dq-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x8-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x8c2-xzp-aarch32-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/4x8c2-xzp-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/6x4-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/8x8-aarch64-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/8x8-dq-aarch64-neon.S 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm/8x8-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/4x4-packA-aarch32-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/4x8c1x4-dq-packedA-aarch32-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/4x8c8x1-dq-packedA-aarch32-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/8x4-packA-aarch32-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/8x4-packA-aarch64-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/8x4c1x4-packed-sse2.c 4 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/8x8c1x4-dq-packedA-aarch64-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8gemm_sparse/8x8c8x1-dq-packedA-aarch64-neon.S 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8vadd/neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/q8vadd/sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/fp32-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/fp32-psimd.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/fp32-scalar.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/fp32-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/gemmlowp-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/gemmlowp-scalar.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/gemmlowp-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/gemmlowp-sse4.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/gemmlowp-ssse3.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/precise-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/precise-psimd.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/precise-scalar.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/precise-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/precise-sse4.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/precise-ssse3.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/q31-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/q31-scalar.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/q31-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/q31-sse4.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/requantization/q31-ssse3.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/sgemm/5x8-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/sgemm/6x8-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/sgemm/6x8-psimd.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8clamp/neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8clamp/sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8lut32norm/scalar.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8maxpool/16x9p8q-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8maxpool/16x9p8q-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8maxpool/sub16-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8maxpool/sub16-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8rmax/neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/u8rmax/sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8lut/scalar.c 1 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/x2-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/x2-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/x3-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/x3-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/x4-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/x4-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/xm-neon.c 3 aten/src/ATen/native/quantized/cpu/qnnpack/wrappers/x8zip/xm-sse2.c 3 aten/src/ATen/native/quantized/cpu/qnnpack_utils.h 415 aten/src/ATen/native/quantized/cpu/qnormalization.cpp 141 aten/src/ATen/native/quantized/cpu/qpool.cpp 407 aten/src/ATen/native/quantized/cpu/qreduction.cpp 125 aten/src/ATen/native/quantized/cpu/qrelu.cpp 156 aten/src/ATen/native/quantized/cpu/qsigmoid.cpp 114 aten/src/ATen/native/quantized/cpu/qsort.cpp 48 aten/src/ATen/native/quantized/cpu/qtanh.cpp 85 aten/src/ATen/native/quantized/cpu/qthreshold.cpp 33 aten/src/ATen/native/quantized/cpu/quant_utils.h 143 aten/src/ATen/native/quantized/cpu/quantized_ops.h 189 aten/src/ATen/native/quantized/cpu/qupsample_bilinear2d.cpp 162 aten/src/ATen/native/quantized/cpu/qupsample_nearest2d.cpp 190 aten/src/ATen/native/quantized/cpu/qupsample_nearest3d.cpp 221 aten/src/ATen/native/quantized/cpu/tensor_operators.cpp 66 aten/src/ATen/native/quantized/cuda/affine_quantizer.cu 233 aten/src/ATen/native/quantized/cuda/embedding_bag.cu 506 aten/src/ATen/native/quantized/cuda/fake_quantize_core.cu 201 aten/src/ATen/native/quantized/cuda/fused_obs_fake_quant.cu 285 aten/src/ATen/native/quantized/cuda/int_repr_quant.cu 26 aten/src/ATen/native/quantized/cuda/make_quantized_tensor.cu 47 aten/src/ATen/native/quantized/fake_quant_affine.h 52 aten/src/ATen/native/quantized/fake_quant_per_channel_affine.cpp 177 aten/src/ATen/native/quantized/fake_quant_per_tensor_affine.cpp 146 aten/src/ATen/native/quantized/library.cpp 190 aten/src/ATen/native/sparse/ParamUtils.cpp 46 aten/src/ATen/native/sparse/ParamUtils.h 19 aten/src/ATen/native/sparse/SoftMax.cpp 364 aten/src/ATen/native/sparse/SparseBlas.cpp 66 aten/src/ATen/native/sparse/SparseBlasImpl.cpp 59 aten/src/ATen/native/sparse/SparseBlasImpl.h 31 aten/src/ATen/native/sparse/SparseCsrTensor.cpp 310 aten/src/ATen/native/sparse/SparseCsrTensorMath.cpp 486 aten/src/ATen/native/sparse/SparseMatMul.cpp 149 aten/src/ATen/native/sparse/SparseTensor.cpp 664 aten/src/ATen/native/sparse/SparseTensorMath.cpp 1140 aten/src/ATen/native/sparse/SparseTensorMath.h 7 aten/src/ATen/native/sparse/SparseUnaryOps.cpp 127 aten/src/ATen/native/sparse/cuda/SoftMax.cu 565 aten/src/ATen/native/sparse/cuda/SparseBlas.cpp 185 aten/src/ATen/native/sparse/cuda/SparseBlasImpl.cpp 1082 aten/src/ATen/native/sparse/cuda/SparseBlasImpl.h 45 aten/src/ATen/native/sparse/cuda/SparseBlasLegacy.cpp 62 aten/src/ATen/native/sparse/cuda/SparseBlasLegacy.h 8 aten/src/ATen/native/sparse/cuda/SparseCUDAApplyUtils.cuh 328 aten/src/ATen/native/sparse/cuda/SparseCUDABlas.cpp 430 aten/src/ATen/native/sparse/cuda/SparseCUDABlas.h 57 aten/src/ATen/native/sparse/cuda/SparseCUDATensor.cpp 53 aten/src/ATen/native/sparse/cuda/SparseCUDATensor.cu 247 aten/src/ATen/native/sparse/cuda/SparseCUDATensorMath.cu 820 aten/src/ATen/native/sparse/cuda/SparseCUDATensorMath.cuh 12 aten/src/ATen/native/sparse/cuda/SparseCsrTensorMath.cu 234 aten/src/ATen/native/sparse/cuda/SparseMatMul.cu 695 aten/src/ATen/native/utils/Factory.cpp 47 aten/src/ATen/native/utils/Factory.h 16 aten/src/ATen/native/utils/ParamUtils.h 23 aten/src/ATen/native/utils/ParamsHash.h 27 aten/src/ATen/native/vol2col.h 109 aten/src/ATen/native/vulkan/Vulkan.cpp 1237 aten/src/ATen/native/vulkan/Vulkan.h 407 aten/src/ATen/native/vulkan/VulkanAten.cpp 632 aten/src/ATen/native/vulkan/VulkanAten.h 20 aten/src/ATen/native/vulkan/VulkanCommon.h 77 aten/src/ATen/native/vulkan/VulkanConvolution.cpp 93 aten/src/ATen/native/vulkan/VulkanConvolution.h 115 aten/src/ATen/native/vulkan/VulkanGuardImpl.cpp 50 aten/src/ATen/native/vulkan/VulkanOpContext.cpp 40 aten/src/ATen/native/vulkan/VulkanOpContext.h 78 aten/src/ATen/native/vulkan/VulkanOpaqueTensorImpl.h 39 aten/src/ATen/native/vulkan/VulkanOps.cpp 1194 aten/src/ATen/native/vulkan/VulkanOps.h 132 aten/src/ATen/native/vulkan/VulkanRegisterOpContextClass.cpp 49 aten/src/ATen/native/vulkan/api/Adapter.h 27 aten/src/ATen/native/vulkan/api/Allocator.cpp 2 aten/src/ATen/native/vulkan/api/Allocator.h 32 aten/src/ATen/native/vulkan/api/Cache.h 51 aten/src/ATen/native/vulkan/api/Command.cpp 413 aten/src/ATen/native/vulkan/api/Command.h 102 aten/src/ATen/native/vulkan/api/Common.cpp 42 aten/src/ATen/native/vulkan/api/Common.h 154 aten/src/ATen/native/vulkan/api/Context.cpp 186 aten/src/ATen/native/vulkan/api/Context.h 131 aten/src/ATen/native/vulkan/api/Descriptor.cpp 343 aten/src/ATen/native/vulkan/api/Descriptor.h 81 aten/src/ATen/native/vulkan/api/Helper.cpp 41 aten/src/ATen/native/vulkan/api/Helper.h 22 aten/src/ATen/native/vulkan/api/Pipeline.cpp 149 aten/src/ATen/native/vulkan/api/Pipeline.h 144 aten/src/ATen/native/vulkan/api/Resource.cpp 578 aten/src/ATen/native/vulkan/api/Resource.h 295 aten/src/ATen/native/vulkan/api/Runtime.cpp 294 aten/src/ATen/native/vulkan/api/Runtime.h 44 aten/src/ATen/native/vulkan/api/Shader.cpp 142 aten/src/ATen/native/vulkan/api/Shader.h 190 aten/src/ATen/native/vulkan/api/ThreadContext.cpp 30 aten/src/ATen/native/vulkan/api/ThreadContext.h 54 aten/src/ATen/native/vulkan/api/Utils.h 81 aten/src/ATen/native/vulkan/api/api.h 13 aten/src/ATen/native/vulkan/api/vk_mem_alloc.h 12820 aten/src/ATen/native/vulkan/glsl/KO4C4HW_to_image.glsl 28 aten/src/ATen/native/vulkan/glsl/adaptive_avg_pool2d.glsl 32 aten/src/ATen/native/vulkan/glsl/add.glsl 26 aten/src/ATen/native/vulkan/glsl/add_.glsl 23 aten/src/ATen/native/vulkan/glsl/add_scalar.glsl 21 aten/src/ATen/native/vulkan/glsl/add_scalar_.glsl 20 aten/src/ATen/native/vulkan/glsl/addmm.glsl 48 aten/src/ATen/native/vulkan/glsl/avg_pool2d.glsl 32 aten/src/ATen/native/vulkan/glsl/cat_feature.glsl 38 aten/src/ATen/native/vulkan/glsl/clamp.glsl 21 aten/src/ATen/native/vulkan/glsl/clamp_.glsl 20 aten/src/ATen/native/vulkan/glsl/conv2d.glsl 47 aten/src/ATen/native/vulkan/glsl/conv2d_dw.glsl 41 aten/src/ATen/native/vulkan/glsl/conv2d_dw_clamp.glsl 53 aten/src/ATen/native/vulkan/glsl/conv2d_nogroup_clamp.glsl 77 aten/src/ATen/native/vulkan/glsl/conv2d_nogroup_clamp_1x.glsl 60 aten/src/ATen/native/vulkan/glsl/conv2d_pw.glsl 38 aten/src/ATen/native/vulkan/glsl/conv2d_pw_2x2.glsl 85 aten/src/ATen/native/vulkan/glsl/conv2d_pw_2x2_buffered.glsl 127 aten/src/ATen/native/vulkan/glsl/conv2d_winograd_2_3.glsl 72 aten/src/ATen/native/vulkan/glsl/conv_transpose2d.glsl 54 aten/src/ATen/native/vulkan/glsl/div.glsl 26 aten/src/ATen/native/vulkan/glsl/div_.glsl 23 aten/src/ATen/native/vulkan/glsl/hardshrink.glsl 21 aten/src/ATen/native/vulkan/glsl/hardshrink_.glsl 20 aten/src/ATen/native/vulkan/glsl/hardsigmoid.glsl 18 aten/src/ATen/native/vulkan/glsl/hardsigmoid_.glsl 17 aten/src/ATen/native/vulkan/glsl/hardswish.glsl 21 aten/src/ATen/native/vulkan/glsl/hardswish_.glsl 20 aten/src/ATen/native/vulkan/glsl/image_to_nchw.glsl 25 aten/src/ATen/native/vulkan/glsl/leaky_relu.glsl 23 aten/src/ATen/native/vulkan/glsl/leaky_relu_.glsl 22 aten/src/ATen/native/vulkan/glsl/log_softmax.glsl 38 aten/src/ATen/native/vulkan/glsl/max_pool2d.glsl 36 aten/src/ATen/native/vulkan/glsl/mean.glsl 42 aten/src/ATen/native/vulkan/glsl/mean2d.glsl 57 aten/src/ATen/native/vulkan/glsl/mm.glsl 50 aten/src/ATen/native/vulkan/glsl/mul.glsl 26 aten/src/ATen/native/vulkan/glsl/mul_.glsl 23 aten/src/ATen/native/vulkan/glsl/mul_scalar.glsl 21 aten/src/ATen/native/vulkan/glsl/mul_scalar_.glsl 20 aten/src/ATen/native/vulkan/glsl/nchw_to_image.glsl 29 aten/src/ATen/native/vulkan/glsl/permute.glsl 49 aten/src/ATen/native/vulkan/glsl/permute_4d.glsl 102 aten/src/ATen/native/vulkan/glsl/reflection_pad2d.glsl 22 aten/src/ATen/native/vulkan/glsl/sigmoid.glsl 17 aten/src/ATen/native/vulkan/glsl/sigmoid_.glsl 16 aten/src/ATen/native/vulkan/glsl/slice_4d.glsl 60 aten/src/ATen/native/vulkan/glsl/softmax.glsl 38 aten/src/ATen/native/vulkan/glsl/sub.glsl 26 aten/src/ATen/native/vulkan/glsl/sub_.glsl 23 aten/src/ATen/native/vulkan/glsl/tanh.glsl 17 aten/src/ATen/native/vulkan/glsl/tanh_.glsl 16 aten/src/ATen/native/vulkan/glsl/transform_winograd_2_3_sh.glsl 47 aten/src/ATen/native/vulkan/glsl/upsample_nearest2d.glsl 26 aten/src/ATen/native/vulkan/ops/Arithmetic.cpp 363 aten/src/ATen/native/vulkan/ops/Clamp.cpp 388 aten/src/ATen/native/vulkan/ops/Clone.cpp 37 aten/src/ATen/native/vulkan/ops/Common.cpp 57 aten/src/ATen/native/vulkan/ops/Common.h 43 aten/src/ATen/native/vulkan/ops/Concat.cpp 215 aten/src/ATen/native/vulkan/ops/Convolution.cpp 731 aten/src/ATen/native/vulkan/ops/Convolution.h 100 aten/src/ATen/native/vulkan/ops/Copy.cpp 63 aten/src/ATen/native/vulkan/ops/Copy.h 13 aten/src/ATen/native/vulkan/ops/Factory.cpp 51 aten/src/ATen/native/vulkan/ops/Mean.cpp 93 aten/src/ATen/native/vulkan/ops/Mm.cpp 345 aten/src/ATen/native/vulkan/ops/Mm.h 41 aten/src/ATen/native/vulkan/ops/Padding.cpp 93 aten/src/ATen/native/vulkan/ops/Permute.cpp 101 aten/src/ATen/native/vulkan/ops/Pool.cpp 254 aten/src/ATen/native/vulkan/ops/Register.cpp 131 aten/src/ATen/native/vulkan/ops/Shape.cpp 50 aten/src/ATen/native/vulkan/ops/Slice.cpp 238 aten/src/ATen/native/vulkan/ops/Softmax.cpp 106 aten/src/ATen/native/vulkan/ops/Tensor.cpp 1054 aten/src/ATen/native/vulkan/ops/Tensor.h 359 aten/src/ATen/native/vulkan/ops/TransposeConvolution2d.cpp 431 aten/src/ATen/native/vulkan/ops/TransposeConvolution2d.h 94 aten/src/ATen/native/vulkan/ops/Upsample.cpp 94 aten/src/ATen/native/vulkan/ops/Utils.h 19 aten/src/ATen/native/xnnpack/Activation.cpp 76 aten/src/ATen/native/xnnpack/AveragePooling.cpp 70 aten/src/ATen/native/xnnpack/ChannelShuffle.cpp 79 aten/src/ATen/native/xnnpack/Common.h 98 aten/src/ATen/native/xnnpack/Convolution.cpp 407 aten/src/ATen/native/xnnpack/Convolution.h 57 aten/src/ATen/native/xnnpack/Engine.h 62 aten/src/ATen/native/xnnpack/Init.cpp 45 aten/src/ATen/native/xnnpack/Linear.cpp 181 aten/src/ATen/native/xnnpack/Linear.h 30 aten/src/ATen/native/xnnpack/MaxPooling.cpp 188 aten/src/ATen/native/xnnpack/OpContext.cpp 140 aten/src/ATen/native/xnnpack/OpContext.h 209 aten/src/ATen/native/xnnpack/Pooling.h 40 aten/src/ATen/native/xnnpack/RegisterOpContextClass.cpp 88 aten/src/ATen/native/xnnpack/Shim.cpp 73 aten/src/ATen/nnapi/NeuralNetworks.h 51 aten/src/ATen/nnapi/codegen.py 73 aten/src/ATen/nnapi/nnapi_bind.cpp 153 aten/src/ATen/nnapi/nnapi_bind.h 38 aten/src/ATen/nnapi/nnapi_model_loader.cpp 197 aten/src/ATen/nnapi/nnapi_model_loader.h 23 aten/src/ATen/nnapi/nnapi_register.cpp 18 aten/src/ATen/nnapi/nnapi_wrapper.cpp 373 aten/src/ATen/nnapi/nnapi_wrapper.h 45 aten/src/ATen/ops/from_blob.h 96 aten/src/ATen/ops/tensor.h 26 aten/src/ATen/quantized/QTensorImpl.cpp 21 aten/src/ATen/quantized/QTensorImpl.h 76 aten/src/ATen/quantized/Quantizer.cpp 360 aten/src/ATen/quantized/Quantizer.h 161 aten/src/ATen/record_function.cpp 553 aten/src/ATen/record_function.h 401 aten/src/ATen/templates/ATenOpList.cpp 28 aten/src/ATen/templates/DispatchKeyFunction.h 29 aten/src/ATen/templates/DispatchKeyFunctions.h 2 aten/src/ATen/templates/DispatchKeyFunctions_inl.h 17 aten/src/ATen/templates/DispatchKeyNativeFunctions.h 7 aten/src/ATen/templates/Function.h 18 aten/src/ATen/templates/FunctionalInverses.h 9 aten/src/ATen/templates/Functions.cpp 81 aten/src/ATen/templates/Functions.h 78 aten/src/ATen/templates/MethodOperators.h 36 aten/src/ATen/templates/NativeFunction.h 17 aten/src/ATen/templates/NativeFunctions.h 30 aten/src/ATen/templates/NativeMetaFunction.h 17 aten/src/ATen/templates/NativeMetaFunctions.h 10 aten/src/ATen/templates/Operator.h 28 aten/src/ATen/templates/Operators.cpp 10 aten/src/ATen/templates/Operators.h 29 aten/src/ATen/templates/RedispatchFunctions.cpp 10 aten/src/ATen/templates/RedispatchFunctions.h 25 aten/src/ATen/templates/RegisterBackendSelect.cpp 33 aten/src/ATen/templates/RegisterDispatchKey.cpp 46 aten/src/ATen/templates/RegisterFunctionalization.cpp 22 aten/src/ATen/templates/RegisterSchema.cpp 8 aten/src/ATen/templates/RegistrationDeclarations.h 1 aten/src/ATen/templates/TensorBody.h 380 aten/src/ATen/templates/TensorMethods.cpp 25 aten/src/ATen/templates/aten_interned_strings.h 12 aten/src/ATen/vulkan/Context.cpp 31 aten/src/ATen/vulkan/Context.h 19 aten/src/THC/THCAtomics.cuh 3 aten/src/THC/THCDeviceUtils.cuh 3 benchmarks/compare-fastrnn-results.py 47 benchmarks/cpp/convolution.cpp 289 benchmarks/cpp/nvfuser/batch_norm.cpp 191 benchmarks/cpp/nvfuser/batch_norm_backward.cpp 211 benchmarks/cpp/nvfuser/bert.cpp 557 benchmarks/cpp/nvfuser/broadcast.cpp 260 benchmarks/cpp/nvfuser/gelu_backward.cpp 148 benchmarks/cpp/nvfuser/heuristic_cache.cpp 135 benchmarks/cpp/nvfuser/heuristic_lookup.cpp 135 benchmarks/cpp/nvfuser/instance_norm.cpp 165 benchmarks/cpp/nvfuser/layer_norm.cpp 168 benchmarks/cpp/nvfuser/layer_norm_backward.cpp 202 benchmarks/cpp/nvfuser/lstm_cell.cpp 163 benchmarks/cpp/nvfuser/main.cpp 2 benchmarks/cpp/nvfuser/reduction.cpp 258 benchmarks/cpp/nvfuser/scale_bias_relu.cpp 301 benchmarks/cpp/nvfuser/shape_inference.cpp 170 benchmarks/cpp/nvfuser/softmax.cpp 320 benchmarks/cpp/nvfuser/softmax_backward.cpp 255 benchmarks/cpp/nvfuser/softmax_dropout.cpp 300 benchmarks/cpp/nvfuser/utils.cpp 150 benchmarks/cpp/nvfuser/utils.h 103 benchmarks/cpp/tensorexpr/bench_approx.cpp 409 benchmarks/cpp/tensorexpr/bench_batchnorm.cpp 203 benchmarks/cpp/tensorexpr/bench_compile.cpp 78 benchmarks/cpp/tensorexpr/bench_concat.cpp 251 benchmarks/cpp/tensorexpr/bench_fuser_overhead.cpp 44 benchmarks/cpp/tensorexpr/bench_gemm.cpp 265 benchmarks/cpp/tensorexpr/bench_ops.py 84 benchmarks/cpp/tensorexpr/bench_parallel.cpp 61 benchmarks/cpp/tensorexpr/bench_prefix_sum.cpp 336 benchmarks/cpp/tensorexpr/bench_reduce.cpp 527 benchmarks/cpp/tensorexpr/bench_signed_log1p.cpp 143 benchmarks/cpp/tensorexpr/main.cpp 2 benchmarks/distributed/ddp/benchmark.py 214 benchmarks/distributed/ddp/compare/compare_ddp.py 181 benchmarks/distributed/ddp/compare/python_ddp.py 112 benchmarks/distributed/ddp/diff.py 56 benchmarks/distributed/pipeline/benchmark_dataset.py 37 benchmarks/distributed/pipeline/pipe.py 208 benchmarks/distributed/rpc/parameter_server/data/DummyData.py 37 benchmarks/distributed/rpc/parameter_server/data/__init__.py 4 benchmarks/distributed/rpc/parameter_server/launcher.py 438 benchmarks/distributed/rpc/parameter_server/metrics/CPUMetric.py 17 benchmarks/distributed/rpc/parameter_server/metrics/CUDAMetric.py 25 benchmarks/distributed/rpc/parameter_server/metrics/MetricBase.py 19 benchmarks/distributed/rpc/parameter_server/metrics/MetricsLogger.py 44 benchmarks/distributed/rpc/parameter_server/metrics/ProcessedMetricsPrinter.py 44 benchmarks/distributed/rpc/parameter_server/models/DummyModel.py 21 benchmarks/distributed/rpc/parameter_server/models/__init__.py 4 benchmarks/distributed/rpc/parameter_server/server/__init__.py 5 benchmarks/distributed/rpc/parameter_server/server/server.py 195 benchmarks/distributed/rpc/parameter_server/trainer/__init__.py 31 benchmarks/distributed/rpc/parameter_server/trainer/criterions.py 4 benchmarks/distributed/rpc/parameter_server/trainer/ddp_models.py 9 benchmarks/distributed/rpc/parameter_server/trainer/hook_states.py 12 benchmarks/distributed/rpc/parameter_server/trainer/hooks.py 58 benchmarks/distributed/rpc/parameter_server/trainer/iteration_steps.py 13 benchmarks/distributed/rpc/parameter_server/trainer/preprocess_data.py 6 benchmarks/distributed/rpc/parameter_server/trainer/trainer.py 144 benchmarks/distributed/rpc/parameter_server/utils.py 43 benchmarks/distributed/rpc/rl/agent.py 103 benchmarks/distributed/rpc/rl/coordinator.py 89 benchmarks/distributed/rpc/rl/launcher.py 146 benchmarks/distributed/rpc/rl/observer.py 39 benchmarks/fastrnns/__init__.py 7 benchmarks/fastrnns/bench.py 210 benchmarks/fastrnns/cells.py 79 benchmarks/fastrnns/conftest.py 13 benchmarks/fastrnns/custom_lstms.py 338 benchmarks/fastrnns/factory.py 350 benchmarks/fastrnns/fuser.py 33 benchmarks/fastrnns/profile.py 99 benchmarks/fastrnns/runner.py 57 benchmarks/fastrnns/scratch.py 35 benchmarks/fastrnns/test.py 119 benchmarks/framework_overhead_benchmark/C2Module.py 29 benchmarks/framework_overhead_benchmark/SimpleAddModule.py 13 benchmarks/framework_overhead_benchmark/framework_overhead_benchmark.py 62 benchmarks/framework_overhead_benchmark/pt_wrapper_module.py 23 benchmarks/framework_overhead_benchmark/utils.py 28 benchmarks/functional_autograd_benchmark/audio_text_models.py 90 benchmarks/functional_autograd_benchmark/compare.py 37 benchmarks/functional_autograd_benchmark/functional_autograd_benchmark.py 136 benchmarks/functional_autograd_benchmark/ppl_models.py 52 benchmarks/functional_autograd_benchmark/torchaudio_models.py 330 benchmarks/functional_autograd_benchmark/torchvision_models.py 466 benchmarks/functional_autograd_benchmark/utils.py 52 benchmarks/functional_autograd_benchmark/vision_models.py 76 benchmarks/fuser/plot_speedups.py 18 benchmarks/fuser/run_benchmarks.py 228 benchmarks/instruction_counts/applications/__init__.py 1 benchmarks/instruction_counts/applications/ci.py 59 benchmarks/instruction_counts/core/__init__.py 1 benchmarks/instruction_counts/core/api.py 236 benchmarks/instruction_counts/core/expand.py 140 benchmarks/instruction_counts/core/types.py 11 benchmarks/instruction_counts/core/utils.py 69 benchmarks/instruction_counts/definitions/__init__.py 1 benchmarks/instruction_counts/definitions/setup.py 23 benchmarks/instruction_counts/definitions/standard.py 78 benchmarks/instruction_counts/execution/__init__.py 1 benchmarks/instruction_counts/execution/runner.py 188 benchmarks/instruction_counts/execution/work.py 152 benchmarks/instruction_counts/main.py 25 benchmarks/instruction_counts/worker/__init__.py 1 benchmarks/instruction_counts/worker/main.py 93 benchmarks/operator_benchmark/__init__.py 1 benchmarks/operator_benchmark/benchmark_caffe2.py 113 benchmarks/operator_benchmark/benchmark_core.py 242 benchmarks/operator_benchmark/benchmark_pytorch.py 101 benchmarks/operator_benchmark/benchmark_runner.py 113 benchmarks/operator_benchmark/benchmark_test_generator.py 12 benchmarks/operator_benchmark/benchmark_utils.py 158 benchmarks/operator_benchmark/c2/__init__.py 1 benchmarks/operator_benchmark/common/__init__.py 1 benchmarks/operator_benchmark/common/repeat_benchmark.py 45 benchmarks/operator_benchmark/operator_benchmark.py 4 benchmarks/operator_benchmark/pt/__init__.py 1 benchmarks/operator_benchmark/pt/configs.py 101 benchmarks/operator_benchmark/pt_extension/extension.cpp 18 benchmarks/operator_benchmark/pt_extension/setup.py 5 benchmarks/overrides_benchmark/bench.py 52 benchmarks/overrides_benchmark/common.py 22 benchmarks/overrides_benchmark/pyspybench.py 22 benchmarks/profiler_benchmark/profiler_bench.py 84 benchmarks/profiler_benchmark/resnet_memory_profiler.py 18 benchmarks/record_function_benchmark/record_function_bench.py 83 benchmarks/serialization/simple_measurement.py 26 benchmarks/sparse/__init__.py 2 benchmarks/sparse/dlmc/__init__.py 2 benchmarks/sparse/dlmc/matmul_bench.py 98 benchmarks/sparse/dlmc/utils.py 131 benchmarks/sparse/spmm.py 82 benchmarks/sparse/spmv.py 80 benchmarks/sparse/utils.py 42 benchmarks/static_runtime/deep_wide_pt.cc 124 benchmarks/static_runtime/deep_wide_pt.h 103 benchmarks/static_runtime/deep_wide_pt_bench.cc 163 benchmarks/tensorexpr/__main__.py 288 benchmarks/tensorexpr/attention.py 63 benchmarks/tensorexpr/benchmark.py 239 benchmarks/tensorexpr/broadcast.py 240 benchmarks/tensorexpr/concat.py 99 benchmarks/tensorexpr/conv.py 87 benchmarks/tensorexpr/elementwise.py 188 benchmarks/tensorexpr/matmul.py 49 benchmarks/tensorexpr/microbenchmarks.py 205 benchmarks/tensorexpr/normalization.py 62 benchmarks/tensorexpr/pooling.py 50 benchmarks/tensorexpr/pt_engine.py 50 benchmarks/tensorexpr/reduction.py 215 benchmarks/tensorexpr/rnn_eltwise.py 87 benchmarks/tensorexpr/softmax.py 42 benchmarks/tensorexpr/swish.py 39 benchmarks/tensorexpr/tensor_engine.py 34 benchmarks/upload_scribe.py 120 binaries/aot_model_compiler.cc 222 binaries/at_launch_benchmark.cc 78 binaries/bench_gen/bench_gen.py 73 binaries/benchmark_args.h 77 binaries/benchmark_helper.cc 467 binaries/benchmark_helper.h 145 binaries/caffe2_benchmark.cc 29 binaries/compare_models_torch.cc 250 binaries/convert_and_benchmark.cc 674 binaries/convert_caffe_image_db.cc 67 binaries/convert_db.cc 31 binaries/convert_encoded_to_raw_leveldb.cc 115 binaries/convert_image_to_tensor.cc 389 binaries/core_overhead_benchmark.cc 35 binaries/core_overhead_benchmark_gpu.cc 179 binaries/db_throughput.cc 80 binaries/dump_operator_names.cc 61 binaries/inspect_gpu.cc 32 binaries/intra_inter_benchmark.cc 140 binaries/lite_interpreter_model_load.cc 26 binaries/make_cifar_db.cc 102 binaries/make_image_db.cc 203 binaries/make_mnist_db.cc 109 binaries/optimize_for_mobile.cc 81 binaries/parallel_info.cc 19 binaries/predictor_verifier.cc 35 binaries/print_core_object_sizes_gpu.cc 21 binaries/print_registered_core_operators.cc 46 binaries/record_function_benchmark.cc 110 binaries/run_plan.cc 20 binaries/run_plan_mpi.cc 28 binaries/speed_benchmark.cc 178 binaries/speed_benchmark_torch.cc 267 binaries/split_db.cc 50 binaries/tsv_2_proto.cc 30 binaries/tutorial_blob.cc 52 binaries/zmq_feeder.cc 33 c10/benchmark/intrusive_ptr_benchmark.cpp 63 c10/core/Allocator.cpp 49 c10/core/Allocator.h 127 c10/core/AutogradState.cpp 15 c10/core/AutogradState.h 35 c10/core/Backend.h 273 c10/core/CPUAllocator.cpp 256 c10/core/CPUAllocator.h 39 c10/core/CompileTimeFunctionPointer.h 24 c10/core/CopyBytes.cpp 41 c10/core/CopyBytes.h 29 c10/core/DefaultDtype.cpp 32 c10/core/DefaultDtype.h 12 c10/core/DefaultTensorOptions.h 33 c10/core/Device.cpp 129 c10/core/Device.h 86 c10/core/DeviceArray.h 19 c10/core/DeviceGuard.h 66 c10/core/DeviceType.cpp 82 c10/core/DeviceType.h 67 c10/core/DispatchKey.cpp 250 c10/core/DispatchKey.h 142 c10/core/DispatchKeySet.cpp 122 c10/core/DispatchKeySet.h 211 c10/core/Event.h 49 c10/core/GeneratorImpl.cpp 60 c10/core/GeneratorImpl.h 49 c10/core/GradMode.cpp 10 c10/core/GradMode.h 31 c10/core/InferenceMode.cpp 7 c10/core/InferenceMode.h 35 c10/core/Layout.h 42 c10/core/MemoryFormat.h 145 c10/core/OptionalRef.h 24 c10/core/QEngine.h 27 c10/core/QScheme.h 37 c10/core/Scalar.cpp 31 c10/core/Scalar.h 161 c10/core/ScalarType.h 353 c10/core/ScalarTypeToTypeMeta.h 30 c10/core/Storage.cpp 2 c10/core/Storage.h 128 c10/core/StorageImpl.cpp 1 c10/core/StorageImpl.h 135 c10/core/Stream.cpp 16 c10/core/Stream.h 81 c10/core/StreamGuard.h 62 c10/core/TensorImpl.cpp 482 c10/core/TensorImpl.h 1402 c10/core/TensorOptions.cpp 28 c10/core/TensorOptions.h 517 c10/core/UndefinedTensorImpl.cpp 31 c10/core/UndefinedTensorImpl.h 25 c10/core/WrapDimMinimal.h 35 c10/core/impl/DeviceGuardImplInterface.cpp 13 c10/core/impl/DeviceGuardImplInterface.h 127 c10/core/impl/FakeGuardImpl.h 85 c10/core/impl/InlineDeviceGuard.h 187 c10/core/impl/InlineEvent.h 91 c10/core/impl/InlineStreamGuard.h 147 c10/core/impl/LocalDispatchKeySet.cpp 72 c10/core/impl/LocalDispatchKeySet.h 78 c10/core/impl/SizesAndStrides.cpp 66 c10/core/impl/SizesAndStrides.h 240 c10/core/impl/VirtualGuardImpl.h 72 c10/core/thread_pool.cpp 105 c10/core/thread_pool.h 90 c10/cuda/CUDACachingAllocator.cpp 1187 c10/cuda/CUDACachingAllocator.h 94 c10/cuda/CUDAException.h 50 c10/cuda/CUDAFunctions.cpp 126 c10/cuda/CUDAFunctions.h 55 c10/cuda/CUDAGraphsC10Utils.h 71 c10/cuda/CUDAGuard.h 145 c10/cuda/CUDAMacros.h 28 c10/cuda/CUDAMathCompat.h 126 c10/cuda/CUDAMiscFunctions.cpp 18 c10/cuda/CUDAMiscFunctions.h 7 c10/cuda/CUDAStream.cpp 197 c10/cuda/CUDAStream.h 101 c10/cuda/impl/CUDAGuardImpl.cpp 9 c10/cuda/impl/CUDAGuardImpl.h 167 c10/cuda/impl/CUDATest.cpp 24 c10/cuda/impl/CUDATest.h 9 c10/cuda/impl/cuda_cmake_macros.h.in 2 c10/macros/Export.h 66 c10/macros/Macros.h 313 c10/macros/build.bzl 30 c10/macros/cmake_configure_file.bzl 59 c10/macros/cmake_macros.h.in 8 c10/mobile/CPUCachingAllocator.cpp 67 c10/mobile/CPUCachingAllocator.h 35 c10/mobile/CPUProfilingAllocator.cpp 347 c10/mobile/CPUProfilingAllocator.h 77 c10/util/AlignOf.h 109 c10/util/Array.cpp 1 c10/util/Array.h 272 c10/util/ArrayRef.h 222 c10/util/BFloat16-inl.h 262 c10/util/BFloat16-math.h 133 c10/util/BFloat16.h 74 c10/util/Backtrace.cpp 275 c10/util/Backtrace.h 13 c10/util/Bitset.h 88 c10/util/C++17.cpp 1 c10/util/C++17.h 291 c10/util/ConstexprCrc.h 121 c10/util/DeadlockDetection.cpp 18 c10/util/DeadlockDetection.h 24 c10/util/Deprecated.h 52 c10/util/Exception.cpp 163 c10/util/Exception.h 341 c10/util/ExclusivelyOwned.h 65 c10/util/Flags.h 101 c10/util/FunctionRef.h 38 c10/util/Half-inl.h 259 c10/util/Half.cpp 11 c10/util/Half.h 243 c10/util/IdWrapper.h 44 c10/util/LeftRight.cpp 1 c10/util/LeftRight.h 116 c10/util/Logging.cpp 260 c10/util/Logging.h 201 c10/util/MathConstants.cpp 8 c10/util/MathConstants.h 27 c10/util/MaybeOwned.h 166 c10/util/Metaprogramming.cpp 1 c10/util/Metaprogramming.h 318 c10/util/Optional.cpp 15 c10/util/Optional.h 928 c10/util/Registry.h 214 c10/util/ScopeExit.h 33 c10/util/SmallBuffer.h 51 c10/util/SmallVector.cpp 110 c10/util/SmallVector.h 946 c10/util/StringUtil.cpp 41 c10/util/StringUtil.h 160 c10/util/ThreadLocal.h 89 c10/util/ThreadLocalDebugInfo.cpp 71 c10/util/ThreadLocalDebugInfo.h 48 c10/util/Type.h 19 c10/util/TypeCast.cpp 8 c10/util/TypeCast.h 114 c10/util/TypeIndex.h 139 c10/util/TypeList.cpp 1 c10/util/TypeList.h 350 c10/util/TypeSafeSignMath.h 85 c10/util/TypeTraits.cpp 1 c10/util/TypeTraits.h 90 c10/util/Type_demangle.cpp 25 c10/util/Type_no_demangle.cpp 8 c10/util/Unicode.cpp 48 c10/util/Unicode.h 12 c10/util/UniqueVoidPtr.cpp 6 c10/util/UniqueVoidPtr.h 71 c10/util/Unroll.h 19 c10/util/accumulate.h 92 c10/util/complex.h 372 c10/util/complex_math.cpp 62 c10/util/complex_math.h 298 c10/util/complex_utils.h 29 c10/util/copysign.h 17 c10/util/either.h 192 c10/util/env.h 26 c10/util/flags_use_gflags.cpp 25 c10/util/flags_use_no_gflags.cpp 169 c10/util/flat_hash_map.h 2001 c10/util/hash.h 235 c10/util/in_place.h 16 c10/util/int128.cpp 126 c10/util/int128.h 264 c10/util/intrusive_ptr.cpp 1 c10/util/intrusive_ptr.h 620 c10/util/irange.h 74 c10/util/llvmMathExtras.h 568 c10/util/logging_is_google_glog.h 31 c10/util/logging_is_not_google_glog.h 191 c10/util/math_compat.h 209 c10/util/numa.cpp 102 c10/util/numa.h 12 c10/util/order_preserving_flat_hash_map.h 2068 c10/util/overloaded.h 23 c10/util/python_stub.h 3 c10/util/qint32.h 11 c10/util/qint8.h 11 c10/util/quint2x4.h 11 c10/util/quint4x2.h 11 c10/util/quint8.h 11 c10/util/reverse_iterator.h 194 c10/util/signal_handler.cpp 317 c10/util/signal_handler.h 74 c10/util/sparse_bitset.h 662 c10/util/string_utils.h 107 c10/util/string_view.h 545 c10/util/tempfile.h 145 c10/util/thread_name.cpp 21 c10/util/thread_name.h 6 c10/util/typeid.cpp 64 c10/util/typeid.h 391 c10/util/variant.h 2317 c10/util/win32-headers.h 54 caffe2/__init__.py 1 caffe2/c2_aten_srcs.bzl 11 caffe2/contrib/__init__.py 1 caffe2/contrib/aten/__init__.py 1 caffe2/contrib/aten/aten_op.cc 45 caffe2/contrib/aten/aten_op.h 1 caffe2/contrib/aten/aten_op_gpu.cc 9 caffe2/contrib/aten/aten_op_template.h 211 caffe2/contrib/aten/gen_op.py 194 caffe2/contrib/fakelowp/batch_matmul_fp16_fake_op.cc 56 caffe2/contrib/fakelowp/batch_matmul_fp16_fake_op.h 420 caffe2/contrib/fakelowp/common.cc 3 caffe2/contrib/fakelowp/common.h 3 caffe2/contrib/fakelowp/elementwise_fp16_fake_op.cc 84 caffe2/contrib/fakelowp/fp16_fc_acc_op.cc 81 caffe2/contrib/fakelowp/fp16_fc_acc_op.h 336 caffe2/contrib/fakelowp/fp16_fma.cc 66 caffe2/contrib/fakelowp/fp16_fma.h 8 caffe2/contrib/fakelowp/fp16_fma_slow.cc 429 caffe2/contrib/fakelowp/fp16_gemm_utils.cc 428 caffe2/contrib/fakelowp/fp16_gemm_utils.h 76 caffe2/contrib/fakelowp/int8_dequantize_op_nnpi.cc 10 caffe2/contrib/fakelowp/int8_dequantize_op_nnpi.h 43 caffe2/contrib/fakelowp/int8_quantize_op_nnpi.cc 12 caffe2/contrib/fakelowp/int8_quantize_op_nnpi.h 93 caffe2/contrib/fakelowp/int8_swish_op_nnpi.cc 23 caffe2/contrib/fakelowp/int8_swish_op_nnpi.h 69 caffe2/contrib/fakelowp/layernorm_fp16_fake_op.cc 177 caffe2/contrib/fakelowp/layernorm_fp16_fake_op.h 159 caffe2/contrib/fakelowp/lengths_reducer_fused_4bit_rowwise_fp16_fake_op.cc 156 caffe2/contrib/fakelowp/lengths_reducer_fused_4bit_rowwise_fp16_fake_op.h 176 caffe2/contrib/fakelowp/lengths_reducer_fused_8bit_rowwise_fp16_fake_op.cc 700 caffe2/contrib/fakelowp/lengths_reducer_fused_8bit_rowwise_fp16_fake_op.h 242 caffe2/contrib/fakelowp/lengths_reducer_ops.cc 198 caffe2/contrib/fakelowp/lengths_reducer_ops.h 217 caffe2/contrib/fakelowp/quant_lut_fp16_fake_op.cc 16 caffe2/contrib/fakelowp/quant_lut_fp16_fake_op.h 68 caffe2/contrib/fakelowp/spatial_batch_norm_fp16_fake_op.cc 9 caffe2/contrib/fakelowp/spatial_batch_norm_fp16_fake_op.h 333 caffe2/contrib/fakelowp/sum_fp16_fake_op.h 55 caffe2/contrib/fakelowp/unary_fp16_fake_op.cc 863 caffe2/contrib/fakelowp/unary_fp16_fake_op.h 63 caffe2/contrib/gloo/__init__.py 1 caffe2/contrib/gloo/allgather_ops.cc 40 caffe2/contrib/gloo/allgather_ops.h 82 caffe2/contrib/gloo/allreduce_ops.cc 98 caffe2/contrib/gloo/allreduce_ops.h 101 caffe2/contrib/gloo/allreduce_ops_gpu.cc 140 caffe2/contrib/gloo/barrier_ops.cc 8 caffe2/contrib/gloo/barrier_ops.h 51 caffe2/contrib/gloo/broadcast_ops.cc 30 caffe2/contrib/gloo/broadcast_ops.h 81 caffe2/contrib/gloo/broadcast_ops_gpu.cc 31 caffe2/contrib/gloo/common.cc 41 caffe2/contrib/gloo/common.h 54 caffe2/contrib/gloo/common_world_ops.cc 20 caffe2/contrib/gloo/common_world_ops.h 209 caffe2/contrib/gloo/common_world_ops_gpu.cc 25 caffe2/contrib/gloo/context.cc 7 caffe2/contrib/gloo/context.h 2 caffe2/contrib/gloo/py_export.cc 11 caffe2/contrib/gloo/reduce_scatter_ops.cc 31 caffe2/contrib/gloo/reduce_scatter_ops.h 83 caffe2/contrib/gloo/store_handler.cc 20 caffe2/contrib/gloo/store_handler.h 24 caffe2/contrib/nccl/__init__.py 1 caffe2/contrib/nccl/cuda_nccl_gpu.cc 276 caffe2/contrib/nccl/cuda_nccl_gpu.h 50 caffe2/contrib/nccl/cuda_nccl_op_gpu.cc 231 caffe2/contrib/nnpack/__init__.py 1 caffe2/contrib/nnpack/nnpack_ops.cc 294 caffe2/contrib/opencl/OpenCL/cl.hpp 10263 caffe2/contrib/opencl/context.cc 86 caffe2/contrib/opencl/context.h 81 caffe2/contrib/playground/AnyExp.py 357 caffe2/contrib/playground/AnyExpOnTerm.py 68 caffe2/contrib/playground/ModuleRegister.py 88 caffe2/contrib/playground/__init__.py 1 caffe2/contrib/playground/checkpoint.py 147 caffe2/contrib/playground/compute_loss.py 24 caffe2/contrib/playground/compute_topk_accuracy.py 42 caffe2/contrib/playground/meter.py 14 caffe2/contrib/playground/module_map.py 24 caffe2/contrib/playground/output_generator.py 10 caffe2/contrib/playground/resnetdemo/IN1k_resnet.py 40 caffe2/contrib/playground/resnetdemo/IN1k_resnet_no_test_model.py 33 caffe2/contrib/playground/resnetdemo/__init__.py 1 caffe2/contrib/playground/resnetdemo/caffe2_resnet50_default_forward.py 16 caffe2/contrib/playground/resnetdemo/caffe2_resnet50_default_param_update.py 32 caffe2/contrib/playground/resnetdemo/explicit_resnet_forward.py 260 caffe2/contrib/playground/resnetdemo/explicit_resnet_param_update.py 50 caffe2/contrib/playground/resnetdemo/gfs_IN1k.py 40 caffe2/contrib/playground/resnetdemo/override_no_test_model_no_checkpoint.py 8 caffe2/contrib/playground/resnetdemo/rendezvous_filestore.py 22 caffe2/contrib/prof/__init__.py 1 caffe2/contrib/prof/cuda_profile_ops.cc 85 caffe2/contrib/prof/prof_dag_stats_op.cc 23 caffe2/contrib/prof/prof_dag_stats_op.h 74 caffe2/contrib/pytorch/script_module_op.cc 132 caffe2/contrib/script/__init__.py 1 caffe2/contrib/shm_mutex/shm_mutex.cc 14 caffe2/contrib/shm_mutex/shm_mutex.h 258 caffe2/contrib/tensorboard/__init__.py 1 caffe2/contrib/tensorboard/tensorboard.py 147 caffe2/contrib/tensorboard/tensorboard_exporter.py 261 caffe2/contrib/tensorrt/tensorrt_op_trt.cc 212 caffe2/contrib/tensorrt/tensorrt_op_trt.h 25 caffe2/contrib/tensorrt/tensorrt_tranformer.cc 443 caffe2/contrib/tensorrt/tensorrt_tranformer.h 66 caffe2/contrib/tensorrt/trt_utils.cc 59 caffe2/contrib/tensorrt/trt_utils.h 44 caffe2/contrib/warpctc/__init__.py 1 caffe2/contrib/warpctc/ctc_op.cpp 33 caffe2/contrib/warpctc/ctc_op.h 103 caffe2/contrib/warpctc/ctc_op_gpu.cpp 15 caffe2/core/__init__.py 1 caffe2/core/allocator.cc 1 caffe2/core/allocator.h 4 caffe2/core/blob.h 106 caffe2/core/blob_serialization.cc 984 caffe2/core/blob_serialization.h 175 caffe2/core/blob_serialization_gpu.cc 8 caffe2/core/blob_serializer_base.h 59 caffe2/core/blob_stats.cc 28 caffe2/core/blob_stats.h 32 caffe2/core/common.cc 27 caffe2/core/common.h 100 caffe2/core/common_cudnn.cc 18 caffe2/core/common_cudnn.h 261 caffe2/core/common_gpu.cc 227 caffe2/core/common_gpu.h 322 caffe2/core/common_omp.h 6 caffe2/core/context.cc 52 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/cudnn_wrappers.h 98 caffe2/core/db.cc 167 caffe2/core/db.h 191 caffe2/core/distributions_stubs.h 48 caffe2/core/event.cc 114 caffe2/core/event.h 292 caffe2/core/event_cpu.h 33 caffe2/core/event_gpu.cc 177 caffe2/core/export_c10_op_to_caffe2.cc 8 caffe2/core/export_c10_op_to_caffe2.h 207 caffe2/core/export_caffe2_op_to_c10.h 159 caffe2/core/flags.h 3 caffe2/core/graph.cc 210 caffe2/core/graph.h 75 caffe2/core/hip/common_miopen.h 126 caffe2/core/hip/miopen_wrapper.h 104 caffe2/core/init.cc 80 caffe2/core/init.h 112 caffe2/core/init_denormals.cc 30 caffe2/core/init_intrinsics_check.cc 67 caffe2/core/init_omp.cc 62 caffe2/core/int8_serialization.cc 97 caffe2/core/logging.h 3 caffe2/core/macros.h 1 caffe2/core/macros.h.in 71 caffe2/core/memonger.cc 548 caffe2/core/memonger.h 23 caffe2/core/module.cc 94 caffe2/core/module.h 27 caffe2/core/net.cc 204 caffe2/core/net.h 116 caffe2/core/net_async_base.cc 534 caffe2/core/net_async_base.h 181 caffe2/core/net_async_scheduling.cc 234 caffe2/core/net_async_scheduling.h 29 caffe2/core/net_async_task.cc 85 caffe2/core/net_async_task.h 26 caffe2/core/net_async_task_future.cc 89 caffe2/core/net_async_task_future.h 52 caffe2/core/net_async_task_graph.cc 110 caffe2/core/net_async_task_graph.h 44 caffe2/core/net_async_tracing.cc 413 caffe2/core/net_async_tracing.h 124 caffe2/core/net_dag_utils.cc 402 caffe2/core/net_dag_utils.h 50 caffe2/core/net_parallel.cc 163 caffe2/core/net_parallel.h 63 caffe2/core/net_simple.cc 297 caffe2/core/net_simple.h 67 caffe2/core/net_simple_refcount.cc 66 caffe2/core/net_simple_refcount.h 26 caffe2/core/nomnigraph/Representations/NeuralNet.cc 317 caffe2/core/nomnigraph/__init__.py 1 caffe2/core/nomnigraph/include/nomnigraph/Converters/Dot.h 209 caffe2/core/nomnigraph/include/nomnigraph/Generated/OpClasses.h 571 caffe2/core/nomnigraph/include/nomnigraph/Generated/OpEnum.h 4 caffe2/core/nomnigraph/include/nomnigraph/Generated/OpNames.h 56 caffe2/core/nomnigraph/include/nomnigraph/Graph/Algorithms.h 154 caffe2/core/nomnigraph/include/nomnigraph/Graph/BinaryMatchImpl.h 76 caffe2/core/nomnigraph/include/nomnigraph/Graph/Graph.h 406 caffe2/core/nomnigraph/include/nomnigraph/Graph/TarjansImpl.h 122 caffe2/core/nomnigraph/include/nomnigraph/Graph/TopoSort.h 59 caffe2/core/nomnigraph/include/nomnigraph/Representations/Compiler.h 84 caffe2/core/nomnigraph/include/nomnigraph/Representations/ControlFlow.h 144 caffe2/core/nomnigraph/include/nomnigraph/Representations/NeuralNet.h 381 caffe2/core/nomnigraph/include/nomnigraph/Support/Casting.h 322 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/nomnigraph/op_gen.py 180 caffe2/core/numa.cc 1 caffe2/core/numa.h 3 caffe2/core/observer.h 118 caffe2/core/operator.cc 782 caffe2/core/operator.h 1182 caffe2/core/operator_gradient.h 233 caffe2/core/operator_schema.cc 453 caffe2/core/operator_schema.h 349 caffe2/core/plan_executor.cc 551 caffe2/core/plan_executor.h 8 caffe2/core/prof_dag_counters.cc 206 caffe2/core/prof_dag_counters.h 80 caffe2/core/qtensor.cc 4 caffe2/core/qtensor.h 161 caffe2/core/qtensor_serialization.cc 9 caffe2/core/qtensor_serialization.h 74 caffe2/core/scope_guard.h 83 caffe2/core/static_tracepoint.h 9 caffe2/core/static_tracepoint_elfx86.h 74 caffe2/core/stats.cc 44 caffe2/core/stats.h 209 caffe2/core/storage.h 26 caffe2/core/tensor.cc 253 caffe2/core/tensor.h 373 caffe2/core/tensor_impl.h 13 caffe2/core/tensor_int8.cc 5 caffe2/core/tensor_int8.h 15 caffe2/core/timer.h 24 caffe2/core/transform.cc 193 caffe2/core/transform.h 68 caffe2/core/types.cc 73 caffe2/core/types.h 57 caffe2/core/workspace.cc 278 caffe2/core/workspace.h 164 caffe2/cuda_rtc/common_rtc.h 109 caffe2/cuda_rtc/elemenntwise_rtc_gpu.cc 95 caffe2/cuda_rtc/pool_op_rtc_gpu.cc 309 caffe2/db/create_db_op.cc 6 caffe2/db/create_db_op.h 35 caffe2/db/create_db_op_gpu.cc 5 caffe2/db/leveldb.cc 101 caffe2/db/lmdb.cc 153 caffe2/db/protodb.cc 97 caffe2/db/zmqdb.cc 100 caffe2/distributed/__init__.py 1 caffe2/distributed/file_store_handler.cc 136 caffe2/distributed/file_store_handler.h 25 caffe2/distributed/file_store_handler_op.cc 21 caffe2/distributed/file_store_handler_op.h 31 caffe2/distributed/file_store_handler_op_gpu.cc 17 caffe2/distributed/py_export.cc 14 caffe2/distributed/python.pyi 2 caffe2/distributed/redis_store_handler.cc 107 caffe2/distributed/redis_store_handler.h 30 caffe2/distributed/redis_store_handler_op.cc 17 caffe2/distributed/redis_store_handler_op.h 35 caffe2/distributed/redis_store_handler_op_gpu.cc 17 caffe2/distributed/store_handler.cc 10 caffe2/distributed/store_handler.h 42 caffe2/distributed/store_ops.cc 109 caffe2/distributed/store_ops.h 40 caffe2/distributed/store_ops_test_util.py 50 caffe2/experiments/__init__.py 1 caffe2/experiments/operators/fully_connected_op_decomposition.cc 21 caffe2/experiments/operators/fully_connected_op_decomposition.h 232 caffe2/experiments/operators/fully_connected_op_decomposition_gpu.cc 8 caffe2/experiments/operators/fully_connected_op_prune.cc 26 caffe2/experiments/operators/fully_connected_op_prune.h 316 caffe2/experiments/operators/fully_connected_op_sparse.cc 7 caffe2/experiments/operators/fully_connected_op_sparse.h 144 caffe2/experiments/operators/funhash_op.cc 64 caffe2/experiments/operators/funhash_op.h 176 caffe2/experiments/operators/sparse_funhash_op.cc 70 caffe2/experiments/operators/sparse_funhash_op.h 180 caffe2/experiments/operators/sparse_matrix_reshape_op.cc 41 caffe2/experiments/operators/sparse_matrix_reshape_op.h 94 caffe2/experiments/operators/tt_contraction_op.cc 32 caffe2/experiments/operators/tt_contraction_op.h 127 caffe2/experiments/operators/tt_contraction_op_gpu.cc 8 caffe2/experiments/operators/tt_pad_op.cc 24 caffe2/experiments/operators/tt_pad_op.h 59 caffe2/experiments/python/SparseTransformer.py 131 caffe2/experiments/python/__init__.py 1 caffe2/experiments/python/convnet_benchmarks.py 567 caffe2/experiments/python/device_reduce_sum_bench.py 82 caffe2/experiments/python/net_construct_bench.py 97 caffe2/ideep/ideep_utils.h 41 caffe2/ideep/operators/adam_op.cc 166 caffe2/ideep/operators/channel_shuffle_op.cc 38 caffe2/ideep/operators/concat_split_op.cc 148 caffe2/ideep/operators/conv_op.cc 310 caffe2/ideep/operators/conv_pool_base_op.h 64 caffe2/ideep/operators/conv_transpose_op.cc 134 caffe2/ideep/operators/conv_transpose_unpool_base_op.h 217 caffe2/ideep/operators/dropout_op.cc 70 caffe2/ideep/operators/elementwise_sum_op.cc 65 caffe2/ideep/operators/expand_squeeze_dims_op.cc 105 caffe2/ideep/operators/fully_connected_op.cc 110 caffe2/ideep/operators/local_response_normalization_op.cc 68 caffe2/ideep/operators/momentum_sgd_op.cc 108 caffe2/ideep/operators/operator_fallback_ideep.cc 296 caffe2/ideep/operators/operator_fallback_ideep.h 136 caffe2/ideep/operators/order_switch_ops.cc 44 caffe2/ideep/operators/pool_op.cc 94 caffe2/ideep/operators/quantization/int8_add_op.cc 62 caffe2/ideep/operators/quantization/int8_conv_op.cc 209 caffe2/ideep/operators/quantization/int8_dequantize_op.cc 32 caffe2/ideep/operators/quantization/int8_fully_connected_op.cc 75 caffe2/ideep/operators/quantization/int8_given_tensor_fill_op.cc 123 caffe2/ideep/operators/quantization/int8_pool_op.cc 50 caffe2/ideep/operators/quantization/int8_quantize_op.cc 47 caffe2/ideep/operators/quantization/int8_relu_op.cc 30 caffe2/ideep/operators/queue_ops.cc 59 caffe2/ideep/operators/relu_op.cc 68 caffe2/ideep/operators/reshape_op.cc 108 caffe2/ideep/operators/shape_op.cc 53 caffe2/ideep/operators/sigmoid_op.cc 44 caffe2/ideep/operators/spatial_batch_norm_op.cc 86 caffe2/ideep/operators/transpose_op.cc 24 caffe2/ideep/operators/utility_ops.cc 112 caffe2/ideep/utils/ideep_context.h 138 caffe2/ideep/utils/ideep_operator.h 114 caffe2/ideep/utils/ideep_register.cc 54 caffe2/image/image_input_op.cc 157 caffe2/image/image_input_op.h 1110 caffe2/image/image_input_op_gpu.cc 33 caffe2/image/transform_gpu.cu 69 caffe2/image/transform_gpu.h 13 caffe2/mobile/contrib/ios/ios_caffe.cc 46 caffe2/mobile/contrib/ios/ios_caffe.h 20 caffe2/mobile/contrib/ios/ios_caffe_defines.h 1 caffe2/mobile/contrib/ios/ios_caffe_predictor.cc 62 caffe2/mobile/contrib/ios/ios_caffe_predictor.h 25 caffe2/mobile/contrib/ios/mpscnn/MPSCNN.metal 1032 caffe2/mobile/contrib/ios/mpscnn/mpscnn.h 14 caffe2/mobile/contrib/ios/mpscnn/mpscnn.mm 2213 caffe2/mobile/contrib/ios/mpscnn/mpscnn_context.h 24 caffe2/mobile/contrib/ios/mpscnn/mpscnn_context.mm 90 caffe2/mobile/contrib/ios/mpscnn/mpscnn_graph.mm 313 caffe2/mobile/contrib/ios/mpscnn/mpscnn_graph_mask.h 9 caffe2/mobile/contrib/ios/mpscnn/mpscnn_graph_mask.mm 506 caffe2/mobile/contrib/ios/mpscnn/mpscnn_kernels.h 1034 caffe2/mobile/contrib/libopencl-stub/include/CL/cl.h 974 caffe2/mobile/contrib/libopencl-stub/include/CL/cl.hpp 9959 caffe2/mobile/contrib/libopencl-stub/include/CL/cl_ext.h 103 caffe2/mobile/contrib/libopencl-stub/include/CL/cl_gl.h 106 caffe2/mobile/contrib/libopencl-stub/include/CL/cl_gl_ext.h 19 caffe2/mobile/contrib/libopencl-stub/include/CL/cl_platform.h 1057 caffe2/mobile/contrib/libopencl-stub/include/CL/opencl.h 20 caffe2/mobile/contrib/libopencl-stub/include/libopencl.h 140 caffe2/mobile/contrib/libopencl-stub/src/libopencl.c 1810 caffe2/mobile/contrib/libvulkan-stub/include/libvulkan-stub.h 358 caffe2/mobile/contrib/libvulkan-stub/include/vulkan/vk_platform.h 58 caffe2/mobile/contrib/libvulkan-stub/include/vulkan/vulkan.h 5488 caffe2/mobile/contrib/libvulkan-stub/src/libvulkan-stub.c 515 caffe2/mobile/contrib/nnapi/NeuralNetworks.h 135 caffe2/mobile/contrib/nnapi/dlnnapi.c 95 caffe2/mobile/contrib/nnapi/dlnnapi.h 69 caffe2/mobile/contrib/nnapi/nnapi.cc 589 caffe2/mobile/contrib/nnapi/nnapi.h 83 caffe2/mobile/contrib/nnapi/nnapi_benchmark.cc 417 caffe2/mobile/contrib/snpe/snpe_ffi.cc 98 caffe2/mobile/contrib/snpe/snpe_ffi.h 21 caffe2/mobile/contrib/snpe/snpe_globals.cc 13 caffe2/mobile/contrib/snpe/snpe_op.cc 108 caffe2/mobile/contrib/snpe/snpe_op_benchmark.cc 164 caffe2/mobile/contrib/ulp2/ulp.cc 344 caffe2/mobile/contrib/ulp2/ulp.h 53 caffe2/mobile/contrib/ulp2/ulp_neon.cc 451 caffe2/mobile/contrib/ulp2/ulp_neon.h 7 caffe2/mpi/mpi_common.cc 125 caffe2/mpi/mpi_common.h 87 caffe2/mpi/mpi_ops.cc 20 caffe2/mpi/mpi_ops.h 212 caffe2/mpi/mpi_ops_gpu.cc 57 caffe2/observers/operator_attaching_net_observer.h 25 caffe2/observers/profile_observer.cc 57 caffe2/observers/profile_observer.h 75 caffe2/observers/runcnt_observer.cc 29 caffe2/observers/runcnt_observer.h 41 caffe2/observers/time_observer.cc 29 caffe2/observers/time_observer.h 59 caffe2/onnx/backend.cc 1595 caffe2/onnx/backend.h 201 caffe2/onnx/backend_rep.cc 27 caffe2/onnx/backend_rep.h 43 caffe2/onnx/device.cc 15 caffe2/onnx/device.h 21 caffe2/onnx/helper.cc 53 caffe2/onnx/helper.h 98 caffe2/onnx/offline_tensor.cc 74 caffe2/onnx/offline_tensor.h 47 caffe2/onnx/onnx_exporter.cc 1260 caffe2/onnx/onnx_exporter.h 106 caffe2/onnx/onnxifi_graph_info.cc 44 caffe2/onnx/onnxifi_graph_info.h 90 caffe2/onnx/onnxifi_init.cc 18 caffe2/onnx/onnxifi_init.h 7 caffe2/onnx/torch_ops/constants.h 5 caffe2/onnx/torch_ops/defs.cc 147 caffe2/onnx/torch_ops/operator_sets.h 39 caffe2/onnx/torch_ops/schema.cc 16 caffe2/onnx/torch_ops/schema.h 6 caffe2/operators/abs_op.cc 85 caffe2/operators/abs_op.cu 52 caffe2/operators/abs_op.h 27 caffe2/operators/accumulate_op.cc 26 caffe2/operators/accumulate_op.cu 5 caffe2/operators/accumulate_op.h 32 caffe2/operators/accuracy_op.cc 70 caffe2/operators/accuracy_op.cu 70 caffe2/operators/accuracy_op.h 20 caffe2/operators/acos_op.cc 63 caffe2/operators/acos_op.cu 52 caffe2/operators/acos_op.h 27 caffe2/operators/activation_ops_cudnn.h 118 caffe2/operators/affine_channel_op.cc 171 caffe2/operators/affine_channel_op.cu 135 caffe2/operators/affine_channel_op.h 103 caffe2/operators/alias_with_name.cc 20 caffe2/operators/alias_with_name.cu 8 caffe2/operators/alias_with_name.h 34 caffe2/operators/apmeter_op.cc 104 caffe2/operators/apmeter_op.h 28 caffe2/operators/arg_ops.cc 199 caffe2/operators/arg_ops.cu 96 caffe2/operators/arg_ops.h 88 caffe2/operators/asin_op.cc 63 caffe2/operators/asin_op.cu 52 caffe2/operators/asin_op.h 27 caffe2/operators/assert_op.cc 46 caffe2/operators/assert_op.cu 5 caffe2/operators/assert_op.h 41 caffe2/operators/async_net_barrier_op.cc 43 caffe2/operators/async_net_barrier_op.cu 5 caffe2/operators/async_net_barrier_op.h 17 caffe2/operators/atan_op.cc 63 caffe2/operators/atan_op.cu 50 caffe2/operators/atan_op.h 27 caffe2/operators/atomic_ops.cc 152 caffe2/operators/batch_box_cox_op.cc 300 caffe2/operators/batch_box_cox_op.h 79 caffe2/operators/batch_bucketize_op.cc 107 caffe2/operators/batch_bucketize_op.h 20 caffe2/operators/batch_gather_ops.cc 55 caffe2/operators/batch_gather_ops.cu 123 caffe2/operators/batch_gather_ops.h 133 caffe2/operators/batch_matmul_op.cc 239 caffe2/operators/batch_matmul_op.cu 19 caffe2/operators/batch_matmul_op.h 311 caffe2/operators/batch_moments_op.cc 107 caffe2/operators/batch_moments_op.cu 143 caffe2/operators/batch_moments_op.h 100 caffe2/operators/batch_permutation_op.cc 144 caffe2/operators/batch_permutation_op.cu 102 caffe2/operators/batch_permutation_op.h 28 caffe2/operators/batch_sparse_to_dense_op.cc 206 caffe2/operators/batch_sparse_to_dense_op.cu 136 caffe2/operators/batch_sparse_to_dense_op.h 174 caffe2/operators/bbox_transform_op.cc 179 caffe2/operators/bbox_transform_op.h 50 caffe2/operators/bisect_percentile_op.cc 82 caffe2/operators/bisect_percentile_op.h 145 caffe2/operators/boolean_mask_ops.cc 515 caffe2/operators/boolean_mask_ops.cu 400 caffe2/operators/boolean_mask_ops.h 71 caffe2/operators/boolean_unmask_ops.cc 122 caffe2/operators/boolean_unmask_ops.cu 111 caffe2/operators/boolean_unmask_ops.h 14 caffe2/operators/box_with_nms_limit_op.cc 285 caffe2/operators/box_with_nms_limit_op.h 87 caffe2/operators/bucketize_op.cc 60 caffe2/operators/bucketize_op.cu 51 caffe2/operators/bucketize_op.h 37 caffe2/operators/byte_weight_dequant_op.cc 6 caffe2/operators/byte_weight_dequant_op.h 46 caffe2/operators/cast_op.cc 195 caffe2/operators/cast_op.cu 124 caffe2/operators/cast_op.h 43 caffe2/operators/cbrt_op.cc 62 caffe2/operators/cbrt_op.cu 50 caffe2/operators/cbrt_op.h 27 caffe2/operators/cc_bmm_bg_op.cc 9 caffe2/operators/cc_bmm_bg_op.h 118 caffe2/operators/ceil_op.cc 49 caffe2/operators/ceil_op.cu 26 caffe2/operators/ceil_op.h 26 caffe2/operators/channel_backprop_stats_op.cc 72 caffe2/operators/channel_backprop_stats_op.cu 186 caffe2/operators/channel_backprop_stats_op.h 25 caffe2/operators/channel_shuffle_op.cc 146 caffe2/operators/channel_shuffle_op.cu 195 caffe2/operators/channel_shuffle_op.h 52 caffe2/operators/channel_stats_op.cc 70 caffe2/operators/channel_stats_op.cu 111 caffe2/operators/channel_stats_op.h 48 caffe2/operators/channelwise_conv3d_op_cudnn.cu 548 caffe2/operators/clip_op.cc 108 caffe2/operators/clip_op.cu 73 caffe2/operators/clip_op.h 52 caffe2/operators/collect_and_distribute_fpn_rpn_proposals_op.cc 501 caffe2/operators/collect_and_distribute_fpn_rpn_proposals_op.h 142 caffe2/operators/communicator_op.cc 187 caffe2/operators/communicator_op_gpu.cc 13 caffe2/operators/concat_split_op.cc 522 caffe2/operators/concat_split_op.h 347 caffe2/operators/concat_split_op_gpu.cc 9 caffe2/operators/conditional_op.cc 53 caffe2/operators/conditional_op.h 17 caffe2/operators/conv_gradient_op.cc 124 caffe2/operators/conv_op.cc 127 caffe2/operators/conv_op.h 79 caffe2/operators/conv_op_cache_cudnn.cc 10 caffe2/operators/conv_op_cache_cudnn.h 45 caffe2/operators/conv_op_cudnn.cc 1351 caffe2/operators/conv_op_eigen.cc 176 caffe2/operators/conv_op_gpu.cc 13 caffe2/operators/conv_op_impl.h 848 caffe2/operators/conv_op_shared.cc 29 caffe2/operators/conv_op_shared.h 12 caffe2/operators/conv_op_shared_gpu.cc 23 caffe2/operators/conv_pool_op_base.h 763 caffe2/operators/conv_transpose_gradient_op.cc 44 caffe2/operators/conv_transpose_op.cc 96 caffe2/operators/conv_transpose_op.h 43 caffe2/operators/conv_transpose_op_cudnn.cc 770 caffe2/operators/conv_transpose_op_gpu.cc 9 caffe2/operators/conv_transpose_op_impl.h 588 caffe2/operators/conv_transpose_op_mobile.cc 10 caffe2/operators/conv_transpose_op_mobile.h 32 caffe2/operators/conv_transpose_op_mobile_impl.h 523 caffe2/operators/conv_transpose_unpool_op_base.h 271 caffe2/operators/copy_op.cc 176 caffe2/operators/copy_op.cu 49 caffe2/operators/copy_op.h 35 caffe2/operators/copy_rows_to_tensor_op.cc 62 caffe2/operators/copy_rows_to_tensor_op.h 71 caffe2/operators/cos_op.cc 80 caffe2/operators/cos_op.cu 50 caffe2/operators/cos_op.h 27 caffe2/operators/cosh_op.cc 87 caffe2/operators/cosh_op.cu 52 caffe2/operators/cosh_op.h 27 caffe2/operators/cosine_embedding_criterion_op.cc 70 caffe2/operators/cosine_embedding_criterion_op.cu 58 caffe2/operators/cosine_embedding_criterion_op.h 32 caffe2/operators/counter_ops.cc 247 caffe2/operators/counter_ops.h 135 caffe2/operators/counter_ops_gpu.cc 12 caffe2/operators/crash_op.cc 19 caffe2/operators/create_scope_op.cc 33 caffe2/operators/create_scope_op.h 148 caffe2/operators/crf_viterbi_op.cc 195 caffe2/operators/cross_entropy_op.cc 594 caffe2/operators/cross_entropy_op.cu 415 caffe2/operators/cross_entropy_op.h 121 caffe2/operators/ctc_beam_search_decoder_op.cc 156 caffe2/operators/ctc_beam_search_decoder_op.h 29 caffe2/operators/ctc_greedy_decoder_op.cc 82 caffe2/operators/ctc_greedy_decoder_op.h 23 caffe2/operators/cube_op.cc 57 caffe2/operators/cube_op.cu 47 caffe2/operators/cube_op.h 27 caffe2/operators/data_couple.cc 13 caffe2/operators/data_couple.h 15 caffe2/operators/data_couple_gpu.cu 5 caffe2/operators/dataset_ops.cc 1347 caffe2/operators/dataset_ops.h 149 caffe2/operators/deform_conv_gradient_op.cc 49 caffe2/operators/deform_conv_op.cc 53 caffe2/operators/deform_conv_op.cu 595 caffe2/operators/deform_conv_op.h 89 caffe2/operators/deform_conv_op_impl.h 358 caffe2/operators/dense_vector_to_id_list_op.cc 24 caffe2/operators/dense_vector_to_id_list_op.h 53 caffe2/operators/depthwise_3x3_conv_op_cudnn.cu 453 caffe2/operators/distance_op.cc 636 caffe2/operators/distance_op.cu 518 caffe2/operators/distance_op.h 248 caffe2/operators/do_op.cc 29 caffe2/operators/do_op.h 159 caffe2/operators/do_op_gpu.cc 5 caffe2/operators/dropout_op.cc 160 caffe2/operators/dropout_op.cu 93 caffe2/operators/dropout_op.h 45 caffe2/operators/dropout_op_cudnn.cc 229 caffe2/operators/elementwise_add_gradient_op.cc 24 caffe2/operators/elementwise_add_op.cc 6 caffe2/operators/elementwise_add_op.h 71 caffe2/operators/elementwise_add_op_gpu.cc 13 caffe2/operators/elementwise_div_gradient_op.cc 304 caffe2/operators/elementwise_div_op.cc 6 caffe2/operators/elementwise_div_op.cu 452 caffe2/operators/elementwise_div_op.h 41 caffe2/operators/elementwise_linear_op.cc 156 caffe2/operators/elementwise_linear_op.cu 102 caffe2/operators/elementwise_linear_op.h 32 caffe2/operators/elementwise_logical_ops.cc 84 caffe2/operators/elementwise_logical_ops.h 145 caffe2/operators/elementwise_mul_gradient_op.cc 292 caffe2/operators/elementwise_mul_op.cc 6 caffe2/operators/elementwise_mul_op.cu 257 caffe2/operators/elementwise_mul_op.h 41 caffe2/operators/elementwise_ops.cc 107 caffe2/operators/elementwise_ops.cu 224 caffe2/operators/elementwise_ops.h 509 caffe2/operators/elementwise_ops_schema.cc 797 caffe2/operators/elementwise_ops_utils.cc 122 caffe2/operators/elementwise_ops_utils.h 26 caffe2/operators/elementwise_sub_gradient_op.cc 24 caffe2/operators/elementwise_sub_op.cc 6 caffe2/operators/elementwise_sub_op.h 71 caffe2/operators/elementwise_sub_op_gpu.cc 13 caffe2/operators/elementwise_sum_op.cc 83 caffe2/operators/elu_op.cc 112 caffe2/operators/elu_op.cu 81 caffe2/operators/elu_op.h 29 caffe2/operators/elu_op_cudnn.cc 95 caffe2/operators/enforce_finite_op.cc 18 caffe2/operators/enforce_finite_op.cu 12 caffe2/operators/enforce_finite_op.h 61 caffe2/operators/ensure_clipped_op.cc 41 caffe2/operators/ensure_clipped_op.h 47 caffe2/operators/ensure_cpu_output_op.cc 25 caffe2/operators/ensure_cpu_output_op.cu 7 caffe2/operators/ensure_cpu_output_op.h 43 caffe2/operators/erf_op.cc 63 caffe2/operators/erf_op.cu 52 caffe2/operators/erf_op.h 28 caffe2/operators/exp_op.cc 67 caffe2/operators/exp_op.h 15 caffe2/operators/exp_op_gpu.cc 10 caffe2/operators/expand_op.cc 47 caffe2/operators/expand_op.h 104 caffe2/operators/expand_op_gpu.cc 14 caffe2/operators/expand_squeeze_dims_op.cc 146 caffe2/operators/expand_squeeze_dims_op.h 107 caffe2/operators/expand_squeeze_dims_op_gpu.cc 6 caffe2/operators/fc_inference.cc 116 caffe2/operators/fc_inference.h 23 caffe2/operators/feature_maps_ops.cc 371 caffe2/operators/feature_maps_ops.h 760 caffe2/operators/feed_blob_op.cc 15 caffe2/operators/feed_blob_op.h 26 caffe2/operators/filler_op.cc 556 caffe2/operators/filler_op.cu 62 caffe2/operators/filler_op.h 510 caffe2/operators/find_duplicate_elements_op.cc 46 caffe2/operators/find_duplicate_elements_op.h 45 caffe2/operators/find_op.cc 22 caffe2/operators/find_op.cu 50 caffe2/operators/find_op.h 61 caffe2/operators/flatten_op.cc 64 caffe2/operators/flatten_op.h 52 caffe2/operators/flexible_top_k.cc 152 caffe2/operators/flexible_top_k.h 26 caffe2/operators/floor_op.cc 51 caffe2/operators/floor_op.cu 26 caffe2/operators/floor_op.h 26 caffe2/operators/free_op.cc 13 caffe2/operators/free_op.h 20 caffe2/operators/free_op_gpu.cc 5 caffe2/operators/fully_connected_op.cc 135 caffe2/operators/fully_connected_op.h 287 caffe2/operators/fully_connected_op_gpu.cc 193 caffe2/operators/fused_rowwise_8bit_conversion_ops.cc 323 caffe2/operators/fused_rowwise_8bit_conversion_ops.h 155 caffe2/operators/fused_rowwise_nbit_conversion_ops.cc 285 caffe2/operators/fused_rowwise_nbit_conversion_ops.h 182 caffe2/operators/fused_rowwise_nbitfake_conversion_ops.cc 238 caffe2/operators/fused_rowwise_nbitfake_conversion_ops.h 106 caffe2/operators/fused_rowwise_random_quantization_ops.cc 171 caffe2/operators/fused_rowwise_random_quantization_ops.h 78 caffe2/operators/gather_fused_8bit_rowwise_op.cc 37 caffe2/operators/gather_fused_8bit_rowwise_op.h 49 caffe2/operators/gather_op.cc 106 caffe2/operators/gather_op.cu 18 caffe2/operators/gather_op.cuh 103 caffe2/operators/gather_op.h 163 caffe2/operators/gather_ranges_to_dense_op.cc 102 caffe2/operators/gather_ranges_to_dense_op.h 203 caffe2/operators/gelu_op.cc 113 caffe2/operators/gelu_op.cu 140 caffe2/operators/gelu_op.h 46 caffe2/operators/generate_proposals_op.cc 323 caffe2/operators/generate_proposals_op.cu 620 caffe2/operators/generate_proposals_op.h 116 caffe2/operators/generate_proposals_op_util_boxes.h 259 caffe2/operators/generate_proposals_op_util_nms.h 549 caffe2/operators/generate_proposals_op_util_nms_gpu.cu 544 caffe2/operators/generate_proposals_op_util_nms_gpu.h 42 caffe2/operators/given_tensor_byte_string_to_uint8_fill_op.cc 55 caffe2/operators/given_tensor_byte_string_to_uint8_fill_op.cu 7 caffe2/operators/given_tensor_byte_string_to_uint8_fill_op.h 64 caffe2/operators/given_tensor_fill_op.cc 210 caffe2/operators/given_tensor_fill_op.cu 18 caffe2/operators/given_tensor_fill_op.h 80 caffe2/operators/glu_op.cc 51 caffe2/operators/glu_op.cu 38 caffe2/operators/glu_op.h 48 caffe2/operators/group_norm_op.cc 442 caffe2/operators/group_norm_op.cu 561 caffe2/operators/group_norm_op.h 288 caffe2/operators/gru_unit_op.cc 51 caffe2/operators/gru_unit_op.h 197 caffe2/operators/gru_unit_op_gpu.cu 132 caffe2/operators/h_softmax_op.cc 534 caffe2/operators/h_softmax_op.h 160 caffe2/operators/half_float_ops.cc 159 caffe2/operators/half_float_ops.cu 77 caffe2/operators/half_float_ops.h 80 caffe2/operators/hard_sigmoid_op.cc 119 caffe2/operators/hard_sigmoid_op.cu 82 caffe2/operators/hard_sigmoid_op.h 30 caffe2/operators/heatmap_max_keypoint_op.cc 127 caffe2/operators/heatmap_max_keypoint_op.h 25 caffe2/operators/hip/activation_ops_miopen.h 127 caffe2/operators/histogram_op.cc 27 caffe2/operators/histogram_op.h 71 caffe2/operators/if_op.cc 16 caffe2/operators/if_op.h 49 caffe2/operators/if_op_gpu.cc 5 caffe2/operators/im2col_op.cc 94 caffe2/operators/im2col_op.h 275 caffe2/operators/im2col_op_gpu.cc 6 caffe2/operators/index_hash_ops.cc 32 caffe2/operators/index_hash_ops.h 60 caffe2/operators/index_ops.cc 309 caffe2/operators/index_ops.h 104 caffe2/operators/inference_lstm_op.cc 55 caffe2/operators/inference_lstm_op.h 276 caffe2/operators/instance_norm_gradient_op.cc 249 caffe2/operators/instance_norm_op.cc 178 caffe2/operators/instance_norm_op.cu 410 caffe2/operators/instance_norm_op.h 255 caffe2/operators/integral_image_op.cc 123 caffe2/operators/integral_image_op.cu 205 caffe2/operators/integral_image_op.h 29 caffe2/operators/is_empty_op.cc 51 caffe2/operators/is_empty_op.h 19 caffe2/operators/jsd_op.cc 85 caffe2/operators/jsd_op.h 23 caffe2/operators/key_split_ops.cc 8 caffe2/operators/key_split_ops.h 48 caffe2/operators/last_n_window_collector.cc 181 caffe2/operators/layer_norm_op.cc 270 caffe2/operators/layer_norm_op.cu 348 caffe2/operators/layer_norm_op.h 260 caffe2/operators/leaky_relu_op.cc 95 caffe2/operators/leaky_relu_op.cu 61 caffe2/operators/leaky_relu_op.h 38 caffe2/operators/length_split_op.cc 29 caffe2/operators/length_split_op.h 60 caffe2/operators/lengths_pad_op.cc 55 caffe2/operators/lengths_pad_op.cu 5 caffe2/operators/lengths_pad_op.h 67 caffe2/operators/lengths_reducer_fused_8bit_rowwise_ops.cc 106 caffe2/operators/lengths_reducer_fused_8bit_rowwise_ops.h 154 caffe2/operators/lengths_reducer_fused_nbit_rowwise_ops.cc 604 caffe2/operators/lengths_reducer_fused_nbit_rowwise_ops.h 615 caffe2/operators/lengths_reducer_ops.cc 184 caffe2/operators/lengths_reducer_ops.h 597 caffe2/operators/lengths_reducer_rowwise_8bit_ops.cc 218 caffe2/operators/lengths_reducer_rowwise_8bit_ops.h 160 caffe2/operators/lengths_tile_op.cc 79 caffe2/operators/lengths_tile_op.cu 98 caffe2/operators/lengths_tile_op.h 21 caffe2/operators/lengths_top_k_op.cc 133 caffe2/operators/lengths_top_k_op.h 42 caffe2/operators/listwise_l2r_op.cc 203 caffe2/operators/listwise_l2r_op.h 53 caffe2/operators/load_save_op.cc 438 caffe2/operators/load_save_op.h 366 caffe2/operators/load_save_op_gpu.cc 16 caffe2/operators/load_save_op_util.cc 128 caffe2/operators/load_save_op_util.h 46 caffe2/operators/local_response_normalization_op.cc 448 caffe2/operators/local_response_normalization_op.cu 315 caffe2/operators/local_response_normalization_op.h 77 caffe2/operators/local_response_normalization_op_cudnn.cc 172 caffe2/operators/locally_connected_op.cc 135 caffe2/operators/locally_connected_op.h 99 caffe2/operators/locally_connected_op_gpu.cc 21 caffe2/operators/locally_connected_op_impl.h 814 caffe2/operators/locally_connected_op_util.cc 56 caffe2/operators/locally_connected_op_util.h 52 caffe2/operators/log1p_op.cc 62 caffe2/operators/log1p_op.cu 50 caffe2/operators/log1p_op.h 27 caffe2/operators/log_op.cc 66 caffe2/operators/log_op.h 15 caffe2/operators/log_op_gpu.cc 10 caffe2/operators/logit_op.cc 84 caffe2/operators/logit_op.cu 65 caffe2/operators/logit_op.h 34 caffe2/operators/loss_op.cc 52 caffe2/operators/loss_op.cu 8 caffe2/operators/loss_op.h 27 caffe2/operators/lp_pool_op.cc 264 caffe2/operators/lp_pool_op.cu 323 caffe2/operators/lpnorm_op.cc 138 caffe2/operators/lpnorm_op.cu 7 caffe2/operators/lpnorm_op.h 40 caffe2/operators/lstm_unit_op.cc 48 caffe2/operators/lstm_unit_op.h 199 caffe2/operators/lstm_unit_op_gpu.cu 262 caffe2/operators/lstm_utils.h 287 caffe2/operators/map_ops.cc 62 caffe2/operators/map_ops.h 224 caffe2/operators/margin_loss_l2r_op.cc 132 caffe2/operators/margin_loss_l2r_op.h 41 caffe2/operators/margin_ranking_criterion_op.cc 94 caffe2/operators/margin_ranking_criterion_op.cu 75 caffe2/operators/margin_ranking_criterion_op.h 32 caffe2/operators/matmul_op.cc 192 caffe2/operators/matmul_op.h 85 caffe2/operators/matmul_op_gpu.cc 5 caffe2/operators/max_pool_with_index.cu 230 caffe2/operators/max_pool_with_index_gpu.h 31 caffe2/operators/mean_op.cc 76 caffe2/operators/mean_op.cu 6 caffe2/operators/mean_op.h 105 caffe2/operators/mem_query_op.cu 39 caffe2/operators/merge_id_lists_op.cc 31 caffe2/operators/merge_id_lists_op.h 63 caffe2/operators/minmax_gradient_ops.cc 54 caffe2/operators/minmax_ops.cc 122 caffe2/operators/minmax_ops.cu 49 caffe2/operators/minmax_ops.h 124 caffe2/operators/mish_op.cc 81 caffe2/operators/mish_op.h 26 caffe2/operators/mod_op.cc 74 caffe2/operators/mod_op.cu 51 caffe2/operators/mod_op.h 31 caffe2/operators/moments_op.cc 70 caffe2/operators/moments_op.cu 111 caffe2/operators/moments_op.h 124 caffe2/operators/multi_class_accuracy_op.cc 73 caffe2/operators/multi_class_accuracy_op.cu 65 caffe2/operators/multi_class_accuracy_op.h 16 caffe2/operators/negate_gradient_op.cc 21 caffe2/operators/negate_gradient_op.h 19 caffe2/operators/negate_gradient_op_gpu.cc 5 caffe2/operators/negative_op.cc 59 caffe2/operators/negative_op.h 15 caffe2/operators/negative_op_gpu.cc 10 caffe2/operators/ngram_ops.cc 10 caffe2/operators/ngram_ops.h 71 caffe2/operators/no_default_engine_op.h 21 caffe2/operators/norm_planar_yuv_op.cc 47 caffe2/operators/normalize_l1_op.cc 35 caffe2/operators/normalize_l1_op.h 30 caffe2/operators/normalize_op.cc 57 caffe2/operators/normalize_op.h 87 caffe2/operators/normalize_ops.cu 167 caffe2/operators/numpy_tile_op.cc 14 caffe2/operators/numpy_tile_op.h 82 caffe2/operators/one_hot_ops.cc 310 caffe2/operators/one_hot_ops.cu 32 caffe2/operators/one_hot_ops.h 77 caffe2/operators/onnx_while_op.cc 69 caffe2/operators/onnx_while_op.h 248 caffe2/operators/op_utils_cudnn.h 33 caffe2/operators/operator_fallback_gpu.h 66 caffe2/operators/order_switch_ops.cc 78 caffe2/operators/order_switch_ops.h 75 caffe2/operators/order_switch_ops_cudnn.cc 147 caffe2/operators/order_switch_ops_gpu.cc 6 caffe2/operators/pack_rnn_sequence_op.cc 88 caffe2/operators/pack_rnn_sequence_op.h 71 caffe2/operators/pack_segments.cc 234 caffe2/operators/pack_segments.cu 314 caffe2/operators/pack_segments.h 79 caffe2/operators/pad_op.cc 402 caffe2/operators/pad_op.h 71 caffe2/operators/pad_op_gpu.cu 532 caffe2/operators/partition_ops.cc 135 caffe2/operators/partition_ops.h 264 caffe2/operators/percentile_op.cc 127 caffe2/operators/percentile_op.h 26 caffe2/operators/perplexity_op.cc 31 caffe2/operators/perplexity_op.cu 37 caffe2/operators/perplexity_op.h 14 caffe2/operators/piecewise_linear_transform_op.cc 91 caffe2/operators/piecewise_linear_transform_op.cu 260 caffe2/operators/piecewise_linear_transform_op.h 209 caffe2/operators/pool_gradient_op.cc 891 caffe2/operators/pool_op.cc 1018 caffe2/operators/pool_op.cu 2151 caffe2/operators/pool_op.h 288 caffe2/operators/pool_op_cudnn.cc 507 caffe2/operators/pool_op_util.cc 302 caffe2/operators/pool_op_util.h 57 caffe2/operators/pow_op.cc 270 caffe2/operators/pow_op.cu 90 caffe2/operators/pow_op.h 123 caffe2/operators/prefetch_op.h 103 caffe2/operators/prelu_op.cc 280 caffe2/operators/prelu_op.cu 257 caffe2/operators/prelu_op.h 33 caffe2/operators/prepend_dim_op.cc 37 caffe2/operators/prepend_dim_op.h 77 caffe2/operators/prepend_dim_op_gpu.cc 6 caffe2/operators/quant_decode_op.cc 54 caffe2/operators/quant_decode_op.h 137 caffe2/operators/quantile_op.cc 15 caffe2/operators/quantile_op.h 132 caffe2/operators/quantized/init_qnnpack.cc 12 caffe2/operators/quantized/int8_add_op.cc 57 caffe2/operators/quantized/int8_add_op.h 92 caffe2/operators/quantized/int8_average_pool_op.cc 57 caffe2/operators/quantized/int8_average_pool_op.h 147 caffe2/operators/quantized/int8_channel_shuffle_op.cc 10 caffe2/operators/quantized/int8_channel_shuffle_op.h 89 caffe2/operators/quantized/int8_concat_op.cc 22 caffe2/operators/quantized/int8_concat_op.h 80 caffe2/operators/quantized/int8_conv_op.cc 72 caffe2/operators/quantized/int8_conv_op.h 153 caffe2/operators/quantized/int8_conv_op_relu.cc 4 caffe2/operators/quantized/int8_conv_transpose_op.cc 45 caffe2/operators/quantized/int8_conv_transpose_op.h 146 caffe2/operators/quantized/int8_dequantize_op.cc 10 caffe2/operators/quantized/int8_dequantize_op.h 41 caffe2/operators/quantized/int8_fc_op.cc 45 caffe2/operators/quantized/int8_fc_op.h 114 caffe2/operators/quantized/int8_flatten_op.cc 28 caffe2/operators/quantized/int8_flatten_op.h 39 caffe2/operators/quantized/int8_given_tensor_fill_op.cc 29 caffe2/operators/quantized/int8_given_tensor_fill_op.h 108 caffe2/operators/quantized/int8_leaky_relu_op.cc 20 caffe2/operators/quantized/int8_leaky_relu_op.h 92 caffe2/operators/quantized/int8_max_pool_op.cc 55 caffe2/operators/quantized/int8_max_pool_op.h 94 caffe2/operators/quantized/int8_quantize_op.cc 24 caffe2/operators/quantized/int8_quantize_op.h 70 caffe2/operators/quantized/int8_relu_op.cc 29 caffe2/operators/quantized/int8_relu_op.h 83 caffe2/operators/quantized/int8_reshape_op.cc 25 caffe2/operators/quantized/int8_reshape_op.h 39 caffe2/operators/quantized/int8_resize_nearest_op.cc 21 caffe2/operators/quantized/int8_resize_nearest_op.h 68 caffe2/operators/quantized/int8_roi_align_op.cc 41 caffe2/operators/quantized/int8_roi_align_op.h 296 caffe2/operators/quantized/int8_sigmoid_op.cc 28 caffe2/operators/quantized/int8_sigmoid_op.h 86 caffe2/operators/quantized/int8_simd.h 10 caffe2/operators/quantized/int8_slice_op.cc 41 caffe2/operators/quantized/int8_slice_op.h 80 caffe2/operators/quantized/int8_softmax_op.cc 41 caffe2/operators/quantized/int8_softmax_op.h 82 caffe2/operators/quantized/int8_test_utils.h 101 caffe2/operators/quantized/int8_transpose_op.cc 23 caffe2/operators/quantized/int8_transpose_op.h 30 caffe2/operators/quantized/int8_utils.h 113 caffe2/operators/rank_loss_op.cc 178 caffe2/operators/rank_loss_op.h 27 caffe2/operators/reciprocal_gradient_op.cc 43 caffe2/operators/reciprocal_op.cc 51 caffe2/operators/reciprocal_op.cu 50 caffe2/operators/reciprocal_op.h 26 caffe2/operators/reduce_front_back_max_ops.cc 227 caffe2/operators/reduce_front_back_max_ops.cu 172 caffe2/operators/reduce_front_back_max_ops.h 113 caffe2/operators/reduce_front_back_mean_ops.cc 232 caffe2/operators/reduce_front_back_sum_mean_ops.cu 251 caffe2/operators/reduce_front_back_sum_mean_ops.h 131 caffe2/operators/reduce_front_back_sum_ops.cc 226 caffe2/operators/reduce_ops.cc 471 caffe2/operators/reduce_ops.cu 168 caffe2/operators/reduce_ops.h 352 caffe2/operators/reducer_functors.h 718 caffe2/operators/reduction_ops.cc 297 caffe2/operators/reduction_ops.cu 124 caffe2/operators/reduction_ops.h 160 caffe2/operators/relu_n_op.cc 89 caffe2/operators/relu_n_op.cu 79 caffe2/operators/relu_n_op.h 33 caffe2/operators/relu_op.cc 131 caffe2/operators/relu_op.cu 213 caffe2/operators/relu_op.h 23 caffe2/operators/remove_data_blocks_op.cc 21 caffe2/operators/remove_data_blocks_op.h 71 caffe2/operators/replace_nan_op.cc 31 caffe2/operators/replace_nan_op.cu 32 caffe2/operators/replace_nan_op.h 32 caffe2/operators/reservoir_sampling.cc 229 caffe2/operators/reshape_op.cc 161 caffe2/operators/reshape_op.h 148 caffe2/operators/reshape_op_gpu.cc 6 caffe2/operators/resize_3d_op.cc 189 caffe2/operators/resize_3d_op.cu 192 caffe2/operators/resize_3d_op.h 75 caffe2/operators/resize_op.cc 305 caffe2/operators/resize_op.cu 157 caffe2/operators/resize_op.h 65 caffe2/operators/reverse_packed_segs_op.cc 30 caffe2/operators/reverse_packed_segs_op.cu 70 caffe2/operators/reverse_packed_segs_op.h 75 caffe2/operators/rmac_regions_op.cc 97 caffe2/operators/rmac_regions_op.cu 194 caffe2/operators/rmac_regions_op.h 21 caffe2/operators/rms_norm_op.cc 153 caffe2/operators/rms_norm_op.cu 173 caffe2/operators/rms_norm_op.h 89 caffe2/operators/rnn/hip/recurrent_op_miopen.h 113 caffe2/operators/rnn/recurrent_network_blob_fetcher_op.cc 24 caffe2/operators/rnn/recurrent_network_blob_fetcher_op.h 55 caffe2/operators/rnn/recurrent_network_blob_fetcher_op_gpu.cc 7 caffe2/operators/rnn/recurrent_network_executor.cc 162 caffe2/operators/rnn/recurrent_network_executor.h 381 caffe2/operators/rnn/recurrent_network_executor_gpu.cc 117 caffe2/operators/rnn/recurrent_network_executor_gpu.h 58 caffe2/operators/rnn/recurrent_network_executor_incl.h 53 caffe2/operators/rnn/recurrent_network_op.cc 218 caffe2/operators/rnn/recurrent_network_op.h 758 caffe2/operators/rnn/recurrent_network_op_gpu.cu 79 caffe2/operators/rnn/recurrent_op_cudnn.cc 524 caffe2/operators/rnn/recurrent_op_cudnn.h 125 caffe2/operators/roi_align_gradient_op.cc 220 caffe2/operators/roi_align_gradient_op.cu 198 caffe2/operators/roi_align_gradient_op.h 39 caffe2/operators/roi_align_op.cc 281 caffe2/operators/roi_align_op.cu 165 caffe2/operators/roi_align_op.h 81 caffe2/operators/roi_align_rotated_gradient_op.cc 25 caffe2/operators/roi_align_rotated_gradient_op.cu 202 caffe2/operators/roi_align_rotated_gradient_op.h 37 caffe2/operators/roi_align_rotated_op.cc 358 caffe2/operators/roi_align_rotated_op.cu 178 caffe2/operators/roi_align_rotated_op.h 43 caffe2/operators/roi_pool_op.cc 163 caffe2/operators/roi_pool_op.cu 179 caffe2/operators/roi_pool_op.h 69 caffe2/operators/rowmul_op.cc 45 caffe2/operators/rowmul_op.h 58 caffe2/operators/rsqrt_op.cc 59 caffe2/operators/rsqrt_op.cu 52 caffe2/operators/rsqrt_op.h 27 caffe2/operators/scale_blobs_op.cc 15 caffe2/operators/scale_blobs_op.cu 155 caffe2/operators/scale_blobs_op.h 49 caffe2/operators/scale_op.cc 22 caffe2/operators/scale_op.h 34 caffe2/operators/scale_op_gpu.cc 9 caffe2/operators/segment_reduction_op.cc 495 caffe2/operators/segment_reduction_op.h 1612 caffe2/operators/segment_reduction_op_gpu.cu 1614 caffe2/operators/segment_reduction_op_gpu.cuh 100 caffe2/operators/self_binning_histogram_op.cc 42 caffe2/operators/self_binning_histogram_op.h 151 caffe2/operators/selu_op.cc 109 caffe2/operators/selu_op.cu 68 caffe2/operators/selu_op.h 45 caffe2/operators/sequence_ops.cc 481 caffe2/operators/sequence_ops.cu 343 caffe2/operators/sequence_ops.h 221 caffe2/operators/shape_op.cc 58 caffe2/operators/shape_op.h 42 caffe2/operators/shape_op_gpu.cc 5 caffe2/operators/sigmoid_gradient_op.cc 43 caffe2/operators/sigmoid_op.cc 64 caffe2/operators/sigmoid_op.cu 75 caffe2/operators/sigmoid_op.h 23 caffe2/operators/sigmoid_op_cudnn.cc 8 caffe2/operators/sin_op.cc 80 caffe2/operators/sin_op.cu 50 caffe2/operators/sin_op.h 27 caffe2/operators/sinh_op.cc 87 caffe2/operators/sinh_op.cu 52 caffe2/operators/sinh_op.h 27 caffe2/operators/sinusoid_position_encoding_op.cc 30 caffe2/operators/sinusoid_position_encoding_op.h 71 caffe2/operators/slice_op.cc 118 caffe2/operators/slice_op.cu 307 caffe2/operators/slice_op.h 285 caffe2/operators/softmax_op.cc 151 caffe2/operators/softmax_op.h 38 caffe2/operators/softmax_op_cudnn.cc 131 caffe2/operators/softmax_ops.cu 784 caffe2/operators/softmax_utils.cc 34 caffe2/operators/softmax_utils.h 18 caffe2/operators/softmax_with_loss_op.cc 309 caffe2/operators/softmax_with_loss_op.h 75 caffe2/operators/softplus_op.cc 88 caffe2/operators/softplus_op.cu 57 caffe2/operators/softplus_op.h 25 caffe2/operators/softsign_op.cc 115 caffe2/operators/softsign_op.cu 77 caffe2/operators/softsign_op.h 24 caffe2/operators/space_batch_op.cc 86 caffe2/operators/space_batch_op.h 188 caffe2/operators/space_batch_op_gpu.cu 166 caffe2/operators/sparse_dropout_with_replacement_op.cc 108 caffe2/operators/sparse_dropout_with_replacement_op.h 27 caffe2/operators/sparse_itemwise_dropout_with_replacement_op.cc 102 caffe2/operators/sparse_itemwise_dropout_with_replacement_op.h 27 caffe2/operators/sparse_lp_regularizer_op.cc 92 caffe2/operators/sparse_lp_regularizer_op.h 36 caffe2/operators/sparse_lp_regularizer_op_gpu.cu 6 caffe2/operators/sparse_normalize_op.cc 147 caffe2/operators/sparse_normalize_op.h 26 caffe2/operators/sparse_normalize_op_gpu.cu 8 caffe2/operators/sparse_to_dense_mask_op.cc 120 caffe2/operators/sparse_to_dense_mask_op.h 257 caffe2/operators/sparse_to_dense_op.cc 63 caffe2/operators/sparse_to_dense_op.cu 63 caffe2/operators/sparse_to_dense_op.h 107 caffe2/operators/spatial_batch_norm_gradient_op.cc 169 caffe2/operators/spatial_batch_norm_op.cc 123 caffe2/operators/spatial_batch_norm_op.cu 6 caffe2/operators/spatial_batch_norm_op.h 442 caffe2/operators/spatial_batch_norm_op_cudnn.cu 447 caffe2/operators/spatial_batch_norm_op_impl.cuh 446 caffe2/operators/spatial_softmax_with_loss_op.cc 216 caffe2/operators/spatial_softmax_with_loss_op.h 59 caffe2/operators/sqr_op.cc 69 caffe2/operators/sqr_op.h 15 caffe2/operators/sqr_op_gpu.cc 10 caffe2/operators/sqrt_op.cc 72 caffe2/operators/sqrt_op.h 15 caffe2/operators/sqrt_op_gpu.cc 10 caffe2/operators/square_root_divide_op.cc 35 caffe2/operators/square_root_divide_op.h 52 caffe2/operators/stats_ops.cc 286 caffe2/operators/stats_put_ops.cc 96 caffe2/operators/stats_put_ops.h 76 caffe2/operators/stop_gradient.cc 15 caffe2/operators/stop_gradient.h 20 caffe2/operators/stop_gradient_gpu.cc 5 caffe2/operators/string_ops.cc 191 caffe2/operators/string_ops.h 56 caffe2/operators/stump_func_op.cc 74 caffe2/operators/stump_func_op.cu 47 caffe2/operators/stump_func_op.h 35 caffe2/operators/stylizer_ops.cc 430 caffe2/operators/summarize_op.cc 58 caffe2/operators/summarize_op.cu 97 caffe2/operators/summarize_op.h 46 caffe2/operators/swish_op.cc 75 caffe2/operators/swish_op.cu 78 caffe2/operators/swish_op.h 26 caffe2/operators/tan_op.cc 57 caffe2/operators/tan_op.cu 52 caffe2/operators/tan_op.h 27 caffe2/operators/tanh_gradient_op.cc 43 caffe2/operators/tanh_op.cc 73 caffe2/operators/tanh_op.cu 50 caffe2/operators/tanh_op.h 27 caffe2/operators/tanh_op_cudnn.cc 8 caffe2/operators/tensor_protos_db_input.cc 29 caffe2/operators/tensor_protos_db_input.h 89 caffe2/operators/tensor_protos_db_input_gpu.cc 6 caffe2/operators/text_file_reader.cc 164 caffe2/operators/text_file_reader_utils.cc 106 caffe2/operators/text_file_reader_utils.h 100 caffe2/operators/thresholded_relu_op.cc 67 caffe2/operators/thresholded_relu_op.cu 56 caffe2/operators/thresholded_relu_op.h 35 caffe2/operators/tile_op.cc 158 caffe2/operators/tile_op.cu 92 caffe2/operators/tile_op.h 207 caffe2/operators/top_k.cc 322 caffe2/operators/top_k.cu 324 caffe2/operators/top_k.h 37 caffe2/operators/top_k_heap_selection.cuh 251 caffe2/operators/top_k_radix_selection.cuh 384 caffe2/operators/transpose_op.cc 95 caffe2/operators/transpose_op.cu 5 caffe2/operators/transpose_op.h 64 caffe2/operators/transpose_op_cudnn.cc 146 caffe2/operators/tt_linear_op.cc 56 caffe2/operators/tt_linear_op.h 151 caffe2/operators/unique_ops.cc 74 caffe2/operators/unique_ops.cu 122 caffe2/operators/unique_ops.h 29 caffe2/operators/unsafe_coalesce.cc 23 caffe2/operators/unsafe_coalesce.cu 5 caffe2/operators/unsafe_coalesce.h 54 caffe2/operators/upsample_op.cc 165 caffe2/operators/upsample_op.cu 226 caffe2/operators/upsample_op.h 51 caffe2/operators/utility_ops.cc 907 caffe2/operators/utility_ops.cu 319 caffe2/operators/utility_ops.h 1250 caffe2/operators/variable_length_sequence_padding.cc 21 caffe2/operators/variable_length_sequence_padding.h 45 caffe2/operators/weighted_multi_sampling_op.cc 104 caffe2/operators/weighted_multi_sampling_op.h 20 caffe2/operators/weighted_sample_op.cc 99 caffe2/operators/weighted_sample_op.cu 87 caffe2/operators/weighted_sample_op.h 21 caffe2/operators/while_op.cc 18 caffe2/operators/while_op.h 59 caffe2/operators/while_op_gpu.cc 5 caffe2/operators/workspace_ops.cc 37 caffe2/operators/zero_gradient_op.cc 23 caffe2/operators/zero_gradient_op.h 14 caffe2/operators/zero_gradient_op_gpu.cc 5 caffe2/opt/annotations.cc 90 caffe2/opt/annotations.h 53 caffe2/opt/backend_cutting.cc 382 caffe2/opt/backend_cutting.h 19 caffe2/opt/backend_transformer_base.cc 157 caffe2/opt/backend_transformer_base.h 63 caffe2/opt/bound_shape_inferencer.cc 1021 caffe2/opt/bound_shape_inferencer.h 120 caffe2/opt/converter.cc 511 caffe2/opt/converter.h 53 caffe2/opt/custom/cc_amrc.cc 7 caffe2/opt/custom/cc_amrc.h 157 caffe2/opt/custom/concat_elim.cc 154 caffe2/opt/custom/concat_elim.h 13 caffe2/opt/custom/concat_elim_test_net.pb 2367 caffe2/opt/custom/converter.cc 313 caffe2/opt/custom/freeze_quantization_params.cc 55 caffe2/opt/custom/freeze_quantization_params.h 6 caffe2/opt/custom/in_batch_broadcast.cc 150 caffe2/opt/custom/in_batch_broadcast.h 17 caffe2/opt/custom/pointwise_elim.cc 32 caffe2/opt/custom/pointwise_elim.h 9 caffe2/opt/dead_code_elim.cc 33 caffe2/opt/device.cc 114 caffe2/opt/device.h 11 caffe2/opt/distributed.cc 90 caffe2/opt/distributed.h 15 caffe2/opt/distributed_converter.cc 11 caffe2/opt/fakefp16_transform.cc 270 caffe2/opt/fakefp16_transform.h 16 caffe2/opt/fusion.cc 95 caffe2/opt/fusion.h 73 caffe2/opt/glow_net_transform.cc 202 caffe2/opt/glow_net_transform.h 34 caffe2/opt/mobile.cc 101 caffe2/opt/mobile.h 11 caffe2/opt/nql/ast.h 68 caffe2/opt/nql/graphmatcher.cc 222 caffe2/opt/nql/graphmatcher.h 85 caffe2/opt/nql/lex.ll 38 caffe2/opt/nql/parse.yy 145 caffe2/opt/onnx_convert.h 29 caffe2/opt/onnxifi_op.cc 752 caffe2/opt/onnxifi_op.h 396 caffe2/opt/onnxifi_transformer.cc 1264 caffe2/opt/onnxifi_transformer.h 119 caffe2/opt/optimize_ideep.cc 828 caffe2/opt/optimize_ideep.h 13 caffe2/opt/optimizer.cc 40 caffe2/opt/optimizer.h 12 caffe2/opt/passes.cc 9 caffe2/opt/passes.h 53 caffe2/opt/shape_info.cc 212 caffe2/opt/shape_info.h 123 caffe2/opt/tvm_transformer.cc 265 caffe2/opt/tvm_transformer.h 58 caffe2/perfkernels/__init__.py 1 caffe2/perfkernels/adagrad.cc 161 caffe2/perfkernels/adagrad.h 139 caffe2/perfkernels/adagrad_avx2.cc 105 caffe2/perfkernels/common.h 56 caffe2/perfkernels/common_avx.cc 15 caffe2/perfkernels/common_avx2.cc 15 caffe2/perfkernels/common_avx512.cc 17 caffe2/perfkernels/cvtsh_ss_bugfix.h 55 caffe2/perfkernels/embedding_lookup.cc 206 caffe2/perfkernels/embedding_lookup.h 21 caffe2/perfkernels/embedding_lookup_avx2.cc 2912 caffe2/perfkernels/embedding_lookup_fused_8bit_rowwise_avx2.cc 2902 caffe2/perfkernels/embedding_lookup_fused_8bit_rowwise_idx_avx2.cc 2968 caffe2/perfkernels/embedding_lookup_idx.cc 211 caffe2/perfkernels/embedding_lookup_idx.h 21 caffe2/perfkernels/embedding_lookup_idx_avx2.cc 2972 caffe2/perfkernels/fused_8bit_rowwise_embedding_lookup.cc 190 caffe2/perfkernels/fused_8bit_rowwise_embedding_lookup.h 20 caffe2/perfkernels/fused_8bit_rowwise_embedding_lookup_idx.cc 193 caffe2/perfkernels/fused_8bit_rowwise_embedding_lookup_idx.h 20 caffe2/perfkernels/fused_nbit_rowwise_conversion.cc 181 caffe2/perfkernels/fused_nbit_rowwise_conversion.h 27 caffe2/perfkernels/hp_emblookup_codegen.py 459 caffe2/perfkernels/lstm_unit_cpu-impl.h 149 caffe2/perfkernels/lstm_unit_cpu.h 64 caffe2/perfkernels/lstm_unit_cpu_avx2.cc 113 caffe2/perfkernels/lstm_unit_cpu_common.cc 115 caffe2/perfkernels/lstm_unit_cpu_common.h 64 caffe2/perfkernels/math.h 17 caffe2/perfkernels/math_cpu_avx2.cc 193 caffe2/perfkernels/math_cpu_base.cc 139 caffe2/perfkernels/typed_axpy.cc 73 caffe2/perfkernels/typed_axpy.h 5 caffe2/perfkernels/typed_axpy_avx.cc 52 caffe2/perfkernels/typed_axpy_avx2.cc 77 caffe2/predictor/InferenceGraph.h 12 caffe2/predictor/ThreadLocalPtr.cc 55 caffe2/predictor/ThreadLocalPtr.h 84 caffe2/predictor/emulator/benchmark.cc 55 caffe2/predictor/emulator/benchmark.h 31 caffe2/predictor/emulator/data_filler.cc 217 caffe2/predictor/emulator/data_filler.h 104 caffe2/predictor/emulator/emulator.h 12 caffe2/predictor/emulator/net_supplier.h 70 caffe2/predictor/emulator/output_formatter.h 14 caffe2/predictor/emulator/profiler.h 12 caffe2/predictor/emulator/std_output_formatter.h 36 caffe2/predictor/emulator/time_profiler.h 14 caffe2/predictor/emulator/utils.h 50 caffe2/predictor/predictor.cc 109 caffe2/predictor/predictor.h 39 caffe2/predictor/predictor_config.cc 82 caffe2/predictor/predictor_config.h 29 caffe2/predictor/predictor_utils.cc 65 caffe2/predictor/predictor_utils.h 19 caffe2/predictor/transforms.cc 263 caffe2/predictor/transforms.h 7 caffe2/proto/__init__.py 5 caffe2/proto/caffe2.proto 463 caffe2/proto/caffe2_legacy.proto 45 caffe2/proto/caffe2_legacy_pb2.pyi 50 caffe2/proto/caffe2_pb.h 125 caffe2/proto/caffe2_pb2.pyi 685 caffe2/proto/gen_proto_typestubs_helper.py 13 caffe2/proto/hsm.proto 54 caffe2/proto/hsm_pb2.pyi 90 caffe2/proto/metanet.proto 42 caffe2/proto/metanet_pb2.pyi 134 caffe2/proto/predictor_consts.proto 29 caffe2/proto/predictor_consts_pb2.pyi 56 caffe2/proto/prof_dag.proto 57 caffe2/proto/prof_dag_pb2.pyi 106 caffe2/proto/torch.proto 84 caffe2/proto/torch_pb.h 6 caffe2/proto/torch_pb2.pyi 186 caffe2/python/__init__.py 77 caffe2/python/_import_c_extension.py 37 caffe2/python/_import_c_extension.pyi 179 caffe2/python/attention.py 336 caffe2/python/benchmark_generator.py 107 caffe2/python/benchmarks/concat_benchmark.py 24 caffe2/python/benchmarks/fused_rowwise_nbit_conversion_bench.py 37 caffe2/python/benchmarks/sparse_lengths_sum_nbit_benchmark.py 95 caffe2/python/benchmarks/sparse_normalize_benchmark.py 96 caffe2/python/binarysize.py 104 caffe2/python/brew.py 117 caffe2/python/build.py 3 caffe2/python/cached_reader.py 47 caffe2/python/caffe_translator.py 738 caffe2/python/checkpoint.py 401 caffe2/python/cnn.py 186 caffe2/python/context.py 67 caffe2/python/context.pyi 8 caffe2/python/control.py 274 caffe2/python/control_ops_grad.py 496 caffe2/python/control_ops_util.py 167 caffe2/python/convert.py 1 caffe2/python/convnet_benchmarks.py 625 caffe2/python/core.py 1443 caffe2/python/crf.py 236 caffe2/python/crf_predict.py 25 caffe2/python/data_parallel_model.py 1765 caffe2/python/data_workers.py 375 caffe2/python/dataio.py 300 caffe2/python/dataset.py 179 caffe2/python/db_file_reader.py 96 caffe2/python/device_checker.py 82 caffe2/python/dlpack.h 70 caffe2/python/dyndep.py 19 caffe2/python/embedding_generation_benchmark.py 148 caffe2/python/experiment_util.py 80 caffe2/python/extension_loader.py 15 caffe2/python/fakefp16_transform_lib.py 8 caffe2/python/fakelowp/__init__.py 1 caffe2/python/fakelowp/init_shared_libs.py 6 caffe2/python/functional.py 84 caffe2/python/gradient_checker.py 282 caffe2/python/gru_cell.py 138 caffe2/python/helpers/__init__.py 1 caffe2/python/helpers/algebra.py 21 caffe2/python/helpers/arg_scope.py 26 caffe2/python/helpers/array_helpers.py 11 caffe2/python/helpers/control_ops.py 15 caffe2/python/helpers/conv.py 312 caffe2/python/helpers/db_input.py 10 caffe2/python/helpers/dropout.py 8 caffe2/python/helpers/elementwise_linear.py 33 caffe2/python/helpers/fc.py 163 caffe2/python/helpers/nonlinearity.py 25 caffe2/python/helpers/normalization.py 250 caffe2/python/helpers/pooling.py 21 caffe2/python/helpers/quantization.py 4 caffe2/python/helpers/tools.py 22 caffe2/python/helpers/train.py 50 caffe2/python/hip_test_util.py 4 caffe2/python/hsm_util.py 53 caffe2/python/hypothesis_test_util.py 556 caffe2/python/ideep/__init__.py 1 caffe2/python/ideep/transform_ideep_net.py 241 caffe2/python/ideep_test_util.py 17 caffe2/python/layer_model_helper.py 570 caffe2/python/layer_model_instantiator.py 84 caffe2/python/layer_test_util.py 91 caffe2/python/layers/__init__.py 22 caffe2/python/layers/adaptive_weight.py 124 caffe2/python/layers/add_bias.py 29 caffe2/python/layers/arc_cosine_feature_map.py 108 caffe2/python/layers/batch_huber_loss.py 90 caffe2/python/layers/batch_lr_loss.py 281 caffe2/python/layers/batch_mse_loss.py 60 caffe2/python/layers/batch_normalization.py 87 caffe2/python/layers/batch_sigmoid_cross_entropy_loss.py 35 caffe2/python/layers/batch_softmax_loss.py 103 caffe2/python/layers/blob_weighted_sum.py 52 caffe2/python/layers/bpr_loss.py 33 caffe2/python/layers/bucket_weighted.py 55 caffe2/python/layers/build_index.py 46 caffe2/python/layers/concat.py 85 caffe2/python/layers/constant_weight.py 34 caffe2/python/layers/conv.py 92 caffe2/python/layers/dropout.py 33 caffe2/python/layers/fc.py 175 caffe2/python/layers/fc_with_bootstrap.py 226 caffe2/python/layers/fc_without_bias.py 47 caffe2/python/layers/feature_sparse_to_dense.py 259 caffe2/python/layers/functional.py 90 caffe2/python/layers/gather_record.py 51 caffe2/python/layers/homotopy_weight.py 103 caffe2/python/layers/label_smooth.py 65 caffe2/python/layers/last_n_window_collector.py 52 caffe2/python/layers/layer_normalization.py 97 caffe2/python/layers/layers.py 344 caffe2/python/layers/margin_rank_loss.py 50 caffe2/python/layers/merge_id_lists.py 28 caffe2/python/layers/pairwise_similarity.py 79 caffe2/python/layers/position_weighted.py 45 caffe2/python/layers/random_fourier_features.py 51 caffe2/python/layers/reservoir_sampling.py 69 caffe2/python/layers/sampling_train.py 53 caffe2/python/layers/sampling_trainable_mixin.py 28 caffe2/python/layers/select_record_by_context.py 53 caffe2/python/layers/semi_random_features.py 84 caffe2/python/layers/sparse_dropout_with_replacement.py 62 caffe2/python/layers/sparse_feature_hash.py 88 caffe2/python/layers/sparse_itemwise_dropout_with_replacement.py 68 caffe2/python/layers/sparse_lookup.py 455 caffe2/python/layers/split.py 56 caffe2/python/layers/tags.py 52 caffe2/python/layers/uniform_sampling.py 59 caffe2/python/lazy.py 8 caffe2/python/lazy_dyndep.py 29 caffe2/python/lstm_benchmark.py 285 caffe2/python/memonger.py 790 caffe2/python/mint/__init__.py 1 caffe2/python/mint/app.py 157 caffe2/python/mint/static/css/simple-sidebar.css 96 caffe2/python/mint/templates/index.html 117 caffe2/python/mkl/__init__.py 1 caffe2/python/mkl/rewrite_graph.py 143 caffe2/python/mkl_test_util.py 21 caffe2/python/model_helper.py 473 caffe2/python/modeling/__init__.py 1 caffe2/python/modeling/compute_histogram_for_blobs.py 62 caffe2/python/modeling/compute_norm_for_blobs.py 63 caffe2/python/modeling/compute_statistics_for_blobs.py 30 caffe2/python/modeling/get_entry_from_blobs.py 40 caffe2/python/modeling/gradient_clipping.py 101 caffe2/python/modeling/initializers.py 110 caffe2/python/modeling/net_modifier.py 15 caffe2/python/modeling/parameter_info.py 34 caffe2/python/modeling/parameter_sharing.py 65 caffe2/python/models/__init__.py 1 caffe2/python/models/__sym_init__.py 11 caffe2/python/models/download.py 165 caffe2/python/models/imagenet_trainer_test_utils.py 161 caffe2/python/models/resnet.py 346 caffe2/python/models/seq2seq/__init__.py 1 caffe2/python/models/seq2seq/beam_search.py 424 caffe2/python/models/seq2seq/seq2seq_model_helper.py 60 caffe2/python/models/seq2seq/seq2seq_util.py 583 caffe2/python/models/seq2seq/train.py 682 caffe2/python/models/seq2seq/translate.py 579 caffe2/python/models/shufflenet.py 156 caffe2/python/modifier_context.py 41 caffe2/python/mpi_python.cc 38 caffe2/python/muji.py 169 caffe2/python/net_builder.py 396 caffe2/python/net_drawer.py 337 caffe2/python/net_printer.py 328 caffe2/python/nomnigraph.py 108 caffe2/python/nomnigraph_transformations.py 55 caffe2/python/normalizer.py 26 caffe2/python/normalizer_context.py 21 caffe2/python/numa_benchmark.py 52 caffe2/python/onnx/__init__.py 1 caffe2/python/onnx/backend.py 720 caffe2/python/onnx/backend_cpp_rep.py 34 caffe2/python/onnx/backend_rep.py 51 caffe2/python/onnx/error.py 2 caffe2/python/onnx/frontend.py 281 caffe2/python/onnx/helper.py 90 caffe2/python/onnx/onnxifi.py 47 caffe2/python/onnx/workspace.py 22 caffe2/python/operator_test/__init__.py 1 caffe2/python/operator_test/adagrad_test_helper.py 144 caffe2/python/operator_test/conftest.py 39 caffe2/python/operator_test/detectron_keypoints.py 149 caffe2/python/operator_test/fused_nbit_rowwise_test_helper.py 52 caffe2/python/operator_test/mul_gradient_benchmark.py 38 caffe2/python/operator_test/sparse_lengths_sum_benchmark.py 110 caffe2/python/optimizer.py 1845 caffe2/python/optimizer_context.py 34 caffe2/python/optimizer_test_util.py 176 caffe2/python/parallel_workers.py 223 caffe2/python/pipeline.py 297 caffe2/python/predictor/__init__.py 1 caffe2/python/predictor/mobile_exporter.py 72 caffe2/python/predictor/predictor_exporter.py 177 caffe2/python/predictor/predictor_py_utils.py 127 caffe2/python/predictor/serde.py 6 caffe2/python/predictor_constants.py 2 caffe2/python/pybind_state.cc 1776 caffe2/python/pybind_state.h 404 caffe2/python/pybind_state_dlpack.cc 67 caffe2/python/pybind_state_dlpack.h 112 caffe2/python/pybind_state_gpu.cc 158 caffe2/python/pybind_state_hip.cc 85 caffe2/python/pybind_state_ideep.cc 175 caffe2/python/pybind_state_int8.cc 39 caffe2/python/pybind_state_nomni.cc 516 caffe2/python/pybind_state_registry.cc 7 caffe2/python/pybind_state_registry.h 23 caffe2/python/queue_util.py 100 caffe2/python/record_queue.py 78 caffe2/python/recurrent.py 232 caffe2/python/regularizer.py 367 caffe2/python/regularizer_context.py 21 caffe2/python/rnn/__init__.py 1 caffe2/python/rnn/lstm_comparison.py 45 caffe2/python/rnn/rnn_cell_test_util.py 53 caffe2/python/rnn_cell.py 1660 caffe2/python/schema.py 800 caffe2/python/scope.py 77 caffe2/python/serialized_test/__init__.py 1 caffe2/python/serialized_test/coverage.py 95 caffe2/python/serialized_test/serialized_test_util.py 238 caffe2/python/session.py 101 caffe2/python/task.py 422 caffe2/python/text_file_reader.py 30 caffe2/python/timeout_guard.py 89 caffe2/python/transformations.py 28 caffe2/python/trt/__init__.py 1 caffe2/python/trt/transform.py 73 caffe2/python/tt_core.py 96 caffe2/python/utils.py 268 caffe2/python/visualize.py 100 caffe2/python/workspace.py 427 caffe2/quantization/__init__.py 1 caffe2/quantization/server/__init__.py 1 caffe2/quantization/server/activation_distribution_observer.cc 754 caffe2/quantization/server/activation_distribution_observer.h 174 caffe2/quantization/server/batch_matmul_dnnlowp_op.cc 591 caffe2/quantization/server/batch_matmul_dnnlowp_op.h 28 caffe2/quantization/server/batch_permutation_dnnlowp_op.cc 48 caffe2/quantization/server/batch_permutation_dnnlowp_op.h 19 caffe2/quantization/server/caffe2_dnnlowp_utils.cc 481 caffe2/quantization/server/caffe2_dnnlowp_utils.h 64 caffe2/quantization/server/channel_shuffle_dnnlowp_op.cc 103 caffe2/quantization/server/channel_shuffle_dnnlowp_op.h 26 caffe2/quantization/server/compute_equalization_scale.cc 70 caffe2/quantization/server/compute_equalization_scale.h 11 caffe2/quantization/server/concat_dnnlowp_op.cc 173 caffe2/quantization/server/concat_dnnlowp_op.h 18 caffe2/quantization/server/conv_dnnlowp_acc16_op.cc 736 caffe2/quantization/server/conv_dnnlowp_acc16_op.h 47 caffe2/quantization/server/conv_dnnlowp_op.cc 1498 caffe2/quantization/server/conv_dnnlowp_op.h 91 caffe2/quantization/server/conv_pool_dnnlowp_op_base.h 193 caffe2/quantization/server/conv_relu_op.cc 58 caffe2/quantization/server/conv_relu_op.h 29 caffe2/quantization/server/dequantize_dnnlowp_op.cc 68 caffe2/quantization/server/dequantize_dnnlowp_op.h 14 caffe2/quantization/server/dnnlowp.cc 298 caffe2/quantization/server/dnnlowp.h 127 caffe2/quantization/server/dnnlowp_op.h 214 caffe2/quantization/server/dnnlowp_partition.cc 45 caffe2/quantization/server/dnnlowp_partition.h 18 caffe2/quantization/server/dnnlowp_pybind11.pyi 35 caffe2/quantization/server/dnnlowp_test_utils.py 326 caffe2/quantization/server/dynamic_histogram.cc 162 caffe2/quantization/server/dynamic_histogram.h 38 caffe2/quantization/server/elementwise_add_dnnlowp_op.cc 164 caffe2/quantization/server/elementwise_dnnlowp_op.h 121 caffe2/quantization/server/elementwise_linear_dnnlowp_op.cc 89 caffe2/quantization/server/elementwise_linear_dnnlowp_op.h 20 caffe2/quantization/server/elementwise_mul_dnnlowp_op.cc 97 caffe2/quantization/server/elementwise_sum_benchmark.cc 30 caffe2/quantization/server/elementwise_sum_dnnlowp_op.cc 204 caffe2/quantization/server/elementwise_sum_dnnlowp_op_avx2.cc 170 caffe2/quantization/server/elementwise_sum_relu_op.cc 37 caffe2/quantization/server/fb_fc_packed_op.cc 74 caffe2/quantization/server/fb_fc_packed_op.h 149 caffe2/quantization/server/fbgemm_fp16_pack_op.cc 45 caffe2/quantization/server/fbgemm_fp16_pack_op.h 60 caffe2/quantization/server/fbgemm_pack_blob.h 24 caffe2/quantization/server/fbgemm_pack_matrix_cache.cc 62 caffe2/quantization/server/fbgemm_pack_matrix_cache.h 13 caffe2/quantization/server/fbgemm_pack_op.cc 805 caffe2/quantization/server/fbgemm_pack_op.h 135 caffe2/quantization/server/fully_connected_dnnlowp_acc16_op.cc 242 caffe2/quantization/server/fully_connected_dnnlowp_acc16_op.h 24 caffe2/quantization/server/fully_connected_dnnlowp_op.cc 832 caffe2/quantization/server/fully_connected_dnnlowp_op.h 41 caffe2/quantization/server/fully_connected_fake_lowp_op.cc 269 caffe2/quantization/server/fully_connected_fake_lowp_op.h 95 caffe2/quantization/server/fully_connected_fake_lowp_op_avx2.cc 107 caffe2/quantization/server/group_norm_dnnlowp_op.cc 639 caffe2/quantization/server/group_norm_dnnlowp_op.h 173 caffe2/quantization/server/group_norm_dnnlowp_op_avx2.cc 277 caffe2/quantization/server/im2col_dnnlowp.h 305 caffe2/quantization/server/int8_gen_quant_params.cc 35 caffe2/quantization/server/int8_gen_quant_params.h 52 caffe2/quantization/server/int8_gen_quant_params_min_max.cc 32 caffe2/quantization/server/int8_gen_quant_params_min_max.h 41 caffe2/quantization/server/int8_quant_scheme_blob_fill.cc 29 caffe2/quantization/server/int8_quant_scheme_blob_fill.h 26 caffe2/quantization/server/kl_minimization.cc 146 caffe2/quantization/server/kl_minimization.h 11 caffe2/quantization/server/kl_minimization_example.cc 41 caffe2/quantization/server/l1_minimization_example.cc 53 caffe2/quantization/server/l2_minimization.h 22 caffe2/quantization/server/l2_minimization_approx_example.cc 53 caffe2/quantization/server/l2_minimization_example.cc 53 caffe2/quantization/server/lstm_unit_dnnlowp_op.cc 294 caffe2/quantization/server/lstm_unit_dnnlowp_op.h 32 caffe2/quantization/server/mmio.h 44 caffe2/quantization/server/norm_minimization.cc 251 caffe2/quantization/server/norm_minimization_avx2.cc 110 caffe2/quantization/server/op_wrapper.h 70 caffe2/quantization/server/p99.cc 54 caffe2/quantization/server/p99_example.cc 36 caffe2/quantization/server/pool_dnnlowp_op.cc 701 caffe2/quantization/server/pool_dnnlowp_op_avx2.cc 314 caffe2/quantization/server/pool_dnnlowp_op_avx2.h 66 caffe2/quantization/server/pybind.cc 471 caffe2/quantization/server/quantization_error_minimization.h 43 caffe2/quantization/server/quantize_dnnlowp_op.cc 84 caffe2/quantization/server/quantize_dnnlowp_op.h 15 caffe2/quantization/server/relu_dnnlowp_op.cc 57 caffe2/quantization/server/relu_dnnlowp_op.h 21 caffe2/quantization/server/relu_dnnlowp_op_avx2.cc 47 caffe2/quantization/server/resize_nearest_3d_dnnlowp_op.cc 48 caffe2/quantization/server/resize_nearest_3d_dnnlowp_op.h 32 caffe2/quantization/server/resize_nearest_dnnlowp_op.cc 42 caffe2/quantization/server/resize_nearest_dnnlowp_op.h 27 caffe2/quantization/server/sigmoid.cc 24 caffe2/quantization/server/sigmoid.h 21 caffe2/quantization/server/sigmoid_dnnlowp_op.cc 32 caffe2/quantization/server/spatial_batch_norm_dnnlowp_op.cc 158 caffe2/quantization/server/spatial_batch_norm_dnnlowp_op.h 43 caffe2/quantization/server/spatial_batch_norm_dnnlowp_op_avx2.cc 135 caffe2/quantization/server/spatial_batch_norm_relu_op.cc 27 caffe2/quantization/server/tanh.cc 104 caffe2/quantization/server/tanh.h 40 caffe2/quantization/server/tanh_dnnlowp_op.cc 28 caffe2/quantization/server/transpose.cc 39 caffe2/quantization/server/transpose.h 5 caffe2/quantization/server/utility_dnnlowp_ops.cc 61 caffe2/quantization/server/utility_dnnlowp_ops.h 84 caffe2/quantization/server/utils.py 318 caffe2/queue/blobs_queue.cc 152 caffe2/queue/blobs_queue.h 54 caffe2/queue/blobs_queue_db.cc 59 caffe2/queue/blobs_queue_db.h 123 caffe2/queue/queue_ops.cc 87 caffe2/queue/queue_ops.h 238 caffe2/queue/queue_ops_gpu.cc 11 caffe2/queue/rebatching_queue.cc 177 caffe2/queue/rebatching_queue.h 44 caffe2/queue/rebatching_queue_ops.cc 63 caffe2/queue/rebatching_queue_ops.h 69 caffe2/serialize/crc.cc 11 caffe2/serialize/crc_alt.h 1037 caffe2/serialize/file_adapter.cc 67 caffe2/serialize/file_adapter.h 27 caffe2/serialize/inline_container.cc 365 caffe2/serialize/inline_container.h 100 caffe2/serialize/istream_adapter.cc 32 caffe2/serialize/istream_adapter.h 20 caffe2/serialize/read_adapter_interface.cc 6 caffe2/serialize/read_adapter_interface.h 15 caffe2/serialize/versions.h 22 caffe2/sgd/adadelta_op.cc 62 caffe2/sgd/adadelta_op.h 163 caffe2/sgd/adadelta_op_gpu.cu 155 caffe2/sgd/adagrad_fused.cc 249 caffe2/sgd/adagrad_fused.h 394 caffe2/sgd/adagrad_fused_op_gpu.cu 1530 caffe2/sgd/adagrad_fused_op_gpu.cuh 224 caffe2/sgd/adagrad_op.cc 201 caffe2/sgd/adagrad_op.h 514 caffe2/sgd/adagrad_op_gpu.cu 227 caffe2/sgd/adam_op.cc 155 caffe2/sgd/adam_op.h 730 caffe2/sgd/adam_op_gpu.cu 410 caffe2/sgd/clip_tensor_op.cc 31 caffe2/sgd/clip_tensor_op.h 51 caffe2/sgd/decay_adagrad_op.cc 43 caffe2/sgd/decay_adagrad_op.h 84 caffe2/sgd/fp16_momentum_sgd_op.cu 246 caffe2/sgd/fp16_momentum_sgd_op.h 61 caffe2/sgd/fp32_momentum_sgd_op.cu 125 caffe2/sgd/fp32_momentum_sgd_op.h 57 caffe2/sgd/ftrl_op.cc 130 caffe2/sgd/ftrl_op.h 65 caffe2/sgd/gftrl_op.cc 91 caffe2/sgd/gftrl_op.h 32 caffe2/sgd/iter_op.cc 54 caffe2/sgd/iter_op.h 77 caffe2/sgd/iter_op_gpu.cc 6 caffe2/sgd/lars_op.cc 44 caffe2/sgd/lars_op.h 75 caffe2/sgd/lars_op_gpu.cu 34 caffe2/sgd/learning_rate_adaption_op.cc 37 caffe2/sgd/learning_rate_adaption_op.h 67 caffe2/sgd/learning_rate_functors.h 413 caffe2/sgd/learning_rate_op.cc 178 caffe2/sgd/learning_rate_op.h 271 caffe2/sgd/learning_rate_op_gpu.cc 5 caffe2/sgd/math_lp.cc 36 caffe2/sgd/math_lp.h 27 caffe2/sgd/momentum_sgd_op.cc 98 caffe2/sgd/momentum_sgd_op.h 158 caffe2/sgd/momentum_sgd_op_gpu.cu 150 caffe2/sgd/rmsprop_op.cc 43 caffe2/sgd/rmsprop_op.h 58 caffe2/sgd/rmsprop_op_gpu.cu 46 caffe2/sgd/rowwise_adagrad_fused.cc 322 caffe2/sgd/rowwise_adagrad_fused.h 859 caffe2/sgd/rowwise_counter.cc 22 caffe2/sgd/rowwise_counter.h 61 caffe2/sgd/storm_op.cc 61 caffe2/sgd/storm_op.h 160 caffe2/sgd/weight_scale_op.cc 29 caffe2/sgd/weight_scale_op.h 62 caffe2/sgd/weight_scale_op_gpu.cc 39 caffe2/sgd/wngrad_op.cc 52 caffe2/sgd/wngrad_op.h 226 caffe2/sgd/yellowfin_op.cc 82 caffe2/sgd/yellowfin_op.h 256 caffe2/sgd/yellowfin_op_gpu.cu 79 caffe2/share/contrib/depthwise/depthwise3x3_conv_op.cc 470 caffe2/share/contrib/nnpack/conv_op.cc 350 caffe2/share/contrib/zstd/quant_decomp_zstd_op.cc 98 caffe2/share/contrib/zstd/quant_decomp_zstd_op.h 15 caffe2/transforms/common_subexpression_elimination.cc 106 caffe2/transforms/common_subexpression_elimination.h 28 caffe2/transforms/conv_to_nnpack_transform.cc 4 caffe2/transforms/conv_to_nnpack_transform.h 18 caffe2/transforms/pattern_net_transform.cc 198 caffe2/transforms/pattern_net_transform.h 51 caffe2/transforms/single_op_transform.cc 33 caffe2/transforms/single_op_transform.h 21 caffe2/unexported_symbols.lds 1 caffe2/utils/GpuAtomics.cuh 20 caffe2/utils/GpuBitonicSort.cuh 142 caffe2/utils/GpuDefs.cuh 132 caffe2/utils/GpuScanUtils.cuh 103 caffe2/utils/bench_utils.cc 79 caffe2/utils/bench_utils.h 8 caffe2/utils/cast.h 39 caffe2/utils/cblas.h 503 caffe2/utils/conversions.h 24 caffe2/utils/cpu_neon.h 39 caffe2/utils/cpuid.cc 72 caffe2/utils/cpuid.h 116 caffe2/utils/cub_namespace.cuh 15 caffe2/utils/eigen_utils.h 166 caffe2/utils/filler.h 112 caffe2/utils/fixed_divisor.h 99 caffe2/utils/knob_patcher.cc 97 caffe2/utils/knob_patcher.h 17 caffe2/utils/knobs.cc 45 caffe2/utils/knobs.h 7 caffe2/utils/map_utils.h 13 caffe2/utils/math-detail.h 74 caffe2/utils/math.h 369 caffe2/utils/math/broadcast.cc 63 caffe2/utils/math/broadcast.cu 100 caffe2/utils/math/broadcast.h 20 caffe2/utils/math/elementwise.cc 668 caffe2/utils/math/elementwise.cu 878 caffe2/utils/math/elementwise.h 138 caffe2/utils/math/half_utils.h 41 caffe2/utils/math/reduce.cc 662 caffe2/utils/math/reduce.cu 571 caffe2/utils/math/reduce.cuh 52 caffe2/utils/math/reduce.h 86 caffe2/utils/math/transpose.cc 241 caffe2/utils/math/transpose.cu 218 caffe2/utils/math/transpose.h 23 caffe2/utils/math/utils.cc 341 caffe2/utils/math/utils.h 133 caffe2/utils/math_cpu.cc 2441 caffe2/utils/math_gpu.cu 2801 caffe2/utils/murmur_hash3.cc 330 caffe2/utils/murmur_hash3.h 13 caffe2/utils/proto_convert.cc 4 caffe2/utils/proto_convert.h 8 caffe2/utils/proto_utils.cc 616 caffe2/utils/proto_utils.h 285 caffe2/utils/proto_wrap.cc 24 caffe2/utils/proto_wrap.h 7 caffe2/utils/signal_handler.cc 44 caffe2/utils/signal_handler.h 15 caffe2/utils/simple_queue.h 47 caffe2/utils/smart_tensor_printer.cc 58 caffe2/utils/smart_tensor_printer.h 27 caffe2/utils/string_utils.cc 98 caffe2/utils/string_utils.h 42 caffe2/utils/threadpool/ThreadPool.cc 157 caffe2/utils/threadpool/ThreadPool.h 36 caffe2/utils/threadpool/ThreadPoolCommon.h 12 caffe2/utils/threadpool/WorkersPool.h 251 caffe2/utils/threadpool/pthreadpool-cpp.cc 77 caffe2/utils/threadpool/pthreadpool-cpp.h 32 caffe2/utils/threadpool/pthreadpool.cc 338 caffe2/utils/threadpool/pthreadpool.h 121 caffe2/utils/threadpool/pthreadpool_impl.cc 74 caffe2/utils/threadpool/thread_pool_guard.cpp 10 caffe2/utils/threadpool/thread_pool_guard.h 16 caffe2/utils/zmq_helper.h 111 caffe2/version_script.lds 4 caffe2/video/optical_flow.cc 71 caffe2/video/optical_flow.h 32 caffe2/video/video_decoder.cc 634 caffe2/video/video_decoder.h 364 caffe2/video/video_input_op.cc 81 caffe2/video/video_input_op.h 882 caffe2/video/video_input_op_gpu.cc 6 caffe2/video/video_io.cc 173 caffe2/video/video_io.h 44 cmake/Allowlist.cmake 26 cmake/BuildVariables.cmake 40 cmake/Caffe2Config.cmake.in 122 cmake/Caffe2ConfigVersion.cmake.in 10 cmake/Codegen.cmake 274 cmake/DebugHelper.cmake 24 cmake/Dependencies.cmake 1814 cmake/External/EigenBLAS.cmake 46 cmake/External/nccl.cmake 79 cmake/External/nnpack.cmake 105 cmake/External/rccl.cmake 17 cmake/FlatBuffers.cmake 10 cmake/GoogleTestPatch.cmake 23 cmake/IncludeSource.cpp.in 1 cmake/Metal.cmake 41 cmake/MiscCheck.cmake 297 cmake/Modules/FindARM.cmake 72 cmake/Modules/FindAVX.cmake 70 cmake/Modules/FindAtlas.cmake 42 cmake/Modules/FindBLAS.cmake 383 cmake/Modules/FindBLIS.cmake 60 cmake/Modules/FindBenchmark.cmake 22 cmake/Modules/FindCUB.cmake 15 cmake/Modules/FindFFmpeg.cmake 58 cmake/Modules/FindFlexiBLAS.cmake 56 cmake/Modules/FindGloo.cmake 42 cmake/Modules/FindHiredis.cmake 19 cmake/Modules/FindLAPACK.cmake 218 cmake/Modules/FindLMDB.cmake 26 cmake/Modules/FindLevelDB.cmake 36 cmake/Modules/FindMAGMA.cmake 30 cmake/Modules/FindMKL.cmake 385 cmake/Modules/FindMKLDNN.cmake 83 cmake/Modules/FindMatlabMex.cmake 39 cmake/Modules/FindNCCL.cmake 80 cmake/Modules/FindNumPy.cmake 50 cmake/Modules/FindNuma.cmake 24 cmake/Modules/FindOpenBLAS.cmake 56 cmake/Modules/FindOpenMP.cmake 588 cmake/Modules/FindRocksDB.cmake 19 cmake/Modules/FindSnappy.cmake 22 cmake/Modules/FindVSX.cmake 33 cmake/Modules/FindZMQ.cmake 26 cmake/Modules/FindZVECTOR.cmake 51 cmake/Modules/Findpybind11.cmake 14 cmake/Modules/FindvecLib.cmake 29 cmake/Modules_CUDA_fix/FindCUDA.cmake 9 cmake/Modules_CUDA_fix/FindCUDNN.cmake 40 cmake/Modules_CUDA_fix/upstream/CMakeInitializeConfigs.cmake 35 cmake/Modules_CUDA_fix/upstream/FindCUDA.cmake 1839 cmake/Modules_CUDA_fix/upstream/FindCUDA/make2cmake.cmake 87 cmake/Modules_CUDA_fix/upstream/FindCUDA/parse_cubin.cmake 84 cmake/Modules_CUDA_fix/upstream/FindCUDA/run_nvcc.cmake 276 cmake/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake 257 cmake/Modules_CUDA_fix/upstream/FindPackageHandleStandardArgs.cmake 319 cmake/Modules_CUDA_fix/upstream/FindPackageMessage.cmake 44 cmake/ProtoBuf.cmake 198 cmake/ProtoBufPatch.cmake 86 cmake/Summary.cmake 191 cmake/TorchConfig.cmake.in 171 cmake/TorchConfigVersion.cmake.in 10 cmake/VulkanCodegen.cmake 68 cmake/VulkanDependencies.cmake 103 cmake/cmake_uninstall.cmake.in 24 cmake/iOS.cmake 175 cmake/public/ComputeLibrary.cmake 28 cmake/public/LoadHIP.cmake 221 cmake/public/cuda.cmake 476 cmake/public/gflags.cmake 74 cmake/public/glog.cmake 62 cmake/public/mkl.cmake 10 cmake/public/mkldnn.cmake 14 cmake/public/protobuf.cmake 86 cmake/public/threads.cmake 24 cmake/public/utils.cmake 527 ios/LibTorch-Lite.h 6 ios/LibTorch.h 4 ios/TestApp/TestApp/AppDelegate.h 4 ios/TestApp/TestApp/AppDelegate.m 19 ios/TestApp/TestApp/ViewController.h 3 ios/TestApp/TestApp/ViewController.mm 5 ios/TestApp/TestApp/main.m 7 ios/TestApp/TestAppTests/TestFullJIT.mm 16 ios/TestApp/TestAppTests/TestLiteInterpreter.mm 27 ios/TestApp/benchmark/coreml_backend.py 35 ios/TestApp/benchmark/setup.rb 106 ios/TestApp/benchmark/trace_model.py 10 ios/TestApp/custom_build/custom_build.py 10 ios/TestApp/custom_build/mobilenetv2.yaml 10 modules/detectron/group_spatial_softmax_op.cc 57 modules/detectron/group_spatial_softmax_op.cu 146 modules/detectron/group_spatial_softmax_op.h 48 modules/detectron/ps_roi_pool_op.cc 83 modules/detectron/ps_roi_pool_op.cu 259 modules/detectron/ps_roi_pool_op.h 65 modules/detectron/roi_pool_f_op.cc 76 modules/detectron/roi_pool_f_op.cu 163 modules/detectron/roi_pool_f_op.h 53 modules/detectron/sample_as_op.cc 58 modules/detectron/sample_as_op.cu 95 modules/detectron/sample_as_op.h 29 modules/detectron/select_smooth_l1_loss_op.cc 84 modules/detectron/select_smooth_l1_loss_op.cu 164 modules/detectron/select_smooth_l1_loss_op.h 49 modules/detectron/sigmoid_cross_entropy_loss_op.cc 72 modules/detectron/sigmoid_cross_entropy_loss_op.cu 176 modules/detectron/sigmoid_cross_entropy_loss_op.h 50 modules/detectron/sigmoid_focal_loss_op.cc 90 modules/detectron/sigmoid_focal_loss_op.cu 153 modules/detectron/sigmoid_focal_loss_op.h 55 modules/detectron/smooth_l1_loss_op.cc 89 modules/detectron/smooth_l1_loss_op.cu 167 modules/detectron/smooth_l1_loss_op.h 47 modules/detectron/softmax_focal_loss_op.cc 76 modules/detectron/softmax_focal_loss_op.cu 216 modules/detectron/softmax_focal_loss_op.h 63 modules/detectron/spatial_narrow_as_op.cc 57 modules/detectron/spatial_narrow_as_op.cu 148 modules/detectron/spatial_narrow_as_op.h 37 modules/detectron/upsample_nearest_op.cc 60 modules/detectron/upsample_nearest_op.cu 200 modules/detectron/upsample_nearest_op.h 74 modules/module_test/module_test_dynamic.cc 18 modules/observers/macros.h 6 modules/observers/net_observer_reporter.h 25 modules/observers/net_observer_reporter_print.cc 137 modules/observers/net_observer_reporter_print.h 11 modules/observers/observer_config.cc 10 modules/observers/observer_config.h 59 modules/observers/perf_observer.cc 271 modules/observers/perf_observer.h 47 modules/rocksdb/rocksdb.cc 87 mypy_plugins/check_mypy_version.py 23 scripts/diagnose_protobuf.py 50 scripts/get_python_cmake_flags.py 7 scripts/model_zoo/update-caffe2-models.py 127 scripts/model_zoo/update-models-from-caffe2.py 249 scripts/proto.ps1 15 scripts/release_notes/categorize.py 116 scripts/release_notes/commitlist.py 143 scripts/release_notes/common.py 153 scripts/release_notes/namespace_check.py 94 scripts/xcode_build.rb 82 setup.py 691 tools/__init__.py 1 tools/actions_local_runner.py 326 tools/amd_build/build_amd.py 147 tools/autograd/__init__.py 1 tools/autograd/context.py 11 tools/autograd/deprecated.yaml 60 tools/autograd/derivatives.yaml 1722 tools/autograd/gen_annotated_fn_args.py 65 tools/autograd/gen_autograd.py 66 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/ADInplaceOrViewType.cpp 24 tools/autograd/templates/Functions.cpp 9 tools/autograd/templates/Functions.h 43 tools/autograd/templates/TraceType.cpp 23 tools/autograd/templates/VariableType.cpp 28 tools/autograd/templates/VariableType.h 42 tools/autograd/templates/annotated_fn_args.py.in 9 tools/autograd/templates/python_fft_functions.cpp 63 tools/autograd/templates/python_functions.cpp 22 tools/autograd/templates/python_functions.h 7 tools/autograd/templates/python_linalg_functions.cpp 51 tools/autograd/templates/python_nn_functions.cpp 94 tools/autograd/templates/python_return_types.cpp 46 tools/autograd/templates/python_sparse_functions.cpp 50 tools/autograd/templates/python_special_functions.cpp 62 tools/autograd/templates/python_torch_functions.cpp 64 tools/autograd/templates/python_variable_methods.cpp 1049 tools/autograd/templates/variable_factories.h 75 tools/bazel.bzl 10 tools/build_libtorch.py 15 tools/build_pytorch_libs.py 53 tools/build_variables.bzl 1266 tools/code_analyzer/gen_op_registration_allowlist.py 56 tools/code_analyzer/gen_oplist.py 131 tools/code_coverage/oss_coverage.py 15 tools/code_coverage/package/__init__.py 1 tools/code_coverage/package/oss/__init__.py 1 tools/code_coverage/package/oss/cov_json.py 18 tools/code_coverage/package/oss/init.py 125 tools/code_coverage/package/oss/run.py 21 tools/code_coverage/package/oss/utils.py 68 tools/code_coverage/package/tool/__init__.py 1 tools/code_coverage/package/tool/clang_coverage.py 145 tools/code_coverage/package/tool/gcc_coverage.py 35 tools/code_coverage/package/tool/parser/__init__.py 1 tools/code_coverage/package/tool/parser/coverage_record.py 11 tools/code_coverage/package/tool/parser/gcov_coverage_parser.py 29 tools/code_coverage/package/tool/parser/llvm_coverage_parser.py 40 tools/code_coverage/package/tool/parser/llvm_coverage_segment.py 43 tools/code_coverage/package/tool/print_report.py 192 tools/code_coverage/package/tool/summarize_jsons.py 168 tools/code_coverage/package/tool/utils.py 15 tools/code_coverage/package/util/__init__.py 1 tools/code_coverage/package/util/setting.py 42 tools/code_coverage/package/util/utils.py 100 tools/code_coverage/package/util/utils_init.py 81 tools/codegen/__init__.py 1 tools/codegen/api/__init__.py 1 tools/codegen/api/autograd.py 169 tools/codegen/api/cpp.py 238 tools/codegen/api/dispatcher.py 37 tools/codegen/api/functionalization.py 66 tools/codegen/api/lazy.py 113 tools/codegen/api/meta.py 3 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/code_template.py 54 tools/codegen/context.py 47 tools/codegen/dest/__init__.py 9 tools/codegen/dest/lazy_ir.py 160 tools/codegen/dest/lazy_ts_lowering.py 24 tools/codegen/dest/native_functions.py 45 tools/codegen/dest/register_dispatch_key.py 434 tools/codegen/gen.py 1190 tools/codegen/gen_backend_stubs.py 261 tools/codegen/gen_functionalization_type.py 185 tools/codegen/gen_lazy_tensor.py 170 tools/codegen/local.py 19 tools/codegen/model.py 1023 tools/codegen/operator_versions/__init__.py 1 tools/codegen/operator_versions/gen_mobile_upgraders.py 227 tools/codegen/selective_build/__init__.py 1 tools/codegen/selective_build/operator.py 91 tools/codegen/selective_build/selector.py 191 tools/codegen/utils.py 162 tools/config/defs.bzl 51 tools/coverage_plugins_package/pyproject.toml 6 tools/coverage_plugins_package/setup.py 24 tools/coverage_plugins_package/src/coverage_plugins/__init__.py 1 tools/coverage_plugins_package/src/coverage_plugins/jit_plugin.py 37 tools/download_mnist.py 78 tools/extract_scripts.py 76 tools/fast_nvcc/fast_nvcc.py 383 tools/fast_nvcc/wrap_nvcc.sh.in 4 tools/gdb/pytorch-gdb.py 30 tools/generate_torch_version.py 44 tools/jit/__init__.py 1 tools/jit/templates/aten_schema_declarations.cpp 5 tools/jit/templates/external_functions_codegen_template.cpp 21 tools/linter/__init__.py 1 tools/linter/adapters/circleci_linter.py 124 tools/linter/adapters/clangformat_linter.py 224 tools/linter/adapters/clangtidy_linter.py 212 tools/linter/adapters/cmake_linter.py 115 tools/linter/adapters/exec_linter.py 65 tools/linter/adapters/flake8_linter.py 299 tools/linter/adapters/grep_linter.py 176 tools/linter/adapters/mypy_linter.py 164 tools/linter/adapters/nativefunctions_linter.py 74 tools/linter/adapters/newlines_linter.py 112 tools/linter/adapters/pip_init.py 43 tools/linter/adapters/s3_init.py 142 tools/linter/adapters/update_s3.py 67 tools/linter/clang_format_all.py 98 tools/linter/clang_format_utils.py 14 tools/linter/clang_tidy/__init__.py 1 tools/linter/clang_tidy/__main__.py 173 tools/linter/clang_tidy/generate_build_files.py 52 tools/linter/clang_tidy/max_tokens_pragma.py 78 tools/linter/clang_tidy/run.py 365 tools/linter/flake8_hook.py 9 tools/linter/install/__init__.py 1 tools/linter/install/clang_tidy.py 17 tools/linter/install/download_bin.py 106 tools/linter/mypy_wrapper.py 89 tools/linter/trailing_newlines.py 25 tools/linter/translate_annotations.py 135 tools/lite_interpreter/__init__.py 1 tools/lite_interpreter/gen_selected_mobile_ops_header.py 91 tools/lldb/deploy_debugger.py 5 tools/nightly.py 470 tools/nvcc_fix_deps.py 75 tools/pyi/__init__.py 1 tools/pyi/gen_pyi.py 486 tools/render_junit.py 76 tools/rules/cu.bzl 4 tools/rules/workspace.bzl 47 tools/setup_helpers/__init__.py 15 tools/setup_helpers/cmake.py 265 tools/setup_helpers/env.py 53 tools/setup_helpers/gen.py 6 tools/setup_helpers/gen_version_header.py 49 tools/setup_helpers/generate_code.py 140 tools/setup_helpers/numpy_.py 10 tools/shared/__init__.py 2 tools/shared/cwrap_common.py 163 tools/shared/module_loader.py 9 tools/stats/__init__.py 1 tools/stats/import_test_stats.py 71 tools/stats/print_test_stats.py 896 tools/stats/s3_stat_parser.py 162 tools/stats/scribe.py 113 tools/stats/upload_binary_size_to_scuba.py 154 tools/stats/upload_sccache_stats.py 61 tools/testing/__init__.py 1 tools/testing/explicit_ci_jobs.py 89 tools/testing/modulefinder_determinator.py 137 tools/vscode_settings.py 15 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