Path Lines of Code android/pytorch_android/src/androidTest/assets/test.pt 17 android/pytorch_android/src/androidTest/cpp/pytorch_jni_common_test.cpp 13 android/pytorch_android/src/androidTest/java/org/pytorch/PytorchHostTests.java 18 android/pytorch_android/src/androidTest/java/org/pytorch/PytorchInstrumentedTests.java 34 android/pytorch_android/src/androidTest/java/org/pytorch/PytorchTestBase.java 341 android/pytorch_android/src/androidTest/java/org/pytorch/suite/PytorchInstrumentedTestSuite.java 7 android/pytorch_android_torchvision/src/androidTest/java/org/pytorch/torchvision/TorchVisionInstrumentedTests.java 20 android/pytorch_android_torchvision/src/androidTest/java/org/pytorch/torchvision/suite/TorchVisionInstrumentedTestSuite.java 7 android/run_tests.sh 33 android/test_app/app/build.gradle 148 android/test_app/app/src/main/cpp/pytorch_testapp_jni.cpp 60 android/test_app/app/src/main/java/org/pytorch/testapp/CameraActivity.java 189 android/test_app/app/src/main/java/org/pytorch/testapp/Constants.java 1007 android/test_app/app/src/main/java/org/pytorch/testapp/LibtorchNativeClient.java 17 android/test_app/app/src/main/java/org/pytorch/testapp/MainActivity.java 153 android/test_app/app/src/main/java/org/pytorch/testapp/Result.java 11 android/test_app/app/src/main/java/org/pytorch/testapp/Utils.java 24 android/test_app/make_assets.py 12 android/test_app/make_assets_custom.py 11 aten/src/ATen/core/List_test.cpp 933 aten/src/ATen/core/TensorImpl_test.cpp 6 aten/src/ATen/core/boxing/KernelFunction_test.cpp 366 aten/src/ATen/core/boxing/impl/kernel_function_legacy_test.cpp 851 aten/src/ATen/core/boxing/impl/kernel_function_test.cpp 659 aten/src/ATen/core/boxing/impl/kernel_lambda_legacy_test.cpp 814 aten/src/ATen/core/boxing/impl/kernel_lambda_test.cpp 586 aten/src/ATen/core/boxing/impl/kernel_stackbased_test.cpp 150 aten/src/ATen/core/boxing/impl/make_boxed_from_unboxed_functor_test.cpp 781 aten/src/ATen/core/boxing/impl/test_helpers.h 102 aten/src/ATen/core/dispatch/CppSignature_test.cpp 30 aten/src/ATen/core/dispatch/backend_fallback_test.cpp 86 aten/src/ATen/core/op_registration/op_allowlist_test.cpp 16 aten/src/ATen/core/op_registration/op_registration_test.cpp 1683 aten/src/ATen/native/metal/mpscnn/tests/MPSCNNTests.h 57 aten/src/ATen/native/metal/mpscnn/tests/MPSCNNTests.mm 819 aten/src/ATen/native/metal/mpscnn/tests/MetalOpTestRunner.h 7 aten/src/ATen/native/metal/mpscnn/tests/MetalOpTestRunner.mm 85 aten/src/ATen/native/quantized/cpu/qnnpack/test/add-operator-tester.h 229 aten/src/ATen/native/quantized/cpu/qnnpack/test/add.cc 357 aten/src/ATen/native/quantized/cpu/qnnpack/test/average-pooling-operator-tester.h 730 aten/src/ATen/native/quantized/cpu/qnnpack/test/average-pooling.cc 1424 aten/src/ATen/native/quantized/cpu/qnnpack/test/avgpool-microkernel-tester.h 357 aten/src/ATen/native/quantized/cpu/qnnpack/test/channel-shuffle-operator-tester.h 123 aten/src/ATen/native/quantized/cpu/qnnpack/test/channel-shuffle.cc 238 aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp-microkernel-tester.h 87 aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp-operator-tester.h 140 aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp.cc 89 aten/src/ATen/native/quantized/cpu/qnnpack/test/convolution-operator-tester.h 719 aten/src/ATen/native/quantized/cpu/qnnpack/test/convolution.cc 2044 aten/src/ATen/native/quantized/cpu/qnnpack/test/deconvolution-operator-tester.h 566 aten/src/ATen/native/quantized/cpu/qnnpack/test/deconvolution.cc 461 aten/src/ATen/native/quantized/cpu/qnnpack/test/dwconv-microkernel-tester.h 483 aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-operator-tester.h 328 aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-sparse-operator-tester.h 511 aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-sparse.cc 133 aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected.cc 586 aten/src/ATen/native/quantized/cpu/qnnpack/test/gavgpool-microkernel-tester.h 246 aten/src/ATen/native/quantized/cpu/qnnpack/test/gemm-block-sparse-microkernel-tester.h 426 aten/src/ATen/native/quantized/cpu/qnnpack/test/gemm-microkernel-tester.h 1013 aten/src/ATen/native/quantized/cpu/qnnpack/test/global-average-pooling-operator-tester.h 206 aten/src/ATen/native/quantized/cpu/qnnpack/test/global-average-pooling.cc 607 aten/src/ATen/native/quantized/cpu/qnnpack/test/hardsigmoid-operator-tester.h 174 aten/src/ATen/native/quantized/cpu/qnnpack/test/hardsigmoid.cc 201 aten/src/ATen/native/quantized/cpu/qnnpack/test/hardswish-operator-tester.h 184 aten/src/ATen/native/quantized/cpu/qnnpack/test/hardswish.cc 227 aten/src/ATen/native/quantized/cpu/qnnpack/test/hgemm.cc 160 aten/src/ATen/native/quantized/cpu/qnnpack/test/leaky-relu-operator-tester.h 193 aten/src/ATen/native/quantized/cpu/qnnpack/test/leaky-relu.cc 139 aten/src/ATen/native/quantized/cpu/qnnpack/test/lut-microkernel-tester.h 65 aten/src/ATen/native/quantized/cpu/qnnpack/test/lut-norm-microkernel-tester.h 74 aten/src/ATen/native/quantized/cpu/qnnpack/test/max-pooling-operator-tester.h 656 aten/src/ATen/native/quantized/cpu/qnnpack/test/max-pooling.cc 1148 aten/src/ATen/native/quantized/cpu/qnnpack/test/maxpool-microkernel-tester.h 204 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8avgpool.cc 1843 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8conv.cc 1004 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8dwconv.cc 2365 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gavgpool.cc 1039 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gemm.cc 3504 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gemm_sparse.cc 1040 aten/src/ATen/native/quantized/cpu/qnnpack/test/q8vadd.cc 257 aten/src/ATen/native/quantized/cpu/qnnpack/test/requantization-tester.h 371 aten/src/ATen/native/quantized/cpu/qnnpack/test/requantization.cc 838 aten/src/ATen/native/quantized/cpu/qnnpack/test/rmax-microkernel-tester.h 45 aten/src/ATen/native/quantized/cpu/qnnpack/test/sconv.cc 86 aten/src/ATen/native/quantized/cpu/qnnpack/test/sgemm.cc 453 aten/src/ATen/native/quantized/cpu/qnnpack/test/sigmoid-operator-tester.h 171 aten/src/ATen/native/quantized/cpu/qnnpack/test/sigmoid.cc 201 aten/src/ATen/native/quantized/cpu/qnnpack/test/softargmax-operator-tester.h 159 aten/src/ATen/native/quantized/cpu/qnnpack/test/softargmax.cc 108 aten/src/ATen/native/quantized/cpu/qnnpack/test/tanh-operator-tester.h 171 aten/src/ATen/native/quantized/cpu/qnnpack/test/tanh.cc 201 aten/src/ATen/native/quantized/cpu/qnnpack/test/test_utils.h 20 aten/src/ATen/native/quantized/cpu/qnnpack/test/u8clamp.cc 103 aten/src/ATen/native/quantized/cpu/qnnpack/test/u8lut32norm.cc 32 aten/src/ATen/native/quantized/cpu/qnnpack/test/u8maxpool.cc 1421 aten/src/ATen/native/quantized/cpu/qnnpack/test/u8rmax.cc 53 aten/src/ATen/native/quantized/cpu/qnnpack/test/vadd-microkernel-tester.h 179 aten/src/ATen/native/quantized/cpu/qnnpack/test/x8lut.cc 29 aten/src/ATen/native/quantized/cpu/qnnpack/test/x8zip.cc 293 aten/src/ATen/native/quantized/cpu/qnnpack/test/zip-microkernel-tester.h 76 aten/src/ATen/test/Dict_test.cpp 443 aten/src/ATen/test/Dimname_test.cpp 63 aten/src/ATen/test/ExclusivelyOwned_test.cpp 74 aten/src/ATen/test/MaybeOwned_test.cpp 265 aten/src/ATen/test/NamedTensor_test.cpp 187 aten/src/ATen/test/apply_utils_test.cpp 114 aten/src/ATen/test/atest.cpp 226 aten/src/ATen/test/basic.cpp 365 aten/src/ATen/test/broadcast_test.cpp 140 aten/src/ATen/test/cpu_caching_allocator_test.cpp 42 aten/src/ATen/test/cpu_generator_test.cpp 147 aten/src/ATen/test/cpu_profiling_allocator_test.cpp 150 aten/src/ATen/test/cpu_rng_test.cpp 321 aten/src/ATen/test/cuda_apply_test.cpp 119 aten/src/ATen/test/cuda_atomic_ops_test.cu 91 aten/src/ATen/test/cuda_complex_math_test.cu 34 aten/src/ATen/test/cuda_complex_test.cu 87 aten/src/ATen/test/cuda_cub_test.cu 175 aten/src/ATen/test/cuda_cudnn_test.cpp 11 aten/src/ATen/test/cuda_distributions_test.cu 176 aten/src/ATen/test/cuda_dlconvertor_test.cpp 40 aten/src/ATen/test/cuda_generator_test.cu 242 aten/src/ATen/test/cuda_half_test.cu 87 aten/src/ATen/test/cuda_integer_divider_test.cu 170 aten/src/ATen/test/cuda_optional_test.cu 18 aten/src/ATen/test/cuda_packedtensoraccessor_test.cu 35 aten/src/ATen/test/cuda_reportMemoryUsage_test.cpp 52 aten/src/ATen/test/cuda_stream_test.cpp 317 aten/src/ATen/test/cuda_tensor_interop_test.cpp 131 aten/src/ATen/test/cuda_vectorized_test.cu 148 aten/src/ATen/test/dispatch_key_set_test.cpp 14 aten/src/ATen/test/dlconvertor_test.cpp 22 aten/src/ATen/test/extension_backend_test.cpp 55 aten/src/ATen/test/half_test.cpp 150 aten/src/ATen/test/ivalue_test.cpp 744 aten/src/ATen/test/lazy_tensor_test.cpp 17 aten/src/ATen/test/math_kernel_test.cpp 114 aten/src/ATen/test/memory_format_test.cpp 154 aten/src/ATen/test/memory_overlapping_test.cpp 64 aten/src/ATen/test/mobile_memory_cleanup.cpp 33 aten/src/ATen/test/native_test.cpp 187 aten/src/ATen/test/operator_name_test.cpp 18 aten/src/ATen/test/operators_test.cpp 41 aten/src/ATen/test/packedtensoraccessor_test.cpp 36 aten/src/ATen/test/pow_test.cpp 299 aten/src/ATen/test/quantized_test.cpp 290 aten/src/ATen/test/reduce_ops_test.cpp 21 aten/src/ATen/test/reportMemoryUsage.h 32 aten/src/ATen/test/reportMemoryUsage_test.cpp 38 aten/src/ATen/test/rng_test.h 189 aten/src/ATen/test/scalar_tensor_test.cpp 228 aten/src/ATen/test/scalar_test.cpp 143 aten/src/ATen/test/stride_properties_test.cpp 49 aten/src/ATen/test/tensor_interop_test.cpp 224 aten/src/ATen/test/tensor_iterator_test.cpp 209 aten/src/ATen/test/test_assert.h 48 aten/src/ATen/test/test_install/main.cpp 4 aten/src/ATen/test/test_parallel.cpp 97 aten/src/ATen/test/test_thread_pool_guard.cpp 37 aten/src/ATen/test/thread_init_test.cpp 33 aten/src/ATen/test/type_ptr_test.cpp 43 aten/src/ATen/test/type_test.cpp 169 aten/src/ATen/test/undefined_tensor_test.cpp 39 aten/src/ATen/test/variant_test.cpp 52 aten/src/ATen/test/vec_test_all_types.cpp 1323 aten/src/ATen/test/vec_test_all_types.h 1301 aten/src/ATen/test/verify_api_visibility.cpp 14 aten/src/ATen/test/vitals.cpp 102 aten/src/ATen/test/vmap_test.cpp 846 aten/src/ATen/test/vulkan_api_test.cpp 2068 aten/src/ATen/test/vulkan_perf_test.cpp 38 aten/src/ATen/test/vulkan_test.cpp 822 aten/src/ATen/test/weakref_test.cpp 54 aten/src/ATen/test/wrapdim_test.cpp 34 aten/src/ATen/test/xla_tensor_test.cpp 27 aten/src/ATen/test/xnnpack_test.cpp 142 aten/tools/run_tests.sh 70 aten/tools/test_install.sh 10 benchmarks/fastrnns/test_bench.py 38 benchmarks/operator_benchmark/benchmark_all_other_test.py 11 benchmarks/operator_benchmark/benchmark_all_quantized_test.py 25 benchmarks/operator_benchmark/benchmark_all_test.py 8 benchmarks/operator_benchmark/c2/add_test.py 34 benchmarks/operator_benchmark/c2/batch_box_cox_test.py 32 benchmarks/operator_benchmark/c2/batch_gather_test.py 44 benchmarks/operator_benchmark/c2/clip_ranges_test.py 37 benchmarks/operator_benchmark/c2/concat_test.py 107 benchmarks/operator_benchmark/c2/matmul_test.py 36 benchmarks/operator_benchmark/c2/quantile_op_test.py 33 benchmarks/operator_benchmark/c2/replace_nan_test.py 29 benchmarks/operator_benchmark/common/tests/add_ops_list_test.py 25 benchmarks/operator_benchmark/common/tests/c2_cpu_gpu_forward_backward_test.py 32 benchmarks/operator_benchmark/common/tests/jit_forward_test.py 25 benchmarks/operator_benchmark/common/tests/pt_backward_test.py 19 benchmarks/operator_benchmark/common/tests/pt_configs_list_test.py 25 benchmarks/operator_benchmark/common/tests/pt_cpu_gpu_forward_backward_test.py 20 benchmarks/operator_benchmark/common/tests/random_sample_test.py 24 benchmarks/operator_benchmark/pt/add_test.py 85 benchmarks/operator_benchmark/pt/as_strided_test.py 42 benchmarks/operator_benchmark/pt/batchnorm_test.py 88 benchmarks/operator_benchmark/pt/binary_test.py 77 benchmarks/operator_benchmark/pt/bmm_test.py 24 benchmarks/operator_benchmark/pt/cat_test.py 102 benchmarks/operator_benchmark/pt/channel_shuffle_test.py 44 benchmarks/operator_benchmark/pt/chunk_test.py 34 benchmarks/operator_benchmark/pt/clip_ranges_test.py 40 benchmarks/operator_benchmark/pt/conv_test.py 73 benchmarks/operator_benchmark/pt/diag_test.py 31 benchmarks/operator_benchmark/pt/embeddingbag_test.py 39 benchmarks/operator_benchmark/pt/fill_test.py 35 benchmarks/operator_benchmark/pt/gather_test.py 37 benchmarks/operator_benchmark/pt/gelu_test.py 20 benchmarks/operator_benchmark/pt/groupnorm_test.py 27 benchmarks/operator_benchmark/pt/hardsigmoid_test.py 43 benchmarks/operator_benchmark/pt/hardswish_test.py 43 benchmarks/operator_benchmark/pt/index_select_test.py 43 benchmarks/operator_benchmark/pt/instancenorm_test.py 25 benchmarks/operator_benchmark/pt/interpolate_test.py 103 benchmarks/operator_benchmark/pt/layernorm_test.py 27 benchmarks/operator_benchmark/pt/linear_prepack_fp16_test.py 32 benchmarks/operator_benchmark/pt/linear_test.py 17 benchmarks/operator_benchmark/pt/linear_unpack_fp16_test.py 34 benchmarks/operator_benchmark/pt/matmul_test.py 39 benchmarks/operator_benchmark/pt/matrix_mult_test.py 94 benchmarks/operator_benchmark/pt/nan_to_num_test.py 46 benchmarks/operator_benchmark/pt/pool_test.py 129 benchmarks/operator_benchmark/pt/qactivation_test.py 86 benchmarks/operator_benchmark/pt/qarithmetic_test.py 66 benchmarks/operator_benchmark/pt/qatembedding_ops_test.py 48 benchmarks/operator_benchmark/pt/qbatchnorm_test.py 73 benchmarks/operator_benchmark/pt/qcat_test.py 56 benchmarks/operator_benchmark/pt/qcomparators_test.py 56 benchmarks/operator_benchmark/pt/qconv_test.py 50 benchmarks/operator_benchmark/pt/qembedding_bag_lookups_test.py 203 benchmarks/operator_benchmark/pt/qembedding_pack_test.py 82 benchmarks/operator_benchmark/pt/qembeddingbag_test.py 26 benchmarks/operator_benchmark/pt/qgroupnorm_test.py 35 benchmarks/operator_benchmark/pt/qinstancenorm_test.py 33 benchmarks/operator_benchmark/pt/qinterpolate_test.py 54 benchmarks/operator_benchmark/pt/qlayernorm_test.py 35 benchmarks/operator_benchmark/pt/qlinear_test.py 37 benchmarks/operator_benchmark/pt/qobserver_test.py 117 benchmarks/operator_benchmark/pt/qpool_test.py 102 benchmarks/operator_benchmark/pt/qrnn_test.py 55 benchmarks/operator_benchmark/pt/qtensor_method_test.py 43 benchmarks/operator_benchmark/pt/quantization_test.py 266 benchmarks/operator_benchmark/pt/qunary_test.py 80 benchmarks/operator_benchmark/pt/remainder_test.py 48 benchmarks/operator_benchmark/pt/softmax_test.py 45 benchmarks/operator_benchmark/pt/split_test.py 34 benchmarks/operator_benchmark/pt/stack_test.py 83 benchmarks/operator_benchmark/pt/sum_test.py 37 benchmarks/operator_benchmark/pt/tensor_to_test.py 34 benchmarks/operator_benchmark/pt/unary_test.py 136 benchmarks/operator_benchmark/pt_extension/cpp_extension_test.py 32 benchmarks/sparse/test_csr.sh 32 benchmarks/static_runtime/test_static_module.cc 1157 benchmarks/static_runtime/test_static_runtime.cc 1899 benchmarks/static_runtime/test_utils.cc 323 benchmarks/static_runtime/test_utils.h 37 c10/cuda/test/impl/CUDATest.cpp 6 c10/test/core/CompileTimeFunctionPointer_test.cpp 63 c10/test/core/DeviceGuard_test.cpp 30 c10/test/core/Device_test.cpp 40 c10/test/core/DispatchKeySet_test.cpp 122 c10/test/core/StreamGuard_test.cpp 1 c10/test/core/impl/InlineDeviceGuard_test.cpp 156 c10/test/core/impl/InlineStreamGuard_test.cpp 191 c10/test/core/impl/SizesAndStrides_test.cpp 281 c10/test/util/Array_test.cpp 83 c10/test/util/Bitset_test.cpp 125 c10/test/util/C++17_test.cpp 114 c10/test/util/ConstexprCrc_test.cpp 11 c10/test/util/Half_test.cpp 89 c10/test/util/LeftRight_test.cpp 173 c10/test/util/Macros.h 7 c10/test/util/Metaprogramming_test.cpp 673 c10/test/util/SmallVectorTest.cpp 1025 c10/test/util/ThreadLocal_test.cpp 175 c10/test/util/TypeIndex_test.cpp 193 c10/test/util/TypeList_test.cpp 352 c10/test/util/TypeTraits_test.cpp 164 c10/test/util/accumulate_test.cpp 50 c10/test/util/bfloat16_test.cpp 119 c10/test/util/complex_math_test.cpp 4 c10/test/util/complex_math_test_common.h 460 c10/test/util/complex_test.cpp 1 c10/test/util/complex_test_common.h 561 c10/test/util/either_test.cpp 1117 c10/test/util/exception_test.cpp 84 c10/test/util/flags_test.cpp 17 c10/test/util/intrusive_ptr_test.cpp 3109 c10/test/util/irange_test.cpp 49 c10/test/util/logging_test.cpp 121 c10/test/util/optional_test.cpp 130 c10/test/util/ordered_preserving_dict_test.cpp 351 c10/test/util/registry_test.cpp 65 c10/test/util/string_view_test.cpp 1530 c10/test/util/tempfile_test.cpp 21 c10/test/util/typeid_test.cpp 106 caffe2/contrib/aten/aten_test.py 101 caffe2/contrib/fakelowp/fp16_fma_test.cc 33 caffe2/contrib/fakelowp/test/test_batchmatmul_nnpi_fp16.py 85 caffe2/contrib/fakelowp/test/test_batchnorm_nnpi_fp16.py 117 caffe2/contrib/fakelowp/test/test_chunking.py 123 caffe2/contrib/fakelowp/test/test_deq_swish_quant_nnpi.py 141 caffe2/contrib/fakelowp/test/test_fc_nnpi_fp16.py 310 caffe2/contrib/fakelowp/test/test_fusions.py 86 caffe2/contrib/fakelowp/test/test_int8_ops_nnpi.py 285 caffe2/contrib/fakelowp/test/test_int8_quant.py 93 caffe2/contrib/fakelowp/test/test_layernorm_nnpi_fp16.py 208 caffe2/contrib/fakelowp/test/test_op_nnpi_fp16.py 301 caffe2/contrib/fakelowp/test/test_sls_4bit_nnpi_fp16.py 194 caffe2/contrib/fakelowp/test/test_sls_8bit_nnpi_fp16.py 487 caffe2/contrib/fakelowp/test/test_sls_8bit_nnpi_fp32.py 233 caffe2/contrib/gloo/gloo_test.py 583 caffe2/contrib/nccl/nccl_ops_test.py 161 caffe2/contrib/nnpack/nnpack_ops_test.py 214 caffe2/contrib/opencl/context_test.cc 7 caffe2/contrib/prof/cuda_profile_ops_test.py 15 caffe2/contrib/tensorboard/tensorboard_exporter_test.py 108 caffe2/contrib/tensorboard/tensorboard_test.py 96 caffe2/contrib/warpctc/ctc_ops_test.py 89 caffe2/core/blob_gpu_test.cc 207 caffe2/core/blob_test.cc 1102 caffe2/core/common_test.cc 29 caffe2/core/context_gpu_test.cc 124 caffe2/core/context_test.cc 29 caffe2/core/event_gpu_test.cc 35 caffe2/core/event_test.cc 31 caffe2/core/graph_test.cc 166 caffe2/core/init_test.cc 56 caffe2/core/module_test.cc 56 caffe2/core/net_async_tracing_test.cc 83 caffe2/core/net_dag_utils_test.cc 267 caffe2/core/net_gpu_test.cc 109 caffe2/core/net_simple_refcount_test.cc 49 caffe2/core/net_test.cc 957 caffe2/core/nomnigraph/tests/AlgorithmsTest.cc 111 caffe2/core/nomnigraph/tests/BinaryMatchImplTest.cc 72 caffe2/core/nomnigraph/tests/GraphTest.cc 143 caffe2/core/nomnigraph/tests/MatchTest.cc 32 caffe2/core/nomnigraph/tests/NeuralNetTest.cc 57 caffe2/core/nomnigraph/tests/SubgraphMatcherTest.cc 405 caffe2/core/nomnigraph/tests/TarjansImplTest.cc 50 caffe2/core/nomnigraph/tests/TopoSortTest.cc 66 caffe2/core/nomnigraph/tests/test_util.cc 118 caffe2/core/nomnigraph/tests/test_util.h 53 caffe2/core/observer_test.cc 156 caffe2/core/operator_gpu_test.cc 52 caffe2/core/operator_schema_test.cc 251 caffe2/core/operator_test.cc 535 caffe2/core/parallel_net_test.cc 264 caffe2/core/plan_executor_test.cc 321 caffe2/core/serialization_test.cc 81 caffe2/core/stats_test.cc 128 caffe2/core/test_utils.cc 125 caffe2/core/test_utils.h 166 caffe2/core/timer_test.cc 52 caffe2/core/transform_test.cc 348 caffe2/core/workspace_test.cc 107 caffe2/db/db_test.cc 168 caffe2/distributed/file_store_handler_op_test.py 41 caffe2/distributed/redis_store_handler_op_test.py 30 caffe2/experiments/python/funhash_op_test.py 45 caffe2/experiments/python/sparse_funhash_op_test.py 41 caffe2/experiments/python/sparse_reshape_op_test.py 62 caffe2/experiments/python/tt_contraction_op_test.py 31 caffe2/experiments/python/tt_pad_op_test.py 30 caffe2/mobile/contrib/ios/mpscnn/mpscnn_test.h 7 caffe2/mobile/contrib/ios/mpscnn/mpscnn_test.mm 3091 caffe2/mobile/contrib/ios/pool_test.cc 133 caffe2/mobile/contrib/ios/resize_test.cc 78 caffe2/mobile/contrib/nnapi/nnapi_test.cc 587 caffe2/mobile/contrib/ulp2/ulp_test.cc 408 caffe2/mpi/mpi_gpu_test.cc 304 caffe2/mpi/mpi_test.cc 281 caffe2/observers/time_observer_test.cc 63 caffe2/onnx/ssa_test.cc 75 caffe2/operators/batch_matmul_op_gpu_test.cc 81 caffe2/operators/batch_matmul_op_test.cc 68 caffe2/operators/batch_permutation_op_gpu_test.cc 209 caffe2/operators/boolean_unmask_ops_test.cc 48 caffe2/operators/conv_op_cache_cudnn_test.cc 46 caffe2/operators/conv_transpose_op_mobile_test.cc 169 caffe2/operators/elementwise_op_gpu_test.cc 34 caffe2/operators/elementwise_op_test.cc 29 caffe2/operators/elementwise_op_test.h 253 caffe2/operators/generate_proposals_op_gpu_test.cc 556 caffe2/operators/generate_proposals_op_test.cc 621 caffe2/operators/generate_proposals_op_util_boxes_test.cc 109 caffe2/operators/generate_proposals_op_util_nms_gpu_test.cc 484 caffe2/operators/generate_proposals_op_util_nms_test.cc 423 caffe2/operators/half_float_ops_test.cc 84 caffe2/operators/operator_fallback_gpu_test.cc 71 caffe2/operators/quantized/int8_roi_align_op_test.cc 59 caffe2/operators/quantized/int8_test.cc 961 caffe2/operators/reshape_op_gpu_test.cc 45 caffe2/operators/roi_align_op_gpu_test.cc 236 caffe2/operators/string_ops_test.cc 112 caffe2/operators/text_file_reader_utils_test.cc 109 caffe2/operators/utility_ops_gpu_test.cc 44 caffe2/operators/utility_ops_test.cc 38 caffe2/opt/backend_cutting_test.cc 138 caffe2/opt/bound_shape_inference_test.cc 1116 caffe2/opt/converter_nomigraph_test.cc 95 caffe2/opt/custom/concat_elim_test.cc 160 caffe2/opt/custom/converter_test.cc 47 caffe2/opt/custom/in_batch_broadcast_test.cc 223 caffe2/opt/custom/test_cc_amcr_net.pb 97 caffe2/opt/dead_code_elim_test.cc 49 caffe2/opt/device_test.cc 63 caffe2/opt/distributed_test.cc 132 caffe2/opt/mobile_test.cc 42 caffe2/opt/nql/tests/GraphMatcherTest.cc 299 caffe2/opt/split_slss_test.cc 100 caffe2/predictor/emulator/data_filler_test.cc 22 caffe2/predictor/predictor_test.cc 192 caffe2/python/allcompare_test.py 62 caffe2/python/brew_test.py 265 caffe2/python/caffe_translator_test.py 73 caffe2/python/checkpoint_test.py 254 caffe2/python/context_test.py 48 caffe2/python/control_ops_grad_test.py 32 caffe2/python/control_test.py 256 caffe2/python/convert_test.py 7 caffe2/python/convnet_benchmarks_test.py 18 caffe2/python/core_gradients_test.py 761 caffe2/python/core_test.py 937 caffe2/python/crf_viterbi_test.py 35 caffe2/python/data_parallel_model_test.py 1174 caffe2/python/data_workers_test.py 143 caffe2/python/dataio_test.py 338 caffe2/python/db_test.py 33 caffe2/python/fakelowp/test_utils.py 50 caffe2/python/filler_test.py 12 caffe2/python/functional_test.py 91 caffe2/python/fused_8bit_rowwise_conversion_ops_test.py 78 caffe2/python/gradient_check_test.py 438 caffe2/python/hypothesis_test.py 2363 caffe2/python/ideep/LRN_op_test.py 34 caffe2/python/ideep/adam_op_test.py 67 caffe2/python/ideep/blobs_queue_db_test.py 90 caffe2/python/ideep/channel_shuffle_op_test.py 34 caffe2/python/ideep/concat_split_op_test.py 131 caffe2/python/ideep/conv_op_test.py 144 caffe2/python/ideep/conv_transpose_test.py 61 caffe2/python/ideep/convfusion_op_test.py 788 caffe2/python/ideep/copy_op_test.py 86 caffe2/python/ideep/dropout_op_test.py 41 caffe2/python/ideep/elementwise_sum_op_test.py 163 caffe2/python/ideep/expanddims_squeeze_op_test.py 117 caffe2/python/ideep/fc_op_test.py 321 caffe2/python/ideep/leaky_relu_op_test.py 69 caffe2/python/ideep/moment_sgd_op_test.py 48 caffe2/python/ideep/operator_fallback_op_test.py 83 caffe2/python/ideep/order_switch_op_test.py 64 caffe2/python/ideep/pool_op_test.py 126 caffe2/python/ideep/pre_convert_test.py 75 caffe2/python/ideep/relu_op_test.py 103 caffe2/python/ideep/reshape_op_test.py 110 caffe2/python/ideep/shape_op_test.py 77 caffe2/python/ideep/sigmoid_op_test.py 22 caffe2/python/ideep/softmax_op_test.py 26 caffe2/python/ideep/spatial_bn_op_test.py 111 caffe2/python/ideep/test_ideep_net.py 115 caffe2/python/ideep/transpose_op_test.py 32 caffe2/python/ideep/weightedsum_op_test.py 44 caffe2/python/layer_parameter_sharing_test.py 189 caffe2/python/layers_test.py 2154 caffe2/python/lazy_dyndep_test.py 99 caffe2/python/lengths_reducer_fused_8bit_rowwise_ops_test.py 168 caffe2/python/lengths_reducer_rowwise_8bit_ops_test.py 113 caffe2/python/memonger_test.py 693 caffe2/python/mkl/mkl_LRN_op_test.py 34 caffe2/python/mkl/mkl_LRN_speed_test.py 52 caffe2/python/mkl/mkl_concat_op_test.py 36 caffe2/python/mkl/mkl_conv_op_test.py 43 caffe2/python/mkl/mkl_copy_op_test.py 59 caffe2/python/mkl/mkl_elementwise_add_op_test.py 32 caffe2/python/mkl/mkl_elementwise_sum_op_test.py 35 caffe2/python/mkl/mkl_fc_op_test.py 25 caffe2/python/mkl/mkl_fc_speed_test.py 67 caffe2/python/mkl/mkl_fill_op_test.py 29 caffe2/python/mkl/mkl_pool_op_test.py 37 caffe2/python/mkl/mkl_pool_speed_test.py 72 caffe2/python/mkl/mkl_relu_op_test.py 27 caffe2/python/mkl/mkl_sbn_op_test.py 65 caffe2/python/mkl/mkl_sbn_speed_test.py 91 caffe2/python/mkl/mkl_sigmoid_op_test.py 23 caffe2/python/mkl/mkl_speed_test.py 49 caffe2/python/mkl/mkl_squeeze_op_test.py 28 caffe2/python/mkl/rewrite_graph_test.py 221 caffe2/python/model_device_test.py 132 caffe2/python/model_helper_test.py 59 caffe2/python/modeling/compute_histogram_for_blobs_test.py 90 caffe2/python/modeling/compute_norm_for_blobs_test.py 156 caffe2/python/modeling/compute_statistics_for_blobs_test.py 53 caffe2/python/modeling/get_entry_from_blobs_test.py 60 caffe2/python/modeling/gradient_clipping_test.py 164 caffe2/python/modeling/initializers_test.py 37 caffe2/python/modeling/parameter_sharing_test.py 120 caffe2/python/models/resnet_test.py 48 caffe2/python/models/seq2seq/seq2seq_beam_search_test.py 191 caffe2/python/models/seq2seq/seq2seq_model_helper_test.py 49 caffe2/python/models/shufflenet_test.py 48 caffe2/python/muji_test.py 69 caffe2/python/net_builder_test.py 259 caffe2/python/net_printer_test.py 78 caffe2/python/nomnigraph_test.py 372 caffe2/python/nomnigraph_transformations_test.py 112 caffe2/python/normalizer_test.py 9 caffe2/python/numa_test.py 35 caffe2/python/observer_test.py 124 caffe2/python/onnx/test_onnxifi.py 168 caffe2/python/onnx/tests/__init__.py 1 caffe2/python/onnx/tests/c2_ref_test.py 741 caffe2/python/onnx/tests/conversion_test.py 316 caffe2/python/onnx/tests/helper_test.py 19 caffe2/python/onnx/tests/onnx_backend_test.py 150 caffe2/python/onnx/tests/ssa_test.py 113 caffe2/python/onnx/tests/test_utils.py 16 caffe2/python/operator_fp_exceptions_test.py 22 caffe2/python/operator_test/activation_ops_test.py 226 caffe2/python/operator_test/adadelta_test.py 165 caffe2/python/operator_test/adagrad_test.py 219 caffe2/python/operator_test/adam_test.py 420 caffe2/python/operator_test/affine_channel_op_test.py 94 caffe2/python/operator_test/alias_with_name_test.py 24 caffe2/python/operator_test/apmeter_test.py 69 caffe2/python/operator_test/arg_ops_test.py 48 caffe2/python/operator_test/assert_test.py 20 caffe2/python/operator_test/async_net_barrier_test.py 24 caffe2/python/operator_test/atomic_ops_test.py 64 caffe2/python/operator_test/basic_rnn_test.py 112 caffe2/python/operator_test/batch_box_cox_test.py 107 caffe2/python/operator_test/batch_bucketize_op_test.py 76 caffe2/python/operator_test/batch_moments_op_test.py 77 caffe2/python/operator_test/batch_sparse_to_dense_op_test.py 96 caffe2/python/operator_test/bbox_transform_test.py 274 caffe2/python/operator_test/bisect_percentile_op_test.py 154 caffe2/python/operator_test/blobs_queue_db_test.py 75 caffe2/python/operator_test/boolean_mask_test.py 333 caffe2/python/operator_test/boolean_unmask_test.py 48 caffe2/python/operator_test/box_with_nms_limit_op_test.py 189 caffe2/python/operator_test/bucketize_op_test.py 24 caffe2/python/operator_test/cast_op_test.py 28 caffe2/python/operator_test/ceil_op_test.py 24 caffe2/python/operator_test/channel_backprop_stats_op_test.py 50 caffe2/python/operator_test/channel_shuffle_test.py 48 caffe2/python/operator_test/channel_stats_op_test.py 69 caffe2/python/operator_test/checkpoint_test.py 24 caffe2/python/operator_test/clip_op_test.py 48 caffe2/python/operator_test/clip_tensor_op_test.py 46 caffe2/python/operator_test/collect_and_distribute_fpn_rpn_proposals_op_test.py 235 caffe2/python/operator_test/concat_op_cost_test.py 72 caffe2/python/operator_test/concat_split_op_test.py 179 caffe2/python/operator_test/conditional_test.py 21 caffe2/python/operator_test/conv_test.py 916 caffe2/python/operator_test/conv_transpose_test.py 393 caffe2/python/operator_test/copy_ops_test.py 148 caffe2/python/operator_test/copy_rows_to_tensor_op_test.py 61 caffe2/python/operator_test/cosine_embedding_criterion_op_test.py 27 caffe2/python/operator_test/counter_ops_test.py 65 caffe2/python/operator_test/crf_test.py 125 caffe2/python/operator_test/cross_entropy_ops_test.py 237 caffe2/python/operator_test/ctc_beam_search_decoder_op_test.py 112 caffe2/python/operator_test/ctc_greedy_decoder_op_test.py 124 caffe2/python/operator_test/cudnn_recurrent_test.py 109 caffe2/python/operator_test/data_couple_op_test.py 19 caffe2/python/operator_test/dataset_ops_test.py 522 caffe2/python/operator_test/decay_adagrad_test.py 55 caffe2/python/operator_test/deform_conv_test.py 538 caffe2/python/operator_test/dense_vector_to_id_list_op_test.py 51 caffe2/python/operator_test/depthwise_3x3_conv_test.py 46 caffe2/python/operator_test/distance_op_test.py 87 caffe2/python/operator_test/dropout_op_test.py 50 caffe2/python/operator_test/duplicate_operands_test.py 19 caffe2/python/operator_test/elementwise_linear_op_test.py 29 caffe2/python/operator_test/elementwise_logical_ops_test.py 112 caffe2/python/operator_test/elementwise_op_broadcast_test.py 310 caffe2/python/operator_test/elementwise_ops_test.py 895 caffe2/python/operator_test/emptysample_ops_test.py 52 caffe2/python/operator_test/enforce_finite_op_test.py 39 caffe2/python/operator_test/ensure_clipped_test.py 37 caffe2/python/operator_test/ensure_cpu_output_op_test.py 36 caffe2/python/operator_test/erf_op_test.py 19 caffe2/python/operator_test/expand_op_test.py 51 caffe2/python/operator_test/fc_operator_test.py 82 caffe2/python/operator_test/feature_maps_ops_test.py 622 caffe2/python/operator_test/filler_ops_test.py 227 caffe2/python/operator_test/find_op_test.py 37 caffe2/python/operator_test/flatten_op_test.py 24 caffe2/python/operator_test/flexible_top_k_test.py 56 caffe2/python/operator_test/floor_op_test.py 24 caffe2/python/operator_test/fused_nbit_rowwise_conversion_ops_test.py 281 caffe2/python/operator_test/fused_nbit_rowwise_test.cc 95 caffe2/python/operator_test/gather_ops_test.py 191 caffe2/python/operator_test/gather_ranges_op_test.py 231 caffe2/python/operator_test/given_tensor_byte_string_to_uint8_fill_op_test.py 38 caffe2/python/operator_test/given_tensor_fill_op_test.py 35 caffe2/python/operator_test/glu_op_test.py 31 caffe2/python/operator_test/group_conv_test.py 64 caffe2/python/operator_test/group_norm_op_test.py 134 caffe2/python/operator_test/gru_test.py 318 caffe2/python/operator_test/heatmap_max_keypoint_op_test.py 111 caffe2/python/operator_test/histogram_test.py 67 caffe2/python/operator_test/hsm_test.py 201 caffe2/python/operator_test/hyperbolic_ops_test.py 33 caffe2/python/operator_test/im2col_col2im_test.py 115 caffe2/python/operator_test/image_input_op_test.py 332 caffe2/python/operator_test/index_hash_ops_test.py 58 caffe2/python/operator_test/index_ops_test.py 112 caffe2/python/operator_test/instance_norm_test.py 219 caffe2/python/operator_test/integral_image_ops_test.py 76 caffe2/python/operator_test/jsd_ops_test.py 28 caffe2/python/operator_test/key_split_ops_test.py 36 caffe2/python/operator_test/lars_test.py 34 caffe2/python/operator_test/layer_norm_op_test.py 341 caffe2/python/operator_test/leaky_relu_test.py 143 caffe2/python/operator_test/learning_rate_adaption_op_test.py 66 caffe2/python/operator_test/learning_rate_op_test.py 213 caffe2/python/operator_test/length_split_op_test.py 119 caffe2/python/operator_test/lengths_pad_op_test.py 45 caffe2/python/operator_test/lengths_reducer_fused_nbit_rowwise_ops_test.py 351 caffe2/python/operator_test/lengths_tile_op_test.py 39 caffe2/python/operator_test/lengths_top_k_ops_test.py 47 caffe2/python/operator_test/listwise_l2r_operator_test.py 214 caffe2/python/operator_test/load_save_test.py 703 caffe2/python/operator_test/locally_connected_op_test.py 196 caffe2/python/operator_test/loss_ops_test.py 29 caffe2/python/operator_test/lpnorm_op_test.py 72 caffe2/python/operator_test/map_ops_test.py 56 caffe2/python/operator_test/margin_loss_l2r_operator_test.py 81 caffe2/python/operator_test/margin_ranking_criterion_op_test.py 33 caffe2/python/operator_test/math_ops_test.py 37 caffe2/python/operator_test/matmul_op_test.py 227 caffe2/python/operator_test/mean_op_test.py 47 caffe2/python/operator_test/merge_id_lists_op_test.py 63 caffe2/python/operator_test/mkl_conv_op_test.py 41 caffe2/python/operator_test/mkl_packed_fc_op_test.py 57 caffe2/python/operator_test/mod_op_test.py 49 caffe2/python/operator_test/moments_op_test.py 42 caffe2/python/operator_test/momentum_sgd_test.py 155 caffe2/python/operator_test/mpi_test.py 170 caffe2/python/operator_test/negate_gradient_op_test.py 30 caffe2/python/operator_test/ngram_ops_test.py 63 caffe2/python/operator_test/normalize_op_test.py 44 caffe2/python/operator_test/numpy_tile_op_test.py 44 caffe2/python/operator_test/one_hot_ops_test.py 176 caffe2/python/operator_test/onnx_while_test.py 84 caffe2/python/operator_test/order_switch_test.py 31 caffe2/python/operator_test/pack_ops_test.py 329 caffe2/python/operator_test/pack_rnn_sequence_op_test.py 67 caffe2/python/operator_test/pad_test.py 39 caffe2/python/operator_test/partition_ops_test.py 148 caffe2/python/operator_test/percentile_op_test.py 113 caffe2/python/operator_test/piecewise_linear_transform_test.py 135 caffe2/python/operator_test/pooling_test.py 413 caffe2/python/operator_test/prepend_dim_test.py 33 caffe2/python/operator_test/python_op_test.py 34 caffe2/python/operator_test/quantile_test.py 62 caffe2/python/operator_test/rand_quantization_op_speed_test.py 84 caffe2/python/operator_test/rank_loss_operator_test.py 126 caffe2/python/operator_test/rebatching_queue_test.py 207 caffe2/python/operator_test/record_queue_test.py 73 caffe2/python/operator_test/recurrent_net_executor_test.py 254 caffe2/python/operator_test/recurrent_network_test.py 320 caffe2/python/operator_test/reduce_ops_test.py 368 caffe2/python/operator_test/reduction_ops_test.py 141 caffe2/python/operator_test/reshape_ops_test.py 168 caffe2/python/operator_test/resize_op_test.py 201 caffe2/python/operator_test/rmac_regions_op_test.py 71 caffe2/python/operator_test/rms_norm_op_test.py 36 caffe2/python/operator_test/rnn_cell_test.py 1612 caffe2/python/operator_test/roi_align_rotated_op_test.py 159 caffe2/python/operator_test/rowwise_counter_test.py 52 caffe2/python/operator_test/scale_op_test.py 51 caffe2/python/operator_test/segment_ops_test.py 624 caffe2/python/operator_test/self_binning_histogram_test.py 254 caffe2/python/operator_test/selu_op_test.py 78 caffe2/python/operator_test/sequence_ops_test.py 390 caffe2/python/operator_test/shape_inference_test.py 538 caffe2/python/operator_test/sinusoid_position_encoding_op_test.py 60 caffe2/python/operator_test/softmax_ops_test.py 523 caffe2/python/operator_test/softplus_op_test.py 14 caffe2/python/operator_test/sparse_dropout_with_replacement_op_test.py 59 caffe2/python/operator_test/sparse_gradient_checker_test.py 32 caffe2/python/operator_test/sparse_itemwise_dropout_with_replacement_op_test.py 59 caffe2/python/operator_test/sparse_lp_regularizer_test.py 54 caffe2/python/operator_test/sparse_normalize_test.py 77 caffe2/python/operator_test/sparse_ops_test.py 75 caffe2/python/operator_test/sparse_to_dense_mask_op_test.py 86 caffe2/python/operator_test/spatial_bn_op_test.py 441 caffe2/python/operator_test/specialized_segment_ops_test.py 283 caffe2/python/operator_test/split_op_cost_test.py 199 caffe2/python/operator_test/square_root_divide_op_test.py 57 caffe2/python/operator_test/stats_ops_test.py 36 caffe2/python/operator_test/stats_put_ops_test.py 147 caffe2/python/operator_test/storm_test.py 125 caffe2/python/operator_test/string_ops_test.py 122 caffe2/python/operator_test/text_file_reader_test.py 57 caffe2/python/operator_test/thresholded_relu_op_test.py 51 caffe2/python/operator_test/tile_op_test.py 84 caffe2/python/operator_test/top_k_test.py 196 caffe2/python/operator_test/torch_integration_test.py 954 caffe2/python/operator_test/transpose_op_test.py 63 caffe2/python/operator_test/trigonometric_op_test.py 38 caffe2/python/operator_test/unique_ops_test.py 44 caffe2/python/operator_test/unique_uniform_fill_op_test.py 42 caffe2/python/operator_test/unsafe_coalesce_test.py 60 caffe2/python/operator_test/upsample_op_test.py 156 caffe2/python/operator_test/utility_ops_test.py 407 caffe2/python/operator_test/video_input_op_test.py 243 caffe2/python/operator_test/weight_scale_test.py 28 caffe2/python/operator_test/weighted_multi_sample_test.py 53 caffe2/python/operator_test/weighted_sample_test.py 63 caffe2/python/operator_test/weighted_sum_test.py 83 caffe2/python/operator_test/wngrad_test.py 175 caffe2/python/optimizer_test.py 639 caffe2/python/parallel_workers_test.py 80 caffe2/python/parallelize_bmuf_distributed_test.py 233 caffe2/python/pipeline_test.py 62 caffe2/python/predictor/mobile_exporter_test.py 87 caffe2/python/predictor/predictor_exporter_test.py 194 caffe2/python/predictor/predictor_test.py 61 caffe2/python/python_op_test.py 193 caffe2/python/regularizer_test.py 213 caffe2/python/schema_test.py 413 caffe2/python/scope_test.py 105 caffe2/python/session_test.py 49 caffe2/python/sparse_to_dense_mask_test.py 139 caffe2/python/sparse_to_dense_test.py 91 caffe2/python/task_test.py 21 caffe2/python/test/__init__.py 1 caffe2/python/test/blob_deallocation_test.py 16 caffe2/python/test/do_op_test.py 56 caffe2/python/test/executor_test.py 77 caffe2/python/test/executor_test_util.py 219 caffe2/python/test/fakefp16_transform_test.py 18 caffe2/python/test/gpu_context_test.py 17 caffe2/python/test/inference_lstm_op_test.py 63 caffe2/python/test/python_protobuf_test.py 9 caffe2/python/test_util.py 68 caffe2/python/toy_regression_test.py 41 caffe2/python/transformations_test.py 268 caffe2/python/trt/test_pt_onnx_trt.py 148 caffe2/python/trt/test_trt.py 232 caffe2/python/tt_core_test.py 63 caffe2/python/utils_test.py 29 caffe2/python/workspace_test.py 779 caffe2/quantization/server/batch_matmul_dnnlowp_op_test.py 201 caffe2/quantization/server/batch_permutation_dnnlowp_op_test.py 29 caffe2/quantization/server/channel_shuffle_dnnlowp_op_test.py 99 caffe2/quantization/server/compute_equalization_scale_test.py 57 caffe2/quantization/server/concat_dnnlowp_op_test.py 77 caffe2/quantization/server/conv_depthwise_dnnlowp_op_test.py 284 caffe2/quantization/server/conv_dnnlowp_acc16_op_test.py 328 caffe2/quantization/server/conv_dnnlowp_op_test.py 468 caffe2/quantization/server/conv_groupwise_dnnlowp_acc16_op_test.py 280 caffe2/quantization/server/conv_groupwise_dnnlowp_op_test.py 253 caffe2/quantization/server/dequantize_dnnlowp_op_test.py 38 caffe2/quantization/server/dynamic_histogram_test.cc 60 caffe2/quantization/server/elementwise_add_dnnlowp_op_test.py 151 caffe2/quantization/server/elementwise_linear_dnnlowp_op_test.py 74 caffe2/quantization/server/elementwise_mul_dnnlowp_op_test.py 148 caffe2/quantization/server/elementwise_sum_dnnlowp_op_test.py 226 caffe2/quantization/server/fc_fake_lowp_test.cc 123 caffe2/quantization/server/fully_connected_dnnlowp_acc16_op_test.py 178 caffe2/quantization/server/fully_connected_dnnlowp_op_test.py 244 caffe2/quantization/server/fully_connected_fp16_test.py 52 caffe2/quantization/server/fully_connected_rowwise_dnnlowp_op_test.py 124 caffe2/quantization/server/gather_dnnlowp_op_test.py 62 caffe2/quantization/server/group_norm_dnnlowp_op_test.py 105 caffe2/quantization/server/int8_gen_quant_params_min_max_test.py 58 caffe2/quantization/server/int8_gen_quant_params_test.py 66 caffe2/quantization/server/int8_quant_scheme_blob_fill_test.py 27 caffe2/quantization/server/lstm_unit_dnnlowp_op_test.py 88 caffe2/quantization/server/observer_test.py 29 caffe2/quantization/server/pool_dnnlowp_op_test.py 169 caffe2/quantization/server/quantize_dnnlowp_op_test.py 56 caffe2/quantization/server/relu_dnnlowp_op_test.py 54 caffe2/quantization/server/requantization_test.cc 106 caffe2/quantization/server/resize_nearest_3d_dnnlowp_op_test.py 52 caffe2/quantization/server/resize_nearest_dnnlowp_op_test.py 47 caffe2/quantization/server/sigmoid_dnnlowp_op_test.py 53 caffe2/quantization/server/sigmoid_test.cc 37 caffe2/quantization/server/spatial_batch_norm_dnnlowp_op_test.py 103 caffe2/quantization/server/tanh_dnnlowp_op_test.py 50 caffe2/quantization/server/tanh_test.cc 38 caffe2/serialize/inline_container_test.cc 89 caffe2/share/contrib/depthwise/depthwise3x3_conv_op_test.cc 177 caffe2/share/contrib/nnpack/nnpack_test.cc 337 caffe2/test/assets/squeeze_predict_net.pb 238 caffe2/test/caffe2_gtest_main.cc 13 caffe2/transforms/common_subexpression_elimination_test.cc 62 caffe2/transforms/conv_to_nnpack_transform_test.cc 34 caffe2/transforms/pattern_net_transform_test.cc 373 caffe2/utils/cast_test.cc 31 caffe2/utils/cpuid_test.cc 7 caffe2/utils/fatal_signal_asan_no_sig_test.cc 119 caffe2/utils/fixed_divisor_test.cc 56 caffe2/utils/hip/math_blas_gpu_test.cc 341 caffe2/utils/knobs_test.cc 25 caffe2/utils/math_gpu_test.cc 388 caffe2/utils/math_test.cc 468 caffe2/utils/proto_utils_test.cc 51 caffe2/utils/simple_queue_test.cc 57 caffe2/utils/smart_tensor_printer_test.cc 37 modules/detectron/upsample_nearest_op_test.py 33 scripts/release_notes/test_release_notes.py 37 test/ao/sparsity/test_kernels.py 226 test/ao/sparsity/test_parametrization.py 115 test/ao/sparsity/test_pruner.py 324 test/ao/sparsity/test_scheduler.py 62 test/ao/sparsity/test_sparsifier.py 279 test/autograd/test_complex.py 68 test/benchmark_utils/test_benchmark_utils.py 490 test/bottleneck_test/test.py 3 test/bottleneck_test/test_args.py 9 test/bottleneck_test/test_cuda.py 20 test/cpp/__init__.py 1 test/cpp/api/any.cpp 366 test/cpp/api/autograd.cpp 954 test/cpp/api/dataloader.cpp 1820 test/cpp/api/dispatch.cpp 57 test/cpp/api/enum.cpp 82 test/cpp/api/expanding-array.cpp 50 test/cpp/api/fft.cpp 102 test/cpp/api/functional.cpp 2635 test/cpp/api/grad_mode.cpp 60 test/cpp/api/imethod.cpp 45 test/cpp/api/inference_mode.cpp 474 test/cpp/api/init.cpp 108 test/cpp/api/init_baseline.h 172 test/cpp/api/init_baseline.py 46 test/cpp/api/integration.cpp 260 test/cpp/api/jit.cpp 96 test/cpp/api/memory.cpp 28 test/cpp/api/meta_tensor.cpp 26 test/cpp/api/misc.cpp 72 test/cpp/api/module.cpp 875 test/cpp/api/moduledict.cpp 267 test/cpp/api/modulelist.cpp 246 test/cpp/api/modules.cpp 4509 test/cpp/api/namespace.cpp 7 test/cpp/api/nn_utils.cpp 661 test/cpp/api/operations.cpp 73 test/cpp/api/optim.cpp 396 test/cpp/api/optim_baseline.h 1334 test/cpp/api/optim_baseline.py 92 test/cpp/api/ordered_dict.cpp 203 test/cpp/api/parallel.cpp 235 test/cpp/api/parallel_benchmark.cpp 83 test/cpp/api/parameterdict.cpp 132 test/cpp/api/parameterlist.cpp 140 test/cpp/api/rnn.cpp 531 test/cpp/api/sequential.cpp 592 test/cpp/api/serialize.cpp 779 test/cpp/api/special.cpp 8 test/cpp/api/static.cpp 85 test/cpp/api/support.cpp 6 test/cpp/api/support.h 136 test/cpp/api/tensor.cpp 939 test/cpp/api/tensor_cuda.cpp 96 test/cpp/api/tensor_flatten.cpp 26 test/cpp/api/tensor_indexing.cpp 629 test/cpp/api/tensor_options.cpp 119 test/cpp/api/tensor_options_cuda.cpp 57 test/cpp/api/torch_include.cpp 9 test/cpp/api/transformer.cpp 1039 test/cpp/c10d/CUDATest.cu 24 test/cpp/c10d/CUDATest.hpp 13 test/cpp/c10d/FileStoreTest.cpp 122 test/cpp/c10d/HashStoreTest.cpp 78 test/cpp/c10d/ProcessGroupGlooAsyncTest.cpp 206 test/cpp/c10d/ProcessGroupGlooTest.cpp 685 test/cpp/c10d/ProcessGroupMPITest.cpp 300 test/cpp/c10d/ProcessGroupNCCLErrorsTest.cpp 203 test/cpp/c10d/ProcessGroupNCCLTest.cpp 557 test/cpp/c10d/StoreTestCommon.hpp 37 test/cpp/c10d/TCPStoreTest.cpp 301 test/cpp/c10d/TestUtils.hpp 110 test/cpp/c10d/example/allreduce.cpp 25 test/cpp/common/main.cpp 27 test/cpp/common/support.h 28 test/cpp/dist_autograd/test_dist_autograd.cpp 74 test/cpp/jit/__init__.py 1 test/cpp/jit/test_alias_analysis.cpp 1323 test/cpp/jit/test_argument_spec.cpp 135 test/cpp/jit/test_autodiff.cpp 235 test/cpp/jit/test_backend.cpp 521 test/cpp/jit/test_backend_compiler_lib.cpp 129 test/cpp/jit/test_backend_compiler_preprocess.cpp 58 test/cpp/jit/test_backend_lib.cpp 70 test/cpp/jit/test_class_import.cpp 128 test/cpp/jit/test_class_parser.cpp 30 test/cpp/jit/test_class_type.cpp 78 test/cpp/jit/test_cleanup_passes.cpp 43 test/cpp/jit/test_code_template.cpp 52 test/cpp/jit/test_concat_opt.cpp 531 test/cpp/jit/test_constant_pooling.cpp 105 test/cpp/jit/test_create_autodiff_subgraphs.cpp 19 test/cpp/jit/test_cs_debug_info_serialization.cpp 141 test/cpp/jit/test_custom_class.cpp 105 test/cpp/jit/test_custom_class_registrations.cpp 321 test/cpp/jit/test_custom_class_registrations.h 30 test/cpp/jit/test_custom_operators.cpp 203 test/cpp/jit/test_dce.cpp 40 test/cpp/jit/test_flatbuffer.cpp 840 test/cpp/jit/test_fuser.cpp 226 test/cpp/jit/test_gpu.cpp 13973 test/cpp/jit/test_gpu_shift.cpp 3178 test/cpp/jit/test_gpu_validator.h 332 test/cpp/jit/test_graph_executor.cpp 49 test/cpp/jit/test_graph_iterator.cpp 181 test/cpp/jit/test_inliner.cpp 39 test/cpp/jit/test_interface.cpp 68 test/cpp/jit/test_interpreter.cpp 218 test/cpp/jit/test_interpreter_async.pt 11 test/cpp/jit/test_ir.cpp 188 test/cpp/jit/test_irparser.cpp 339 test/cpp/jit/test_jit_logging_levels.cpp 41 test/cpp/jit/test_jit_type.cpp 54 test/cpp/jit/test_lite_interpreter.cpp 1518 test/cpp/jit/test_lite_interpreter_direct.cpp 694 test/cpp/jit/test_lite_trainer.cpp 210 test/cpp/jit/test_load_upgraders.cpp 35 test/cpp/jit/test_memory_dag.cpp 93 test/cpp/jit/test_misc.cpp 2378 test/cpp/jit/test_mobile_type_parser.cpp 192 test/cpp/jit/test_module_api.cpp 318 test/cpp/jit/test_op_replacement.cpp 134 test/cpp/jit/test_peephole_optimize.cpp 108 test/cpp/jit/test_qualified_name.cpp 57 test/cpp/jit/test_save_load.cpp 124 test/cpp/jit/test_schema_matching.cpp 80 test/cpp/jit/test_script_profile.cpp 53 test/cpp/jit/test_shape_analysis.cpp 196 test/cpp/jit/test_stack_opt.cpp 210 test/cpp/jit/test_subgraph_matcher.cpp 486 test/cpp/jit/test_subgraph_rewriter.cpp 249 test/cpp/jit/test_subgraph_utils.cpp 124 test/cpp/jit/test_union.cpp 95 test/cpp/jit/test_upgrader_utils.cpp 75 test/cpp/jit/test_utils.cpp 243 test/cpp/jit/test_utils.h 86 test/cpp/jit/tests_setup.py 78 test/cpp/jit/torch_python_test.cpp 66 test/cpp/lazy/test_backend_device.cpp 77 test/cpp/lazy/test_cache.cpp 50 test/cpp/lazy/test_ir.cpp 83 test/cpp/lazy/test_ir_util.cpp 57 test/cpp/lazy/test_misc.cpp 53 test/cpp/lazy/test_permutation_util.cpp 38 test/cpp/lazy/test_shape.cpp 65 test/cpp/lazy/test_tensor_impl.cpp 12 test/cpp/lazy/test_util.cpp 57 test/cpp/lite_interpreter_runtime/main.cpp 21 test/cpp/lite_interpreter_runtime/test_lite_interpreter_runtime.cpp 69 test/cpp/lite_interpreter_runtime/test_mobile_profiler.cpp 99 test/cpp/monitor/test_counters.cpp 279 test/cpp/monitor/test_events.cpp 27 test/cpp/rpc/e2e_test_base.cpp 15 test/cpp/rpc/e2e_test_base.h 131 test/cpp/rpc/test_e2e_tensorpipe.cpp 47 test/cpp/rpc/test_tensorpipe_serialization.cpp 150 test/cpp/rpc/test_wire_serialization.cpp 92 test/cpp/tensorexpr/gtest_assert_float_eq.h 31 test/cpp/tensorexpr/padded_buffer.cpp 31 test/cpp/tensorexpr/padded_buffer.h 203 test/cpp/tensorexpr/test_approx.cpp 76 test/cpp/tensorexpr/test_aten.cpp 860 test/cpp/tensorexpr/test_base.h 78 test/cpp/tensorexpr/test_boundsinference.cpp 714 test/cpp/tensorexpr/test_conv.cpp 185 test/cpp/tensorexpr/test_cpp_codegen.cpp 222 test/cpp/tensorexpr/test_cuda.cpp 1827 test/cpp/tensorexpr/test_dynamic_shapes.cpp 340 test/cpp/tensorexpr/test_expr.cpp 576 test/cpp/tensorexpr/test_external_calls.cpp 772 test/cpp/tensorexpr/test_graph_opt.cpp 264 test/cpp/tensorexpr/test_ir_printer.cpp 77 test/cpp/tensorexpr/test_ir_verifier.cpp 143 test/cpp/tensorexpr/test_kernel.cpp 1450 test/cpp/tensorexpr/test_llvm.cpp 1516 test/cpp/tensorexpr/test_loopnest.cpp 5071 test/cpp/tensorexpr/test_memdependency.cpp 1641 test/cpp/tensorexpr/test_memplanning.cpp 294 test/cpp/tensorexpr/test_ops.cpp 36 test/cpp/tensorexpr/test_quantization.cpp 270 test/cpp/tensorexpr/test_reductions.cpp 1492 test/cpp/tensorexpr/test_registerizer.cpp 1972 test/cpp/tensorexpr/test_simplify.cpp 3692 test/cpp/tensorexpr/test_te_fuser_pass.cpp 289 test/cpp/tensorexpr/test_type.cpp 150 test/cpp/tensorexpr/test_utils.h 65 test/cpp/tensorexpr/tutorial.cpp 173 test/cpp_api_parity/__init__.py 1 test/cpp_api_parity/functional_impl_check.py 146 test/cpp_api_parity/module_impl_check.py 174 test/cpp_api_parity/parity_table_parser.py 47 test/cpp_api_parity/sample_functional.py 41 test/cpp_api_parity/sample_module.py 56 test/cpp_api_parity/utils.py 169 test/cpp_extensions/cpp_c10d_extension.cpp 98 test/cpp_extensions/cpp_c10d_extension.hpp 86 test/cpp_extensions/cpp_frontend_extension.cpp 43 test/cpp_extensions/cublas_extension.cpp 13 test/cpp_extensions/cuda_extension.cpp 13 test/cpp_extensions/cuda_extension.cu 27 test/cpp_extensions/cuda_extension_kernel.cu 22 test/cpp_extensions/cuda_extension_kernel2.cu 22 test/cpp_extensions/cudnn_extension.cpp 46 test/cpp_extensions/cusolver_extension.cpp 12 test/cpp_extensions/dangling_impl_extension.cpp 8 test/cpp_extensions/doubler.h 15 test/cpp_extensions/extension.cpp 33 test/cpp_extensions/jit_extension.cpp 15 test/cpp_extensions/jit_extension2.cpp 5 test/cpp_extensions/no_python_abi_suffix_test/no_python_abi_suffix_test.cpp 1 test/cpp_extensions/no_python_abi_suffix_test/setup.py 9 test/cpp_extensions/ort_extension.cpp 108 test/cpp_extensions/rng_extension.cpp 55 test/cpp_extensions/self_compiler_include_dirs_test/tmp.h 1 test/cpp_extensions/setup.py 62 test/cpp_extensions/torch_library.cu 10 test/cpp_extensions/torch_test_cpp_extension/__init__.py 1 test/custom_backend/backend.py 35 test/custom_backend/custom_backend.cpp 14 test/custom_backend/custom_backend.h 68 test/custom_backend/test_custom_backend.cpp 30 test/custom_backend/test_custom_backend.py 30 test/custom_operator/model.py 32 test/custom_operator/op.cpp 63 test/custom_operator/op.h 18 test/custom_operator/test_custom_classes.py 66 test/custom_operator/test_custom_ops.cpp 156 test/custom_operator/test_custom_ops.py 70 test/delete.py 1 test/distributed/_sharded_optim/test_sharded_optim.py 145 test/distributed/_sharded_tensor/ops/test_binary_cmp.py 122 test/distributed/_sharded_tensor/ops/test_embedding.py 141 test/distributed/_sharded_tensor/ops/test_embedding_bag.py 250 test/distributed/_sharded_tensor/ops/test_init.py 105 test/distributed/_sharded_tensor/ops/test_linear.py 194 test/distributed/_sharded_tensor/test_sharded_tensor.py 1812 test/distributed/_sharding_spec/test_sharding_spec.py 213 test/distributed/algorithms/ddp_comm_hooks/test_ddp_hooks.py 156 test/distributed/algorithms/quantization/test_quantization.py 220 test/distributed/algorithms/test_join.py 366 test/distributed/argparse_util_test.py 96 test/distributed/elastic/agent/server/test/__init__.py 1 test/distributed/elastic/agent/server/test/api_test.py 468 test/distributed/elastic/agent/server/test/local_elastic_agent_test.py 847 test/distributed/elastic/events/lib_test.py 113 test/distributed/elastic/metrics/__init__.py 1 test/distributed/elastic/metrics/api_test.py 77 test/distributed/elastic/multiprocessing/api_test.py 682 test/distributed/elastic/multiprocessing/errors/api_test.py 149 test/distributed/elastic/multiprocessing/errors/error_handler_test.py 68 test/distributed/elastic/multiprocessing/redirects_test.py 97 test/distributed/elastic/multiprocessing/tail_log_test.py 70 test/distributed/elastic/rendezvous/__init__.py 1 test/distributed/elastic/rendezvous/api_test.py 195 test/distributed/elastic/rendezvous/c10d_rendezvous_backend_test.py 179 test/distributed/elastic/rendezvous/dynamic_rendezvous_test.py 974 test/distributed/elastic/rendezvous/etcd_rendezvous_backend_test.py 91 test/distributed/elastic/rendezvous/etcd_rendezvous_test.py 56 test/distributed/elastic/rendezvous/etcd_server_test.py 42 test/distributed/elastic/rendezvous/rendezvous_backend_test.py 63 test/distributed/elastic/rendezvous/static_rendezvous_test.py 82 test/distributed/elastic/rendezvous/utils_test.py 264 test/distributed/elastic/timer/__init__.py 1 test/distributed/elastic/timer/api_test.py 40 test/distributed/elastic/timer/local_timer_example.py 68 test/distributed/elastic/timer/local_timer_test.py 196 test/distributed/elastic/utils/__init__.py 1 test/distributed/elastic/utils/data/__init__.py 1 test/distributed/elastic/utils/data/cycling_iterator_test.py 26 test/distributed/elastic/utils/distributed_test.py 112 test/distributed/elastic/utils/logging_test.py 19 test/distributed/elastic/utils/util_test.py 49 test/distributed/fsdp/test_flatten_params_wrapper.py 138 test/distributed/fsdp/test_fsdp_checkpoint.py 180 test/distributed/fsdp/test_fsdp_core.py 262 test/distributed/fsdp/test_fsdp_freezing_weights.py 144 test/distributed/fsdp/test_fsdp_input.py 61 test/distributed/fsdp/test_fsdp_memory.py 178 test/distributed/fsdp/test_fsdp_multiple_forward.py 60 test/distributed/fsdp/test_fsdp_multiple_wrapping.py 44 test/distributed/fsdp/test_fsdp_overlap.py 174 test/distributed/fsdp/test_fsdp_pure_fp16.py 70 test/distributed/fsdp/test_fsdp_uneven.py 54 test/distributed/fsdp/test_utils.py 61 test/distributed/fsdp/test_wrap.py 281 test/distributed/launcher/__init__.py 1 test/distributed/launcher/api_test.py 306 test/distributed/launcher/launch_test.py 66 test/distributed/launcher/run_test.py 467 test/distributed/nn/jit/__init__.py 1 test/distributed/nn/jit/test_instantiator.py 65 test/distributed/optim/test_zero_redundancy_optimizer.py 826 test/distributed/pipeline/sync/__init__.py 1 test/distributed/pipeline/sync/conftest.py 38 test/distributed/pipeline/sync/skip/__init__.py 1 test/distributed/pipeline/sync/skip/test_api.py 26 test/distributed/pipeline/sync/skip/test_gpipe.py 79 test/distributed/pipeline/sync/skip/test_inspect_skip_layout.py 67 test/distributed/pipeline/sync/skip/test_leak.py 83 test/distributed/pipeline/sync/skip/test_portal.py 94 test/distributed/pipeline/sync/skip/test_stash_pop.py 92 test/distributed/pipeline/sync/skip/test_tracker.py 85 test/distributed/pipeline/sync/skip/test_verify_skippables.py 104 test/distributed/pipeline/sync/test_balance.py 154 test/distributed/pipeline/sync/test_bugs.py 82 test/distributed/pipeline/sync/test_checkpoint.py 97 test/distributed/pipeline/sync/test_copy.py 43 test/distributed/pipeline/sync/test_deferred_batch_norm.py 117 test/distributed/pipeline/sync/test_dependency.py 95 test/distributed/pipeline/sync/test_inplace.py 37 test/distributed/pipeline/sync/test_microbatch.py 87 test/distributed/pipeline/sync/test_phony.py 30 test/distributed/pipeline/sync/test_pipe.py 573 test/distributed/pipeline/sync/test_pipeline.py 19 test/distributed/pipeline/sync/test_stream.py 134 test/distributed/pipeline/sync/test_transparency.py 22 test/distributed/pipeline/sync/test_worker.py 66 test/distributed/rpc/cuda/test_tensorpipe_agent.py 24 test/distributed/rpc/test_faulty_agent.py 25 test/distributed/rpc/test_tensorpipe_agent.py 26 test/distributed/test_c10d_common.py 689 test/distributed/test_c10d_gloo.py 1803 test/distributed/test_c10d_nccl.py 2089 test/distributed/test_c10d_spawn.py 197 test/distributed/test_c10d_spawn_gloo.py 220 test/distributed/test_c10d_spawn_nccl.py 132 test/distributed/test_data_parallel.py 702 test/distributed/test_distributed_spawn.py 27 test/distributed/test_launcher.py 43 test/distributed/test_nccl.py 141 test/distributed/test_pg_wrapper.py 313 test/distributed/test_store.py 321 test/distributions/test_constraints.py 109 test/distributions/test_distributions.py 4449 test/distributions/test_transforms.py 384 test/distributions/test_utils.py 19 test/error_messages/storage.py 54 test/expect/__init__.py 1 test/forward_backward_compatibility/check_forward_backward_compatibility.py 237 test/forward_backward_compatibility/dump_all_function_schemas.py 19 test/fx/named_tup.py 5 test/fx/quantization.py 241 test/fx/test_dce_pass.py 110 test/fx/test_future.py 33 test/fx/test_fx_const_fold.py 378 test/fx/test_fx_param_shape_control_flow.py 99 test/fx/test_gradual_type.py 795 test/fx/test_subgraph_rewriter.py 328 test/fx2trt/converters/acc_op/test_adaptive_avgpool.py 44 test/fx2trt/converters/acc_op/test_avgpool.py 44 test/fx2trt/converters/acc_op/test_batchnorm.py 33 test/fx2trt/converters/acc_op/test_binary_ops.py 106 test/fx2trt/converters/acc_op/test_cat.py 32 test/fx2trt/converters/acc_op/test_chunk.py 26 test/fx2trt/converters/acc_op/test_clamp.py 27 test/fx2trt/converters/acc_op/test_convolution.py 55 test/fx2trt/converters/acc_op/test_dequantize.py 38 test/fx2trt/converters/acc_op/test_elu.py 28 test/fx2trt/converters/acc_op/test_flatten.py 61 test/fx2trt/converters/acc_op/test_gelu.py 33 test/fx2trt/converters/acc_op/test_getitem.py 51 test/fx2trt/converters/acc_op/test_hard_sigmoid.py 36 test/fx2trt/converters/acc_op/test_hardtanh.py 14 test/fx2trt/converters/acc_op/test_layernorm.py 55 test/fx2trt/converters/acc_op/test_leaky_relu.py 28 test/fx2trt/converters/acc_op/test_linear.py 46 test/fx2trt/converters/acc_op/test_matmul.py 79 test/fx2trt/converters/acc_op/test_max.py 69 test/fx2trt/converters/acc_op/test_maximum.py 26 test/fx2trt/converters/acc_op/test_maxpool.py 54 test/fx2trt/converters/acc_op/test_min.py 69 test/fx2trt/converters/acc_op/test_minimum.py 26 test/fx2trt/converters/acc_op/test_narrow.py 26 test/fx2trt/converters/acc_op/test_pad.py 61 test/fx2trt/converters/acc_op/test_permute.py 47 test/fx2trt/converters/acc_op/test_quantize_per_tensor.py 36 test/fx2trt/converters/acc_op/test_reduce_ops.py 60 test/fx2trt/converters/acc_op/test_relu.py 28 test/fx2trt/converters/acc_op/test_reshape.py 46 test/fx2trt/converters/acc_op/test_selu.py 28 test/fx2trt/converters/acc_op/test_sigmoid.py 14 test/fx2trt/converters/acc_op/test_silu.py 28 test/fx2trt/converters/acc_op/test_size.py 30 test/fx2trt/converters/acc_op/test_softmax.py 58 test/fx2trt/converters/acc_op/test_softsign.py 28 test/fx2trt/converters/acc_op/test_split.py 30 test/fx2trt/converters/acc_op/test_squeeze.py 28 test/fx2trt/converters/acc_op/test_tanh.py 28 test/fx2trt/converters/acc_op/test_tile.py 64 test/fx2trt/converters/acc_op/test_topk.py 40 test/fx2trt/converters/acc_op/test_unary_ops.py 40 test/fx2trt/converters/acc_op/test_unsqueeze.py 48 test/fx2trt/converters/vanilla/test_add.py 18 test/fx2trt/converters/vanilla/test_convolution.py 38 test/fx2trt/core/test_import_fx2trt.py 7 test/fx2trt/core/test_input_tensor_spec.py 41 test/fx2trt/core/test_trt_module.py 45 test/fx2trt/passes/test_fuse_permute_linear_trt.py 59 test/fx2trt/passes/test_fuse_permute_matmul_trt.py 104 test/fx2trt/passes/test_fuse_unsqueeze_cat_sum_trt.py 33 test/fx2trt/passes/test_multi_fuse_trt.py 47 test/fx2trt/passes/test_remove_duplicate_output_args.py 45 test/fx2trt/test_quant_trt.py 583 test/fx2trt/trt_lower/test_fx2trt_lower.py 38 test/fx2trt/trt_lower/trt_operator_supported_test.py 63 test/fx2trt/trt_lower/trt_splitter_test.py 748 test/fx_acc/test_acc_tracer.py 1583 test/jit/__init__.py 1 test/jit/_imported_class_test/__init__.py 1 test/jit/_imported_class_test/bar.py 5 test/jit/_imported_class_test/foo.py 7 test/jit/_imported_class_test/very/__init__.py 1 test/jit/_imported_class_test/very/very/__init__.py 1 test/jit/_imported_class_test/very/very/nested.py 5 test/jit/fixtures/test_versioned_div_scalar_float_v3.pt 13 test/jit/fixtures/test_versioned_div_scalar_inplace_float_v3.pt 15 test/jit/fixtures/test_versioned_div_scalar_inplace_int_v3.pt 15 test/jit/fixtures/test_versioned_div_scalar_int_v3.pt 13 test/jit/fixtures/test_versioned_div_scalar_reciprocal_float_v3.pt 12 test/jit/fixtures/test_versioned_div_scalar_reciprocal_int_v3.pt 17 test/jit/fixtures/test_versioned_div_scalar_scalar_v3.pt 10 test/jit/fixtures/test_versioned_div_tensor_inplace_v3.pt 14 test/jit/fixtures/test_versioned_div_tensor_out_v3.pt 14 test/jit/fixtures/test_versioned_div_tensor_v3.pt 8 test/jit/fixtures/test_versioned_full_integer_value_v4.pt 11 test/jit/fixtures/test_versioned_full_preserved_v4.pt 10 test/jit/test_alias_analysis.py 27 test/jit/test_async.py 397 test/jit/test_aten_pow.py 78 test/jit/test_attr.py 25 test/jit/test_autodiff_subgraph_slicing.py 312 test/jit/test_backend_nnapi.py 69 test/jit/test_backends.py 433 test/jit/test_batch_mm.py 261 test/jit/test_builtins.py 213 test/jit/test_class_type.py 1084 test/jit/test_complex.py 245 test/jit/test_complexity.py 70 test/jit/test_convert_activation.py 140 test/jit/test_cuda.py 413 test/jit/test_custom_operators.py 109 test/jit/test_data_parallel.py 116 test/jit/test_dce.py 32 test/jit/test_device_analysis.py 273 test/jit/test_dtype_analysis.py 280 test/jit/test_enum.py 266 test/jit/test_export_modes.py 80 test/jit/test_freezing.py 1876 test/jit/test_functional_blocks.py 29 test/jit/test_fuser_common.py 13 test/jit/test_graph_rewrite_passes.py 48 test/jit/test_hash.py 80 test/jit/test_hooks.py 280 test/jit/test_hooks_modules.py 352 test/jit/test_if_hoisting.py 152 test/jit/test_ignorable_args.py 26 test/jit/test_ignore_context_manager.py 90 test/jit/test_isinstance.py 235 test/jit/test_jit_utils.py 66 test/jit/test_legacy_upgraders.py 394 test/jit/test_list_dict.py 1779 test/jit/test_logging.py 97 test/jit/test_misc.py 170 test/jit/test_models.py 547 test/jit/test_module_apis.py 100 test/jit/test_module_containers.py 485 test/jit/test_module_interface.py 502 test/jit/test_modules.py 18 test/jit/test_onnx_export.py 277 test/jit/test_optimize_for_mobile_preserve_debug_info.py 233 test/jit/test_parametrization.py 42 test/jit/test_pdt.py 361 test/jit/test_peephole.py 678 test/jit/test_profiler.py 192 test/jit/test_python_bindings.py 64 test/jit/test_python_builtins.py 296 test/jit/test_python_ir.py 16 test/jit/test_recursive_script.py 559 test/jit/test_remove_mutation.py 244 test/jit/test_save_load.py 344 test/jit/test_save_load_for_op_version.py 324 test/jit/test_script_profile.py 88 test/jit/test_scriptmod_ann.py 199 test/jit/test_slice.py 124 test/jit/test_sparse.py 84 test/jit/test_string_formatting.py 128 test/jit/test_symbolic_shape_analysis.py 387 test/jit/test_tensor_creation_ops.py 41 test/jit/test_torchbind.py 348 test/jit/test_tracer.py 1891 test/jit/test_type_sharing.py 415 test/jit/test_types.py 217 test/jit/test_typing.py 425 test/jit/test_union.py 711 test/jit/test_unsupported_ops.py 49 test/jit/test_upgraders.py 173 test/jit/test_warn.py 132 test/jit/test_with.py 351 test/jit_hooks/model.py 50 test/jit_hooks/test_jit_hooks.cpp 120 test/linear.py 7 test/mobile/custom_build/build.sh 64 test/mobile/custom_build/predictor.cpp 34 test/mobile/custom_build/prepare_model.py 13 test/mobile/nnc/aot_test_model.py 10 test/mobile/nnc/test_aot_compile.sh 20 test/mobile/nnc/test_context.cpp 125 test/mobile/nnc/test_nnc_backend.cpp 187 test/mobile/nnc/test_registry.cpp 29 test/mobile/test_bytecode.py 213 test/mobile/test_lite_script_module.py 398 test/mobile/test_lite_script_type.py 88 test/mobile/test_quantize_fx_lite_script_module.py 72 test/mobile/test_upgrader_bytecode_table_example.cpp 278 test/mobile/test_upgrader_codegen.py 26 test/mobile/test_upgraders.py 45 test/onnx/debug_embed_params.py 35 test/onnx/export_onnx_tests_filter.py 71 test/onnx/export_onnx_tests_generator.py 115 test/onnx/model_defs/__init__.py 4 test/onnx/model_defs/dcgan.py 66 test/onnx/model_defs/emb_seq.py 23 test/onnx/model_defs/lstm_flattening_result.py 26 test/onnx/model_defs/mnist.py 18 test/onnx/model_defs/op_test.py 40 test/onnx/model_defs/rnn_model_with_packed_sequence.py 36 test/onnx/model_defs/squeezenet.py 80 test/onnx/model_defs/srresnet.py 60 test/onnx/model_defs/super_resolution.py 23 test/onnx/model_defs/word_language_model.py 76 test/onnx/pytorch_helper.py 40 test/onnx/test_caffe2_common.py 30 test/onnx/test_custom_ops.py 117 test/onnx/test_models.py 202 test/onnx/test_models_onnxruntime.py 30 test/onnx/test_onnx_common.py 6 test/onnx/test_onnx_opset.py 308 test/onnx/test_operators.py 824 test/onnx/test_pytorch_common.py 86 test/onnx/test_pytorch_helper.py 48 test/onnx/test_pytorch_onnx_caffe2.py 1983 test/onnx/test_pytorch_onnx_caffe2_quantized.py 257 test/onnx/test_pytorch_onnx_onnxruntime.py 8292 test/onnx/test_pytorch_onnx_onnxruntime_cuda.py 81 test/onnx/test_pytorch_onnx_shape_inference.py 86 test/onnx/test_utility_funs.py 961 test/onnx/test_verify.py 79 test/onnx/verify.py 258 test/package/__init__.py 1 test/package/common.py 29 test/package/generate_bc_packages.py 28 test/package/module_a.py 1 test/package/package_a/__init__.py 7 test/package/package_a/fake_interface.py 29 test/package/package_a/fake_script_class.py 25 test/package/package_a/std_sys_module_hacks.py 11 test/package/package_a/subpackage.py 5 test/package/package_a/test_all_leaf_modules_tracer.py 4 test/package/package_a/test_module.py 43 test/package/package_a/test_nn_module.py 22 test/package/package_a/use_dunder_package.py 6 test/package/package_a/use_torch_package_importer.py 4 test/package/package_b/__init__.py 10 test/package/package_b/subpackage_0/__init__.py 1 test/package/package_b/subpackage_0/subsubpackage_0/__init__.py 4 test/package/package_b/subpackage_1.py 7 test/package/package_b/subpackage_2.py 7 test/package/package_c/__init__.py 7 test/package/package_c/test_module.py 14 test/package/test_analyze.py 16 test/package/test_dependency_api.py 226 test/package/test_dependency_hooks.py 80 test/package/test_digraph.py 100 test/package/test_directory_reader.py 178 test/package/test_glob_group.py 95 test/package/test_importer.py 101 test/package/test_load_bc_packages.py 33 test/package/test_mangling.py 74 test/package/test_misc.py 144 test/package/test_model.py 87 test/package/test_package_fx.py 123 test/package/test_package_script.py 500 test/package/test_resources.py 85 test/package/test_save_load.py 184 test/package/test_trace_dep/__init__.py 5 test/quantization/__init__.py 1 test/quantization/ao_migration/__init__.py 1 test/quantization/ao_migration/common.py 37 test/quantization/ao_migration/test_ao_migration.py 109 test/quantization/ao_migration/test_quantization.py 227 test/quantization/ao_migration/test_quantization_fx.py 178 test/quantization/bc/__init__.py 1 test/quantization/bc/test_backward_compatibility.py 277 test/quantization/core/__init__.py 1 test/quantization/core/test_quantized_functional.py 201 test/quantization/core/test_quantized_module.py 1101 test/quantization/core/test_quantized_op.py 4127 test/quantization/core/test_quantized_tensor.py 926 test/quantization/core/test_workflow_module.py 994 test/quantization/core/test_workflow_ops.py 1038 test/quantization/dbr/__init__.py 1 test/quantization/dbr/test_quantize_dbr.py 1105 test/quantization/eager/__init__.py 1 test/quantization/eager/test_bias_correction_eager.py 84 test/quantization/eager/test_equalize_eager.py 150 test/quantization/eager/test_fuse_eager.py 335 test/quantization/eager/test_model_numerics.py 105 test/quantization/eager/test_numeric_suite_eager.py 437 test/quantization/eager/test_quantize_eager_ptq.py 979 test/quantization/eager/test_quantize_eager_qat.py 798 test/quantization/fx/__init__.py 1 test/quantization/fx/test_equalize_fx.py 629 test/quantization/fx/test_numeric_suite_fx.py 1659 test/quantization/fx/test_quantize_fx.py 4958 test/quantization/fx/test_subgraph_rewriter.py 328 test/quantization/jit/__init__.py 1 test/quantization/jit/test_deprecated_jit_quant.py 218 test/quantization/jit/test_fusion_passes.py 92 test/quantization/jit/test_quantize_jit.py 3139 test/quantization/serialized/TestSerialization.test_conv2d.expected.pt 14 test/quantization/serialized/TestSerialization.test_conv2d.input.pt 14 test/quantization/serialized/TestSerialization.test_conv2d.scripted.pt 38 test/quantization/serialized/TestSerialization.test_conv2d.state_dict.pt 24 test/quantization/serialized/TestSerialization.test_conv2d.traced.pt 36 test/quantization/serialized/TestSerialization.test_conv2d_graph.expected.pt 9 test/quantization/serialized/TestSerialization.test_conv2d_graph.input.pt 11 test/quantization/serialized/TestSerialization.test_conv2d_graph_v2.expected.pt 9 test/quantization/serialized/TestSerialization.test_conv2d_graph_v2.input.pt 11 test/quantization/serialized/TestSerialization.test_conv2d_graph_v3.expected.pt 9 test/quantization/serialized/TestSerialization.test_conv2d_graph_v3.input.pt 11 test/quantization/serialized/TestSerialization.test_conv2d_graph_v3.scripted.pt 32 test/quantization/serialized/TestSerialization.test_conv2d_nobias.expected.pt 12 test/quantization/serialized/TestSerialization.test_conv2d_nobias.input.pt 12 test/quantization/serialized/TestSerialization.test_conv2d_nobias.scripted.pt 37 test/quantization/serialized/TestSerialization.test_conv2d_nobias.state_dict.pt 23 test/quantization/serialized/TestSerialization.test_conv2d_nobias.traced.pt 32 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph.expected.pt 12 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph.input.pt 11 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v2.expected.pt 12 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v2.input.pt 11 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v2.scripted.pt 32 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v2.traced.pt 31 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v3.expected.pt 12 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v3.input.pt 11 test/quantization/serialized/TestSerialization.test_conv2d_nobias_graph_v3.traced.pt 29 test/quantization/serialized/TestSerialization.test_conv2d_relu.expected.pt 15 test/quantization/serialized/TestSerialization.test_conv2d_relu.input.pt 14 test/quantization/serialized/TestSerialization.test_conv2d_relu.scripted.pt 40 test/quantization/serialized/TestSerialization.test_conv2d_relu.state_dict.pt 23 test/quantization/serialized/TestSerialization.test_conv2d_relu.traced.pt 38 test/quantization/serialized/TestSerialization.test_conv3d.expected.pt 14 test/quantization/serialized/TestSerialization.test_conv3d.input.pt 14 test/quantization/serialized/TestSerialization.test_conv3d.state_dict.pt 23 test/quantization/serialized/TestSerialization.test_conv3d.traced.pt 38 test/quantization/serialized/TestSerialization.test_conv3d_relu.expected.pt 14 test/quantization/serialized/TestSerialization.test_conv3d_relu.input.pt 14 test/quantization/serialized/TestSerialization.test_conv3d_relu.scripted.pt 33 test/quantization/serialized/TestSerialization.test_conv3d_relu.state_dict.pt 23 test/quantization/serialized/TestSerialization.test_conv3d_relu.traced.pt 44 test/quantization/serialized/TestSerialization.test_default_qat_qconfig.expected.pt 9 test/quantization/serialized/TestSerialization.test_default_qat_qconfig.input.pt 11 test/quantization/serialized/TestSerialization.test_linear.expected.pt 14 test/quantization/serialized/TestSerialization.test_linear.input.pt 14 test/quantization/serialized/TestSerialization.test_linear.scripted.pt 32 test/quantization/serialized/TestSerialization.test_linear.state_dict.pt 25 test/quantization/serialized/TestSerialization.test_linear_dynamic_float16.expected.pt 11 test/quantization/serialized/TestSerialization.test_linear_dynamic_float16.input.pt 11 test/quantization/serialized/TestSerialization.test_linear_dynamic_float16.state_dict.pt 19 test/quantization/serialized/TestSerialization.test_linear_dynamic_qint8.expected.pt 11 test/quantization/serialized/TestSerialization.test_linear_dynamic_qint8.input.pt 11 test/quantization/serialized/TestSerialization.test_linear_dynamic_qint8.state_dict.pt 25 test/quantization/serialized/TestSerialization.test_linear_relu.expected.pt 14 test/quantization/serialized/TestSerialization.test_linear_relu.input.pt 14 test/quantization/serialized/TestSerialization.test_linear_relu.state_dict.pt 26 test/quantization/serialized/TestSerialization.test_linear_relu_package_quantization_transforms.expected.pt 10 test/quantization/serialized/TestSerialization.test_linear_relu_package_quantization_transforms.get_attr_targets.pt 8 test/quantization/serialized/TestSerialization.test_linear_relu_package_quantization_transforms.input.pt 9 test/quantization/serialized/TestSerialization.test_lstm.expected.pt 16 test/quantization/serialized/TestSerialization.test_lstm.input.pt 13 test/quantization/serialized/TestSerialization.test_per_channel_observer.expected.pt 12 test/quantization/serialized/TestSerialization.test_per_channel_observer.input.pt 12 test/quantization/serialized/TestSerialization.test_per_channel_observer.state_dict.pt 10 test/quantization/serialized/TestSerialization.test_per_tensor_observer.expected.pt 12 test/quantization/serialized/TestSerialization.test_per_tensor_observer.input.pt 12 test/quantization/serialized/TestSerialization.test_per_tensor_observer.state_dict.pt 10 test/run_test.py 862 test/scripts/cuda_memcheck_common.py 43 test/scripts/run_cuda_memcheck.py 112 test/simulate_nccl_errors.py 34 test/test_ao_sparsity.py 10 test/test_autocast.py 101 test/test_autograd.py 6957 test/test_binary_ufuncs.py 2861 test/test_bundled_images.py 58 test/test_bundled_inputs.py 321 test/test_complex.py 19 test/test_cpp_api_parity.py 40 test/test_cpp_extensions_aot.py 162 test/test_cpp_extensions_jit.py 603 test/test_cuda.py 3153 test/test_cuda_primary_ctx.py 64 test/test_dataloader.py 1836 test/test_datapipe.py 1344 test/test_deploy.py 23 test/test_determination.py 99 test/test_dispatch.py 714 test/test_foreach.py 513 test/test_function_schema.py 135 test/test_functional_autograd_benchmark.py 36 test/test_functional_optim.py 72 test/test_functionalization.py 188 test/test_futures.py 232 test/test_fx.py 2846 test/test_fx_experimental.py 1368 test/test_import_stats.py 54 test/test_indexing.py 1154 test/test_jit.py 12372 test/test_jit_autocast.py 545 test/test_jit_cuda_fuser.py 2664 test/test_jit_disabled.py 37 test/test_jit_fuser.py 762 test/test_jit_fuser_legacy.py 5 test/test_jit_fuser_te.py 1866 test/test_jit_legacy.py 8 test/test_jit_profiling.py 8 test/test_jit_simple.py 8 test/test_jit_string.py 288 test/test_kernel_launch_checks.py 17 test/test_license.py 36 test/test_linalg.py 6342 test/test_logging.py 10 test/test_masked.py 91 test/test_metal.py 141 test/test_mkldnn.py 917 test/test_mobile_optimizer.py 492 test/test_model_dump.py 185 test/test_module_init.py 387 test/test_modules.py 455 test/test_monitor.py 85 test/test_multiprocessing.py 685 test/test_multiprocessing_spawn.py 166 test/test_namedtensor.py 1559 test/test_namedtuple_return_api.py 138 test/test_native_functions.py 124 test/test_nn.py 15704 test/test_nnapi.py 562 test/test_numba_integration.py 229 test/test_numpy_interop.py 328 test/test_openmp.py 49 test/test_ops.py 805 test/test_optim.py 2223 test/test_overrides.py 697 test/test_package.py 9 test/test_profiler.py 699 test/test_pruning_op.py 59 test/test_public_bindings.py 248 test/test_python_dispatch.py 405 test/test_pytree.py 158 test/test_quantization.py 76 test/test_reductions.py 2497 test/test_scatter_gather_ops.py 124 test/test_segment_reductions.py 279 test/test_serialization.py 704 test/test_set_default_mobile_cpu_allocator.py 19 test/test_shape_ops.py 509 test/test_show_pickle.py 26 test/test_sort_and_select.py 849 test/test_sparse.py 2869 test/test_sparse_csr.py 1063 test/test_spectral_ops.py 1150 test/test_stateless.py 117 test/test_static_runtime.py 273 test/test_tensor_creation_ops.py 3114 test/test_tensorboard.py 680 test/test_tensorexpr.py 1269 test/test_tensorexpr_pybind.py 318 test/test_testing.py 1204 test/test_throughput_benchmark.py 60 test/test_torch.py 6372 test/test_type_hints.py 76 test/test_type_info.py 57 test/test_type_promotion.py 774 test/test_typing.py 147 test/test_unary_ufuncs.py 1035 test/test_utils.py 659 test/test_view_ops.py 1325 test/test_vmap.py 1793 test/test_vulkan.py 144 test/test_xnnpack_integration.py 986 test/typing/fail/bitwise_ops.py 4 test/typing/fail/creation_ops.py 4 test/typing/fail/random.py 2 test/typing/pass/creation_ops.py 74 test/typing/pass/math_ops.py 161 test/typing/reveal/module_list.py 8 test/typing/reveal/namedtuple.py 11 test/typing/reveal/opt_size.py 9 test/typing/reveal/size.py 8 test/typing/reveal/tensor_constructors.py 71 test/typing/reveal/tensor_copy.py 9 test/typing/reveal/tensor_sampling.py 27 test/typing/reveal/torch_optim.py 9 tools/stats/export_slow_tests.py 81 tools/stats/test_history.py 285 tools/test/test_actions_local_runner.py 133 tools/test/test_cmake.py 64 tools/test/test_codegen.py 106 tools/test/test_extract_scripts.py 73 tools/test/test_gen_backend_stubs.py 190 tools/test/test_max_tokens_pragma.py 35 tools/test/test_mypy_wrapper.py 144 tools/test/test_stats.py 576 tools/test/test_test_history.py 62 tools/test/test_test_selections.py 89 tools/test/test_trailing_newlines.py 33 tools/test/test_translate_annotations.py 204 tools/testing/test_selections.py 217 torch/csrc/deploy/interpreter/test_builtin_registry.cpp 46 torch/csrc/deploy/test_deploy.cpp 386 torch/csrc/deploy/test_deploy_from_python.py 5 torch/csrc/deploy/test_deploy_gpu.cpp 99 torch/csrc/deploy/test_deploy_lib.cpp 81 torch/csrc/deploy/test_deploy_missing_interpreter.cpp 11 torch/csrc/deploy/test_deploy_python.py 19 torch/csrc/deploy/test_deploy_python_ext.cpp 22 torch/csrc/deploy/unity/tests/simple_model.py 13 torch/csrc/deploy/unity/tests/sum.py 4 torch/csrc/deploy/unity/tests/test_unity.h 3 torch/csrc/deploy/unity/tests/test_unity_simple_model.cpp 30 torch/csrc/deploy/unity/tests/test_unity_sum.cpp 23 torch/fx/experimental/fx2trt/example/quantized_resnet_test.py 113 torch/testing/_internal/codegen/random_topo_test.py 263 torch/testing/_internal/distributed/ddp_under_dist_autograd_test.py 571 torch/testing/_internal/distributed/distributed_test.py 6862 torch/testing/_internal/distributed/nn/api/remote_module_test.py 589 torch/testing/_internal/distributed/pipe_with_ddp_test.py 116 torch/testing/_internal/distributed/rpc/dist_autograd_test.py 2077 torch/testing/_internal/distributed/rpc/dist_optimizer_test.py 184 torch/testing/_internal/distributed/rpc/jit/dist_autograd_test.py 90 torch/testing/_internal/distributed/rpc/jit/rpc_test.py 1021 torch/testing/_internal/distributed/rpc/rpc_test.py 5071 torch/testing/_internal/test_module/__init__.py 1 torch/testing/_internal/test_module/future_div.py 5 torch/testing/_internal/test_module/no_future_div.py 5