Path Lines of Code 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