Structural vibration monitoring with diffractive optical processors.
Paper
Loaded from Europe PMC by your browser, not stored by OSCR: doi.org · Europe PMC
The paper is loaded when this pane is shown.
The authors' code
Shell · 36 lines · 1.2 KB · BSD-3-Clause
- set -ex
- LOCAL_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
- ROOT_DIR=$(cd "$LOCAL_DIR"/../.. && pwd)
- TEST_DIR="$ROOT_DIR/test"
- gtest_reports_dir="${TEST_DIR}/test-reports/cpp"
- pytest_reports_dir="${TEST_DIR}/test-reports/python"
- # Figure out which Python to use
- PYTHON="$(which python)"
- if [[ "${BUILD_ENVIRONMENT}" =~ py((2|3)\.?[0-9]?\.?[0-9]?) ]]; then
- PYTHON=$(which "python${BASH_REMATCH[1]}")
- fi
- if [[ "${BUILD_ENVIRONMENT}" == *rocm* ]]; then
- # HIP_PLATFORM is auto-detected by hipcc; unset to avoid build errors
- unset HIP_PLATFORM
- if which sccache > /dev/null; then
- # Save sccache logs to file
- sccache --stop-server || true
- rm -f ~/sccache_error.log || true
- SCCACHE_ERROR_LOG=~/sccache_error.log SCCACHE_IDLE_TIMEOUT=0 sccache --start-server
- # Report sccache stats for easier debugging
- sccache --zero-stats
- fi
- fi
- # /usr/local/caffe2 is where the cpp bits are installed to in cmake-only
- # builds. In +python builds the cpp tests are copied to /usr/local/caffe2 so
- # that the test code in .ci/test.sh is the same
- INSTALL_PREFIX="/usr/local/caffe2"
- mkdir -p "$gtest_reports_dir" || true
- mkdir -p "$pytest_reports_dir" || true
- mkdir -p "$INSTALL_PREFIX" || true
common.sh at commit 35d22a1, under BSD-3-Clause · at the source
Overview
- Electrical and Computer Engineering Department, University of California, Los Angeles, Los Angeles, CA 90095, USA
- Bioengineering Department, University of California, Los Angeles, Los Angeles, CA 90095, USA
- California NanoSystems Institute (CNSI), University of California, Los Angeles, Los Angeles, CA 90095, USA
- Civil and Environmental Engineering Department, University of California, Los Angeles, Los Angeles, CA 90095, USA
- California Geological Survey, California Department of Conservation, Sacramento, CA 95814, USA
Abstract
Structural health monitoring (SHM) is vital for maintaining the safety and longevity of civil infrastructure, yet current solutions remain constrained by cost, power consumption, scalability, and the complexity of data processing. Here, we present a diffractive vibration monitoring system, integrating a jointly optimized diffractive layer with a shallow neural network-based backend to remotely extract three-dimensional (3D) structural vibration spectra, offering a low-power, cost-effective, and scalable solution. Unlike prior diffractive processors designed primarily for static image classification or reconstruction tasks, this framework establishes a dynamic computational sensing modality where the optical front-end is co-optimized to encode time-varying mechanical vibrations into distinct spatiotemporal optical patterns. This architecture eliminates the need for dense sensor arrays or extensive data acquisition; instead, it uses a spatially optimized passive diffractive layer that encodes 3D structural displacements into modulated light, captured by a minimal number of detectors and decoded in real time by shallow and low-power neural networks to reconstruct the 3D displacement spectra of structures. The diffractive system’s efficacy was demonstrated both numerically and experimentally using millimeter-wave illumination on a laboratory-scale building model with a programmable shake table. Our system achieves more than an order-of-magnitude improvement in accuracy over conventional optics or separately trained modules, establishing a foundation for high-throughput 3D monitoring of structures. Beyond SHM, the 3D vibration monitoring capabilities of this cost-effective and data-efficient framework establish a distinct computational sensing modality with potential applications in disaster resilience, aerospace diagnostics, and autonomous navigation—where energy efficiency, low latency, and high-throughput are critical.
Reproduced under the paper's license (CC BY), from the paper cited above.
Repository
Its files are read in the Code ↔ Paper reader above.
pytorch/pytorch
35d22a1a2c36bc86b5a9bfbf7e625b369a77ff51, 30 September 2026Availability: 1 check, the latest on 30 September 2026: the link answers
- 30 September 2026: the link answers
2,000 files
- .ci/
caffe2/ , Shell, 36 linescommon.sh - .ci/
caffe2/ , Shell, 168 linestest.sh - .ci/
docker/ , Shell, 83 linesalmalinux/ build.sh - .ci/
docker/ , Shell, 612 linesbuild.sh - .ci/
docker/ , Shell, 18 linescommon/ cache_vision_models.sh - .ci/
docker/ , Shell, 51 linescommon/ common_utils.sh - .ci/
docker/ , Shell, 27 linescommon/ install_acl.sh - .ci/
docker/ , Shell, 20 linescommon/ install_amdsmi.sh - .ci/
docker/ , Shell, 101 linescommon/ install_base.sh - .ci/
docker/ , Shell, 128 linescommon/ install_cache.sh - .ci/
docker/ , Shell, 45 linescommon/ install_clang.sh - .ci/
docker/ , Shell, 102 linescommon/ install_conda.sh - .ci/
docker/ , Shell, 20 linescommon/ install_conda_docker.sh - .ci/
docker/ , Shell, 120 linescommon/ install_cpython.sh - .ci/
docker/ , Shell, 243 linescommon/ install_cuda.sh - .ci/
docker/ , Shell, 25 linescommon/ install_cudss.sh - .ci/
docker/ , Shell, 36 linescommon/ install_cusparselt.sh - .ci/
docker/ , Shell, 27 linescommon/ install_docs_reqs.sh - .ci/
docker/ , Shell, 75 linescommon/ install_executorch.sh - .ci/
docker/ , Shell, 20 linescommon/ install_gcc.sh - .ci/
docker/ , Shell, 54 linescommon/ install_halide.sh - .ci/
docker/ , Shell, 78 linescommon/ install_inductor_benchma rk_deps.sh - .ci/
docker/ , Shell, 62 linescommon/ install_jax.sh - .ci/
docker/ , Shell, 6 linescommon/ install_jni.sh - .ci/
docker/ , Shell, 8 linescommon/ install_lcov.sh - .ci/
docker/ , Shell, 56 linescommon/ install_libgomp.sh - .ci/
docker/ , Shell, 23 linescommon/ install_libpng.sh - .ci/
docker/ , Shell, 27 linescommon/ install_linter.sh - .ci/
docker/ , Shell, 23 linescommon/ install_llvm_tools.sh - .ci/
docker/ , Shell, 27 linescommon/ install_magma.sh - .ci/
docker/ , Shell, 23 linescommon/ install_magma_conda.sh - .ci/
docker/ , Shell, 10 linescommon/ install_mingw.sh - .ci/
docker/ , Shell, 16 linescommon/ install_mkl.sh - .ci/
docker/ , Shell, 13 linescommon/ install_mnist.sh - .ci/
docker/ , Shell, 37 linescommon/ install_nccl.sh - .ci/
docker/ , Shell, 20 linescommon/ install_nvpl.sh - .ci/
docker/ , Shell, 33 linescommon/ install_onnx.sh - .ci/
docker/ , Shell, 38 linescommon/ install_openblas.sh - .ci/
docker/ , Shell, 10 linescommon/ install_openmpi.sh - .ci/
docker/ , Shell, 16 linescommon/ install_patchelf.sh - .ci/
docker/ , Shell, 35 linescommon/ install_python.sh - .ci/
docker/ , Shell, 106 linescommon/ install_rocm.sh - .ci/
docker/ , Shell, 37 linescommon/ install_rust.sh - .ci/
docker/ , Shell, 212 linescommon/ install_torch_tpu.sh - .ci/
docker/ , Shell, 109 linescommon/ install_triton.sh - .ci/
docker/ , Shell, 40 linescommon/ install_user.sh - .ci/
docker/ , Shell, 244 linescommon/ install_xpu.sh - .ci/
docker/ , Shell, 9 linescommon/ patch_libstdc.sh - .ci/
docker/ , C/C++, 1,143 linesjava/ jni.h - .ci/
docker/ , Shell, 125 linesmanywheel/ build.sh - .ci/
docker/ , Shell, 105 linesmanywheel/ s390_scripts/ build.sh - .ci/
docker/ , Shell, 91 linesmanywheel/ s390_scripts/ build_utils.sh - .ci/
docker/ , Shell, 66 linesmanywheel/ s390_scripts/ dependencies_s390x.sh - .ci/
flash-attention/ , Shell, 112 linesbuild.sh - .ci/
libtorch/ , Python, 528 linesextract_libtorch_from_wh eel.py - .ci/
libtorch/ , Python, 42 linessmoke_test_extract_libto rch.py - .ci/
lumen_cli/ , Python, 1 linecli/ build_cli/ __init__.py - .ci/
lumen_cli/ , Python, 39 linescli/ build_cli/ register_build.py - .ci/
lumen_cli/ , Python, 1 linecli/ lib/ __init__.py - .ci/
lumen_cli/ , Python, 75 linescli/ lib/ common/ cli_helper.py - .ci/
lumen_cli/ , Python, 43 linescli/ lib/ common/ docker_helper.py - .ci/
lumen_cli/ , Python, 111 linescli/ lib/ common/ envs_helper.py - .ci/
lumen_cli/ , Python, 143 linescli/ lib/ common/ gh_summary.py - .ci/
lumen_cli/ , Python, 69 linescli/ lib/ common/ git_helper.py - .ci/
lumen_cli/ , Python, 14 linescli/ lib/ common/ logger.py - .ci/
lumen_cli/ , Python, 63 linescli/ lib/ common/ path_helper.py - .ci/
lumen_cli/ , Python, 72 linescli/ lib/ common/ pip_helper.py - .ci/
lumen_cli/ , Python, 140 linescli/ lib/ common/ utils.py - .ci/
lumen_cli/ , Python, 78 linescli/ lib/ core/ torchtitan/ lib.py - .ci/
lumen_cli/ , Python, 61 linescli/ lib/ core/ torchtitan/ torchtitan_test.py - .ci/
lumen_cli/ , Python, 290 linescli/ lib/ core/ vllm/ lib.py - .ci/
lumen_cli/ , Python, 297 linescli/ lib/ core/ vllm/ vllm_build.py - .ci/
lumen_cli/ , Python, 304 linescli/ lib/ core/ vllm/ vllm_test.py - .ci/
lumen_cli/ , Python, 40 linescli/ run.py - .ci/
lumen_cli/ , Python, 1 linecli/ test_cli/ __init__.py - .ci/
lumen_cli/ , Python, 68 linescli/ test_cli/ register_test.py - .ci/
lumen_cli/ , Python, 47 linestests/ test_app.py - .ci/
lumen_cli/ , Python, 117 linestests/ test_cli_helper.py - .ci/
lumen_cli/ , Python, 75 linestests/ test_docker_helper.py - .ci/
lumen_cli/ , Python, 149 linestests/ test_envs_helper.py - .ci/
lumen_cli/ , Python, 122 linestests/ test_path_helper.py - .ci/
lumen_cli/ , Python, 522 linestests/ test_run_plan.py - .ci/
lumen_cli/ , Python, 143 linestests/ test_utils.py - .ci/
lumen_cli/ , Python, 205 linestests/ test_vllm.py - .ci/
magma/ , Shell, 52 linesbuild_magma.sh - .ci/
magma/ , Shell, 12 linespackage_files/ build.sh - .ci/
pytorch/ , Shell, 111 linesbinary_linux_test.sh - .ci/
pytorch/ , Shell, 212 linesbinary_populate_env.sh - .ci/
pytorch/ , Shell, 132 linesbinary_upload.sh - .ci/
pytorch/ , Shell, 72 linesbinary_windows_build.sh - .ci/
pytorch/ , Shell, 23 linesbinary_windows_test.sh - .ci/
pytorch/ , Shell, 511 linesbuild.sh - .ci/
pytorch/ , Shell, 285 linescheck_binary.sh - .ci/
pytorch/ , Python, 172 linescheck_wheel_soabi.py - .ci/
pytorch/ , Shell, 58 linescodegen-test.sh - .ci/
pytorch/ , Shell, 66 linescommon-build.sh - .ci/
pytorch/ , Shell, 39 linescommon.sh - .ci/
pytorch/ , Shell, 444 linescommon_utils.sh - .ci/
pytorch/ , Shell, 122 linescpp_doc_push_script.sh - .ci/
pytorch/ , Shell, 6 linesdocker-build-test.sh - .ci/
pytorch/ , Shell, 9 linesdocs-test.sh - .ci/
pytorch/ , Python, 1 linefake_numpy/ numpy.py - .ci/
pytorch/ , Shell, 56 linesinstall_cache_xla.sh - .ci/
pytorch/ , Shell, 11 linesmacos-build-test.sh - .ci/
pytorch/ , Shell, 58 linesmacos-build.sh - .ci/
pytorch/ , Shell, 14 linesmacos-common.sh - .ci/
pytorch/ , Shell, 417 linesmacos-test.sh - .ci/
pytorch/ , Shell, 75 linesmultigpu-test.sh - .ci/
pytorch/ , Python, 18 linesprint_sccache_log.py - .ci/
pytorch/ , Shell, 156 linespython_doc_push_script.s h - .ci/
pytorch/ , Python, 66 linesrocm_preflight.py - .ci/
pytorch/ , Shell, 118 linesrocm_utils.sh - .ci/
pytorch/ , Shell, 425 linesrun_tests.sh - .ci/
pytorch/ , Python, 506 linessmoke_test/ check_binary_symbols.py - .ci/
pytorch/ , Python, 77 linessmoke_test/ check_gomp.py - .ci/
pytorch/ , Python, 272 linessmoke_test/ check_wheel_tags.py - .ci/
pytorch/ , Python, 209 linessmoke_test/ max_autotune.py - .ci/
pytorch/ , Python, 711 linessmoke_test/ smoke_test.py - .ci/
pytorch/ , Shell, 2,802 linestest.sh - .ci/
pytorch/ , C++, 15 linestest_example_code/ check-torch-cuda.cpp - .ci/
pytorch/ , C++, 6 linestest_example_code/ check-torch-mkl.cpp - .ci/
pytorch/ , C++, 7 linestest_example_code/ check-torch-xnnpack.cpp - .ci/
pytorch/ , Python, 38 linestest_example_code/ cnn_smoke.py - .ci/
pytorch/ , Python, 41 linestest_example_code/ cnn_smoke_win_arm64.py - .ci/
pytorch/ , Python, 14 linestest_example_code/ rnn_smoke.py - .ci/
pytorch/ , Python, 13 linestest_example_code/ rnn_smoke_win_arm64.py - .ci/
pytorch/ , C++, 6 linestest_example_code/ simple-torch-test.cpp - .ci/
pytorch/ , Shell, 54 linestest_fa3_abi_stable.sh - .ci/
pytorch/ , Shell, 58 linestorch_trace.sh - .ci/
pytorch/ , Shell, 24 lineswin-arm64-test.sh - .ci/
pytorch/ , Shell, 49 lineswin-build.sh - .ci/
pytorch/ , Python, 58 lineswin-test-helpers/ run_python_nn_smoketests .py - .ci/
pytorch/ , Shell, 78 lineswin-test.sh - .ci/
wheel/ , Python, 205 lines_common.py - .ci/
wheel/ , Shell, 83 lineslinux/ build.sh - .ci/
wheel/ , Shell, 63 lineslinux/ build_all.sh - .ci/
wheel/ , Shell, 497 lineslinux/ build_common.sh - .ci/
wheel/ , Shell, 62 lineslinux/ build_cpu.sh - .ci/
wheel/ , Python, 524 lineslinux/ build_env_setup.py - .ci/
wheel/ , Python, 98 lineslinux/ build_install_deps.py - .ci/
wheel/ , Shell, 332 lineslinux/ build_libtorch.sh - .ci/
wheel/ , Python, 94 lineslinux/ build_wheel.py - .ci/
wheel/ , Python, 558 lineslinux/ repair_wheel.py - .ci/
wheel/ , Shell, 30 lineslinux/ set_desired_python.sh - .ci/
wheel/ , Shell, 45 linesmac/ build.sh - .ci/
wheel/ , Shell, 95 linesmac/ build_all_macos_wheels.s h - .ci/
wheel/ , Python, 64 linesmac/ build_env_setup.py - .ci/
wheel/ , Python, 65 linesmac/ build_install_deps.py - .ci/
wheel/ , Python, 49 linesmac/ build_wheel.py - .ci/
wheel/ , Shell, 27 linesmac/ install_libomp.sh - .ci/
wheel/ , Python, 55 linesmac/ repair_wheel.py - .ci/
wheel/ , Python, 471 lineswindows/ build_env_setup.py - .ci/
wheel/ , Python, 111 lineswindows/ build_install_deps.py - .ci/
wheel/ , Python, 45 lineswindows/ build_wheel.py - .ci/
wheel/ , Shell, 132 lineswindows/ set_desired_python.sh - .claude/
hooks/ , Shell, 109 linesclaude_code/ time-budget.sh - .claude/
hooks/ , Shell, 117 linespr_review/ restrict-write.sh - .claude/
hooks/ , Shell, 43 linespr_review/ validate-findings.sh - .claude/
hooks/ , Shell, 27 linespr_review/ validate-on-stop.sh - .claude/
hooks/ , Shell, 47 linespr_review/ validate-post-write.sh - .claude/
skills/ , Shell, 92 linesci-metrics/ gcx-wrapper.sh - .claude/
skills/ , Python, 78 linestriaging-issues/ scripts/ add_bot_triaged.py - .claude/
skills/ , Python, 88 linestriaging-issues/ scripts/ validate_issue_target.py - .claude/
skills/ , Python, 248 linestriaging-issues/ scripts/ validate_labels.py - .devcontainer/
scripts/ , Shell, 13 linesinstall-dev-tools.sh - .devcontainer/
scripts/ , Shell, 38 linesupdate_alternatives_clan g.sh - .github/
actions/ , Python, 390 linesreuse-old-whl/ reuse_old_whl.py - .github/
ci_configs/ , Python, 66 linesvllm/ use_existing_torch.py - .github/
regenerate.sh , Shell, 6 lines - .github/
scripts/ , Shell, 92 linesamd/ package_triton_wheel.sh - .github/
scripts/ , Shell, 101 linesamd/ patch_triton_wheel.sh - .github/
scripts/ , Python, 183 linesbuild_triton_wheel.py - .github/
scripts/ , Shell, 40 linesbuild_with_remote_buildk it.sh - .github/
scripts/ , Python, 252 linescheck_doc_redirects.py - .github/
scripts/ , Python, 473 linescherry_pick.py - .github/
scripts/ , Python, 212 linesclose_nonexistent_disabl e_issues.py - .github/
scripts/ , Python, 74 linescollect_ciflow_labels.py - .github/
scripts/ , Python, 37 linescomment_on_pr.py - .github/
scripts/ , Python, 71 linesconvert_lintrunner_annot ations_to_github.py - .github/
scripts/ , Python, 325 linesdelete_old_branches.py - .github/
scripts/ , Python, 54 linesdocathon-label-sync.py - .github/
scripts/ , Python, 84 linesensure_actions_will_canc el.py - .github/
scripts/ , Python, 51 linesexport_pytorch_labels.py - .github/
scripts/ , Python, 110 linesfile_io_utils.py - .github/
scripts/ , Python, 720 linesfilter_test_configs.py - .github/
scripts/ , Python, 580 linesgenerate_binary_build_ma trix.py - .github/
scripts/ , Python, 304 linesgenerate_ci_workflows.py - .github/
scripts/ , Python, 59 linesgenerate_docker_release_ matrix.py - .github/
scripts/ , Python, 133 linesgenerate_pytorch_version .py - .github/
scripts/ , Python, 13 linesget_aws_session_tokens.p y - .github/
scripts/ , Python, 30 linesget_ci_variable.py - .github/
scripts/ , Python, 193 linesget_workflow_job_id.py - .github/
scripts/ , Python, 273 linesgithub_utils.py - .github/
scripts/ , Python, 465 linesgitutils.py - .github/
scripts/ , Python, 380 linesgreenlight_guard.py - .github/
scripts/ , Python, 51 linesgreenlight_identity.py - .github/
scripts/ , Python, 198 linesgreenlight_ledger.py - .github/
scripts/ , Python, 200 linesgreenlight_messages.py - .github/
scripts/ , Python, 250 linesgreenlight_preflight.py - .github/
scripts/ , Python, 132 lineslabel_utils.py - .github/
scripts/ , Python, 61 lineslint_native_functions.py - .github/
scripts/ , Shell, 62 lineslintrunner.sh - .github/
scripts/ , Python, 30 linesparse_ref.py - .github/
scripts/ , Shell, 60 linespr-sanity-check.sh - .github/
scripts/ , Shell, 94 linesprepare_vllm_wheels.sh - .github/
scripts/ , Python, 115 linespytest_cache.py - .github/
scripts/ , Python, 266 linespytest_caching_utils.py - .github/
scripts/ , Shell, 9 linesreport_git_status.sh - .github/
scripts/ , Python, 819 linesrunner_determinator.py - .github/
scripts/ , Shell, 84 liness390x-ci/ self-hosted-builder/ helpers/ app_token.sh - .github/
scripts/ , Shell, 8 liness390x-ci/ self-hosted-builder/ helpers/ gh_cat_token.sh - .github/
scripts/ , Shell, 10 liness390x-ci/ self-hosted-builder/ helpers/ gh_token_generator.sh - .github/
scripts/ , Shell, 24 linesstop_runner_service.sh - .github/
scripts/ , Shell, 21 linestd_llm_indexer.sh - .github/
scripts/ , Python, 177 linestest_check_ci.py - .github/
scripts/ , Python, 288 linestest_cherry_pick.py - .github/
scripts/ , Python, 56 linestest_delete_old_branches .py - .github/
scripts/ , Python, 895 linestest_filter_test_configs .py - .github/
scripts/ , Python, 102 linestest_gitutils.py - .github/
scripts/ , Python, 758 linestest_greenlight_guard.py - .github/
scripts/ , Python, 821 linestest_greenlight_ledger.p y - .github/
scripts/ , Python, 241 linestest_greenlight_prefligh t.py - .github/
scripts/ , Python, 95 linestest_label_utils.py - .github/
scripts/ , Python, 84 linestest_pytest_caching_util s.py - .github/
scripts/ , Python, 950 linestest_runner_determinator .py - .github/
scripts/ , Python, 1,077 linestest_time_budget.py - .github/
scripts/ , Python, 3,009 linestest_trymerge.py - .github/
scripts/ , Python, 233 linestest_tryrebase.py - .github/
scripts/ , Python, 3,401 linestrymerge.py - .github/
scripts/ , Python, 98 linestrymerge_explainer.py - .github/
scripts/ , Python, 263 linestryrebase.py - .github/
scripts/ , Python, 160 linesupdate_metamates_rule.py - .github/
scripts/ , Shell, 41 linesupload_aws_ossci.sh - .spin/
cmds.py , Python, 658 lines - android/
common.sh , Shell, 74 lines - android/
pytorch_android/ , C++, 20 linesgenerate_test_asset.cpp - android/
pytorch_android/ , Python, 152 linesgenerate_test_torchscrip ts.py - android/
pytorch_android/ , C++, 21 linessrc/ androidTest/ cpp/ pytorch_jni_common_test. cpp - android/
pytorch_android/ , Java, 25 linessrc/ androidTest/ java/ org/ pytorch/ PytorchHostTests.java - android/
pytorch_android/ , Java, 42 linessrc/ androidTest/ java/ org/ pytorch/ PytorchInstrumentedTests .java - android/
pytorch_android/ , Java, 42 linessrc/ androidTest/ java/ org/ pytorch/ PytorchLiteInstrumentedT ests.java - android/
pytorch_android/ , Java, 694 linessrc/ androidTest/ java/ org/ pytorch/ PytorchTestBase.java - android/
pytorch_android/ , Java, 9 linessrc/ androidTest/ java/ org/ pytorch/ suite/ PytorchInstrumentedTestS uite.java - android/
pytorch_android/ , Java, 9 linessrc/ androidTest/ java/ org/ pytorch/ suite/ PytorchLiteInstrumentedT estSuite.java - android/
pytorch_android/ , C/C++, 3 linessrc/ main/ cpp/ cmake_macros.h - android/
pytorch_android/ , C++, 697 linessrc/ main/ cpp/ pytorch_jni_common.cpp - android/
pytorch_android/ , C/C++, 149 linessrc/ main/ cpp/ pytorch_jni_common.h - android/
pytorch_android/ , C++, 245 linessrc/ main/ cpp/ pytorch_jni_jit.cpp - android/
pytorch_android/ , C++, 209 linessrc/ main/ cpp/ pytorch_jni_lite.cpp - android/
pytorch_android/ , Java, 27 linessrc/ main/ java/ org/ pytorch/ DType.java - android/
pytorch_android/ , Java, 15 linessrc/ main/ java/ org/ pytorch/ Device.java - android/
pytorch_android/ , Java, 9 linessrc/ main/ java/ org/ pytorch/ INativePeer.java - android/
pytorch_android/ , Java, 343 linessrc/ main/ java/ org/ pytorch/ IValue.java - android/
pytorch_android/ , Java, 49 linessrc/ main/ java/ org/ pytorch/ LiteModuleLoader.java - android/
pytorch_android/ , Java, 64 linessrc/ main/ java/ org/ pytorch/ LiteNativePeer.java - android/
pytorch_android/ , Java, 14 linessrc/ main/ java/ org/ pytorch/ MemoryFormat.java - android/
pytorch_android/ , Java, 75 linessrc/ main/ java/ org/ pytorch/ Module.java - android/
pytorch_android/ , Java, 46 linessrc/ main/ java/ org/ pytorch/ NativePeer.java - android/
pytorch_android/ , Java, 50 linessrc/ main/ java/ org/ pytorch/ PyTorchAndroid.java - android/
pytorch_android/ , Java, 16 linessrc/ main/ java/ org/ pytorch/ PyTorchCodegenLoader.jav a - android/
pytorch_android/ , Java, 735 linessrc/ main/ java/ org/ pytorch/ Tensor.java - android/
pytorch_android_torchvis , Java, 24 linesion/ src/ androidTest/ java/ org/ pytorch/ torchvision/ TorchVisionInstrumentedT ests.java - android/
pytorch_android_torchvis , Java, 9 linesion/ src/ androidTest/ java/ org/ pytorch/ torchvision/ suite/ TorchVisionInstrumentedT estSuite.java - android/
pytorch_android_torchvis , C++, 187 linesion/ src/ main/ cpp/ pytorch_vision_jni.cpp - android/
pytorch_android_torchvis , Java, 398 linesion/ src/ main/ java/ org/ pytorch/ torchvision/ TensorImageUtils.java - android/
run_tests.sh , Shell, 57 lines - aten/
src/ , C/C++, 36 linesATen/ ATen.h - aten/
src/ , C++, 31 linesATen/ AccumulateType.cpp - aten/
src/ , C/C++, 173 linesATen/ AccumulateType.h - aten/
src/ , C/C++, 2 linesATen/ ArrayRef.h - aten/
src/ , C/C++, 2 linesATen/ Backend.h - aten/
src/ , C/C++, 2 linesATen/ Backtrace.h - aten/
src/ , C/C++, 52 linesATen/ BlasBackend.h - aten/
src/ , C/C++, 351 linesATen/ CPUApplyUtils.h - aten/
src/ , C++, 311 linesATen/ CPUGeneratorImpl.cpp - aten/
src/ , C/C++, 49 linesATen/ CPUGeneratorImpl.h - aten/
src/ , C++, 48 linesATen/ CachedTensorUtils.cpp - aten/
src/ , C/C++, 24 linesATen/ CachedTensorUtils.h - aten/
src/ , C/C++, 94 linesATen/ CollapseDims.h - aten/
src/ , C++, 65 linesATen/ ConjugateFallback.cpp - aten/
src/ , C++, 1,067 linesATen/ Context.cpp - aten/
src/ , C/C++, 771 linesATen/ Context.h - aten/
src/ , C++, 567 linesATen/ DLConvertor.cpp - aten/
src/ , C/C++, 84 linesATen/ DLConvertor.h - aten/
src/ , C++, 17 linesATen/ DTensorState.cpp - aten/
src/ , C/C++, 34 linesATen/ DTensorState.h - aten/
src/ , C/C++, 2 linesATen/ Device.h - aten/
src/ , C++, 157 linesATen/ DeviceAccelerator.cpp - aten/
src/ , C/C++, 126 linesATen/ DeviceAccelerator.h - aten/
src/ , C/C++, 41 linesATen/ DeviceGuard.h - aten/
src/ , C/C++, 2 linesATen/ DimVector.h - aten/
src/ , C++, 15 linesATen/ Dispatch.cpp - aten/
src/ , C/C++, 798 linesATen/ Dispatch.h - aten/
src/ , C/C++, 177 linesATen/ Dispatch_v2.h - aten/
src/ , C++, 107 linesATen/ DynamicLibrary.cpp - aten/
src/ , C/C++, 36 linesATen/ DynamicLibrary.h - aten/
src/ , C++, 483 linesATen/ EmptyTensor.cpp - aten/
src/ , C/C++, 166 linesATen/ EmptyTensor.h - aten/
src/ , C/C++, 30 linesATen/ ExpandBase.h - aten/
src/ , C++, 230 linesATen/ ExpandUtils.cpp - aten/
src/ , C/C++, 548 linesATen/ ExpandUtils.h - aten/
src/ , C++, 69 linesATen/ FakeTensorDispatchTables .cpp - aten/
src/ , C/C++, 42 linesATen/ FakeTensorDispatchTables .h - aten/
src/ , C/C++, 1 lineATen/ Formatting.h - aten/
src/ , C++, 31 linesATen/ FuncTorchTLS.cpp - aten/
src/ , C/C++, 46 linesATen/ FuncTorchTLS.h - aten/
src/ , C++, 467 linesATen/ FunctionalInverses.cpp - aten/
src/ , C++, 140 linesATen/ FunctionalStorageImpl.cp p - aten/
src/ , C/C++, 300 linesATen/ FunctionalStorageImpl.h - aten/
src/ , C++, 962 linesATen/ FunctionalTensorWrapper. cpp - aten/
src/ , C/C++, 491 linesATen/ FunctionalTensorWrapper. h - aten/
src/ , C++, 425 linesATen/ FunctionalizeFallbackKer nel.cpp - aten/
src/ , C/C++, 58 linesATen/ FunctionalizeFallbackKer nel.h - aten/
src/ , C/C++, 2 linesATen/ Generator.h - aten/
src/ , C/C++, 128 linesATen/ InferSize.h - aten/
src/ , C/C++, 15 linesATen/ InitialTensorOptions.h - aten/
src/ , C/C++, 2 linesATen/ Layout.h - aten/
src/ , C++, 378 linesATen/ LegacyBatchedFallback.cp p - aten/
src/ , C/C++, 25 linesATen/ LegacyBatchedFallback.h - aten/
src/ , C++, 161 linesATen/ LegacyBatchedTensorImpl. cpp - aten/
src/ , C/C++, 161 linesATen/ LegacyBatchedTensorImpl. h - aten/
src/ , C++, 1,291 linesATen/ LegacyBatchingRegistrati ons.cpp - aten/
src/ , C++, 26 linesATen/ LegacyVmapMode.cpp - aten/
src/ , C/C++, 26 linesATen/ LegacyVmapMode.h - aten/
src/ , C++, 295 linesATen/ LegacyVmapTransforms.cpp - aten/
src/ , C/C++, 183 linesATen/ LegacyVmapTransforms.h - aten/
src/ , C/C++, 31 linesATen/ LinalgBackend.h - aten/
src/ , C++, 644 linesATen/ MapAllocator.cpp - aten/
src/ , C/C++, 147 linesATen/ MapAllocator.h - aten/
src/ , C/C++, 109 linesATen/ MatrixRef.h - aten/
src/ , C++, 117 linesATen/ MemoryOverlap.cpp - aten/
src/ , C/C++, 42 linesATen/ MemoryOverlap.h - aten/
src/ , C++, 364 linesATen/ NestedTensorImpl.cpp - aten/
src/ , C/C++, 287 linesATen/ NestedTensorImpl.h - aten/
src/ , C++, 36 linesATen/ NodeCreationHooks.cpp - aten/
src/ , C/C++, 33 linesATen/ NodeCreationHooks.h - aten/
src/ , C/C++, 104 linesATen/ NumericUtils.h - aten/
src/ , C/C++, 79 linesATen/ OpMathType.h - aten/
src/ , C/C++, 206 linesATen/ OpaqueTensorImpl.h - aten/
src/ , C/C++, 17 linesATen/ PTThreadPool.h - aten/
src/ , C/C++, 12 linesATen/ PadNd.h - aten/
src/ , C/C++, 93 linesATen/ Parallel-inl.h - aten/
src/ , C/C++, 164 linesATen/ Parallel.h - aten/
src/ , C++, 135 linesATen/ ParallelCommon.cpp - aten/
src/ , C/C++, 13 linesATen/ ParallelFuture.h - aten/
src/ , C++, 320 linesATen/ ParallelNative.cpp - aten/
src/ , C/C++, 15 linesATen/ ParallelNative.h - aten/
src/ , C++, 116 linesATen/ ParallelOpenMP.cpp - aten/
src/ , C/C++, 54 linesATen/ ParallelOpenMP.h - aten/
src/ , C++, 95 linesATen/ ParallelThreadPoolNative .cpp - aten/
src/ , C++, 71 linesATen/ PythonTorchFunctionTLS.c pp - aten/
src/ , C/C++, 43 linesATen/ PythonTorchFunctionTLS.h - aten/
src/ , C/C++, 31 linesATen/ ROCmFABackend.h - aten/
src/ , C/C++, 16 linesATen/ SDPBackend.h - aten/
src/ , C++, 83 linesATen/ SavedTensorHooks.cpp - aten/
src/ , C/C++, 68 linesATen/ SavedTensorHooks.h - aten/
src/ , C/C++, 3 linesATen/ Scalar.h - aten/
src/ , C++, 55 linesATen/ ScalarOps.cpp - aten/
src/ , C/C++, 53 linesATen/ ScalarOps.h - aten/
src/ , C/C++, 4 linesATen/ ScalarType.h - aten/
src/ , C++, 17 linesATen/ SequenceNumber.cpp - aten/
src/ , C/C++, 13 linesATen/ SequenceNumber.h - aten/
src/ , C/C++, 2 linesATen/ SmallVector.h - aten/
src/ , C++, 254 linesATen/ SparseCsrTensorImpl.cpp - aten/
src/ , C/C++, 207 linesATen/ SparseCsrTensorImpl.h - aten/
src/ , C/C++, 464 linesATen/ SparseCsrTensorUtils.h - aten/
src/ , C++, 113 linesATen/ SparseTensorImpl.cpp - aten/
src/ , C/C++, 423 linesATen/ SparseTensorImpl.h - aten/
src/ , C/C++, 2 linesATen/ Storage.h - aten/
src/ , C++, 56 linesATen/ StorageUtils.cpp - aten/
src/ , C/C++, 49 linesATen/ StorageUtils.h - aten/
src/ , C/C++, 3 linesATen/ Tensor.h - aten/
src/ , C/C++, 2 linesATen/ TensorAccessor.h - aten/
src/ , C++, 38 linesATen/ TensorGeometry.cpp - aten/
src/ , C/C++, 155 linesATen/ TensorGeometry.h - aten/
src/ , C++, 73 linesATen/ TensorIndexing.cpp - aten/
src/ , C/C++, 904 linesATen/ TensorIndexing.h - aten/
src/ , C++, 1,742 linesATen/ TensorIterator.cpp - aten/
src/ , C/C++, 1,028 linesATen/ TensorIterator.h - aten/
src/ , C/C++, 72 linesATen/ TensorIteratorInternal.h - aten/
src/ , C++, 5 linesATen/ TensorMeta.cpp - aten/
src/ , C/C++, 133 linesATen/ TensorMeta.h - aten/
src/ , C/C++, 51 linesATen/ TensorOperators.h - aten/
src/ , C/C++, 2 linesATen/ TensorOptions.h - aten/
src/ , C/C++, 88 linesATen/ TensorSubclassLikeUtils. h - aten/
src/ , C++, 434 linesATen/ TensorUtils.cpp - aten/
src/ , C/C++, 190 linesATen/ TensorUtils.h - aten/
src/ , C++, 37 linesATen/ ThreadLocalPythonObjects .cpp - aten/
src/ , C/C++, 22 linesATen/ ThreadLocalPythonObjects .h - aten/
src/ , C++, 82 linesATen/ ThreadLocalState.cpp - aten/
src/ , C/C++, 131 linesATen/ ThreadLocalState.h - aten/
src/ , C/C++, 132 linesATen/ TracerMode.h - aten/
src/ , C/C++, 25 linesATen/ TypeDefault.h - aten/
src/ , C++, 77 linesATen/ Utils.cpp - aten/
src/ , C/C++, 138 linesATen/ Utils.h - aten/
src/ , C++, 228 linesATen/ Version.cpp - aten/
src/ , C/C++, 18 linesATen/ Version.h - aten/
src/ , C++, 111 linesATen/ VmapModeRegistrations.cp p - aten/
src/ , C/C++, 156 linesATen/ WrapDimUtils.h - aten/
src/ , C/C++, 43 linesATen/ WrapDimUtilsMulti.h - aten/
src/ , C++, 136 linesATen/ ZeroTensorFallback.cpp - aten/
src/ , C++, 16 linesATen/ accelerator/ Graph.cpp - aten/
src/ , C/C++, 64 linesATen/ accelerator/ Graph.h - aten/
src/ , C++, 588 linesATen/ autocast_mode.cpp - aten/
src/ , C/C++, 956 linesATen/ autocast_mode.h - aten/
src/ , C++, 85 linesATen/ benchmarks/ quantize_per_channel.cpp - aten/
src/ , C++, 85 linesATen/ benchmarks/ stateful_conv1d.cpp - aten/
src/ , C++, 30 linesATen/ benchmarks/ tensor_add.cpp - aten/
src/ , C/C++, 24 linesATen/ ceil_div.h - aten/
src/ , C/C++, 244 linesATen/ code_template.h - aten/
src/ , C++, 1 lineATen/ core/ ATenGeneral.cpp - aten/
src/ , C/C++, 3 linesATen/ core/ ATenGeneral.h - aten/
src/ , C/C++, 13 linesATen/ core/ ATenOpList.h - aten/
src/ , C/C++, 44 linesATen/ core/ ATen_fwd.h - aten/
src/ , C/C++, 157 linesATen/ core/ ATen_pch.h - aten/
src/ , C/C++, 48 linesATen/ core/ Array.h - aten/
src/ , C++, 5 linesATen/ core/ BackendSelectFallbackKer nel.cpp - aten/
src/ , C/C++, 2 linesATen/ core/ Backtrace.h - aten/
src/ , C++, 33 linesATen/ core/ CachingHostAllocator.cpp - aten/
src/ , C/C++, 1,508 linesATen/ core/ CachingHostAllocator.h - aten/
src/ , C/C++, 24 linesATen/ core/ CheckMemoryFormat.h - aten/
src/ , C++, 22 linesATen/ core/ DeprecatedTypeProperties .cpp - aten/
src/ , C/C++, 139 linesATen/ core/ DeprecatedTypeProperties .h - aten/
src/ , C++, 34 linesATen/ core/ DeprecatedTypeProperties Registry.cpp - aten/
src/ , C/C++, 33 linesATen/ core/ DeprecatedTypeProperties Registry.h - aten/
src/ , C++, 28 linesATen/ core/ Dict.cpp - aten/
src/ , C/C++, 380 linesATen/ core/ Dict.h - aten/
src/ , C/C++, 208 linesATen/ core/ Dict_inl.h - aten/
src/ , C/C++, 13 linesATen/ core/ DimVector.h - aten/
src/ , C/C++, 309 linesATen/ core/ DistributionsHelper.h - aten/
src/ , C++, 406 linesATen/ core/ Formatting.cpp - aten/
src/ , C/C++, 29 linesATen/ core/ Formatting.h - aten/
src/ , C++, 32 linesATen/ core/ Generator.cpp - aten/
src/ , C/C++, 187 linesATen/ core/ Generator.h - aten/
src/ , C++, 43 linesATen/ core/ GeneratorForPrivateuseon e.cpp - aten/
src/ , C/C++, 39 linesATen/ core/ GeneratorForPrivateuseon e.h - aten/
src/ , C++, 7 linesATen/ core/ GraphImplInterface.cpp - aten/
src/ , C/C++, 100 linesATen/ core/ GraphImplInterface.h - aten/
src/ , C/C++, 633 linesATen/ core/ IListRef.h - aten/
src/ , C/C++, 203 linesATen/ core/ IListRef_inl.h - aten/
src/ , C++, 264 linesATen/ core/ IListRef_test.cpp - aten/
src/ , C/C++, 111 linesATen/ core/ LegacyTypeDispatch.h - aten/
src/ , C++, 19 linesATen/ core/ List.cpp - aten/
src/ , C/C++, 546 linesATen/ core/ List.h - aten/
src/ , C/C++, 366 linesATen/ core/ List_inl.h - aten/
src/ , C++, 1,271 linesATen/ core/ List_test.cpp - aten/
src/ , C/C++, 194 linesATen/ core/ MT19937RNGEngine.h - aten/
src/ , C++, 27 linesATen/ core/ MetaFallbackKernel.cpp - aten/
src/ , C++, 80 linesATen/ core/ NestedIntSymNodeImpl.cpp - aten/
src/ , C/C++, 187 linesATen/ core/ NestedIntSymNodeImpl.h - aten/
src/ , C/C++, 240 linesATen/ core/ PhiloxRNGEngine.h - aten/
src/ , C++, 206 linesATen/ core/ PythonFallbackKernel.cpp - aten/
src/ , C/C++, 35 linesATen/ core/ PythonFallbackKernel.h - aten/
src/ , C++, 18 linesATen/ core/ PythonOpRegistrationTram poline.cpp - aten/
src/ , C/C++, 22 linesATen/ core/ PythonOpRegistrationTram poline.h - aten/
src/ , C/C++, 84 linesATen/ core/ QuantizerBase.h - aten/
src/ , C++, 12 linesATen/ core/ Range.cpp - aten/
src/ , C/C++, 25 linesATen/ core/ Range.h - aten/
src/ , C/C++, 14 linesATen/ core/ Reduction.h - aten/
src/ , C/C++, 1 lineATen/ core/ Scalar.h - aten/
src/ , C/C++, 1 lineATen/ core/ ScalarType.h - aten/
src/ , C++, 181 linesATen/ core/ Tensor.cpp - aten/
src/ , C/C++, 95 linesATen/ core/ Tensor.h - aten/
src/ , C/C++, 61 linesATen/ core/ TensorAccessor.h - aten/
src/ , C/C++, 1,059 linesATen/ core/ TensorBase.h - aten/
src/ , C++, 30 linesATen/ core/ TorchDispatchUtils.cpp - aten/
src/ , C/C++, 17 linesATen/ core/ TorchDispatchUtils.h - aten/
src/ , C/C++, 175 linesATen/ core/ TransformationHelper.h - aten/
src/ , C/C++, 1 lineATen/ core/ UndefinedTensorImpl.h - aten/
src/ , C/C++, 21 linesATen/ core/ UnsafeFromTH.h - aten/
src/ , C++, 117 linesATen/ core/ VariableFallbackKernel.c pp - aten/
src/ , C++, 20 linesATen/ core/ VariableHooksInterface.c pp - aten/
src/ , C/C++, 86 linesATen/ core/ VariableHooksInterface.h - aten/
src/ , C/C++, 92 linesATen/ core/ Variadic.h - aten/
src/ , C++, 12 linesATen/ core/ adaption.cpp - aten/
src/ , C/C++, 162 linesATen/ core/ alias_info.h - aten/
src/ , C++, 1 lineATen/ core/ blob.cpp - aten/
src/ , C/C++, 204 linesATen/ core/ blob.h - aten/
src/ , C/C++, 208 linesATen/ core/ boxing/ BoxedKernel.h - aten/
src/ , C/C++, 110 linesATen/ core/ boxing/ BoxedKernel_impl.h - aten/
src/ , C++, 56 linesATen/ core/ boxing/ KernelFunction.cpp - aten/
src/ , C/C++, 346 linesATen/ core/ boxing/ KernelFunction.h - aten/
src/ , C/C++, 389 linesATen/ core/ boxing/ KernelFunction_impl.h - aten/
src/ , C++, 525 linesATen/ core/ boxing/ KernelFunction_test.cpp - aten/
src/ , C/C++, 27 linesATen/ core/ boxing/ OperatorKernel.h - aten/
src/ , C/C++, 38 linesATen/ core/ boxing/ impl/ WrapFunctionIntoFunctor. h - aten/
src/ , C/C++, 41 linesATen/ core/ boxing/ impl/ WrapFunctionIntoRuntimeF unctor.h - aten/
src/ , C/C++, 405 linesATen/ core/ boxing/ impl/ boxing.h - aten/
src/ , C++, 1,098 linesATen/ core/ boxing/ impl/ kernel_function_legacy_t est.cpp - aten/
src/ , C++, 853 linesATen/ core/ boxing/ impl/ kernel_function_test.cpp - aten/
src/ , C++, 1,028 linesATen/ core/ boxing/ impl/ kernel_lambda_legacy_tes t.cpp - aten/
src/ , C++, 756 linesATen/ core/ boxing/ impl/ kernel_lambda_test.cpp - aten/
src/ , C++, 205 linesATen/ core/ boxing/ impl/ kernel_stackbased_test.c pp - aten/
src/ , C/C++, 802 linesATen/ core/ boxing/ impl/ make_boxed_from_unboxed_ functor.h - aten/
src/ , C++, 995 linesATen/ core/ boxing/ impl/ make_boxed_from_unboxed_ functor_test.cpp - aten/
src/ , C/C++, 140 linesATen/ core/ boxing/ impl/ test_helpers.h - aten/
src/ , C/C++, 90 linesATen/ core/ builtin_function.h - aten/
src/ , C++, 694 linesATen/ core/ class_type.cpp - aten/
src/ , C/C++, 441 linesATen/ core/ class_type.h - aten/
src/ , C++, 188 linesATen/ core/ custom_class.cpp - aten/
src/ , C/C++, 28 linesATen/ core/ custom_class.h - aten/
src/ , C/C++, 68 linesATen/ core/ dispatch/ CppSignature.h - aten/
src/ , C++, 38 linesATen/ core/ dispatch/ CppSignature_test.cpp - aten/
src/ , C++, 65 linesATen/ core/ dispatch/ DispatchKeyExtractor.cpp - aten/
src/ , C/C++, 280 linesATen/ core/ dispatch/ DispatchKeyExtractor.h - aten/
src/ , C++, 584 linesATen/ core/ dispatch/ Dispatcher.cpp - aten/
src/ , C/C++, 958 linesATen/ core/ dispatch/ Dispatcher.h - aten/
src/ , C++, 29 linesATen/ core/ dispatch/ ObservedOperators.cpp - aten/
src/ , C/C++, 17 linesATen/ core/ dispatch/ ObservedOperators.h - aten/
src/ , C++, 731 linesATen/ core/ dispatch/ OperatorEntry.cpp - aten/
src/ , C/C++, 336 linesATen/ core/ dispatch/ OperatorEntry.h - aten/
src/ , C/C++, 30 linesATen/ core/ dispatch/ OperatorOptions.h - aten/
src/ , C/C++, 36 linesATen/ core/ dispatch/ RegistrationHandleRAII.h - aten/
src/ , C++, 126 linesATen/ core/ dispatch/ backend_fallback_test.cp p - aten/
src/ , C++, 399 linesATen/ core/ dynamic_type.cpp - aten/
src/ , C/C++, 247 linesATen/ core/ dynamic_type.h - aten/
src/ , C/C++, 4 linesATen/ core/ enum_tag.h - aten/
src/ , C/C++, 104 linesATen/ core/ enum_type.h - aten/
src/ , C/C++, 114 linesATen/ core/ function.h - aten/
src/ , C++, 613 linesATen/ core/ function_schema.cpp - aten/
src/ , C/C++, 688 linesATen/ core/ function_schema.h - aten/
src/ , C/C++, 78 linesATen/ core/ function_schema_inl.h - aten/
src/ , C/C++, 151 linesATen/ core/ functional.h - aten/
src/ , C/C++, 10 linesATen/ core/ grad_mode.h - aten/
src/ , C++, 132 linesATen/ core/ interned_strings.cpp - aten/
src/ , C/C++, 353 linesATen/ core/ interned_strings.h - aten/
src/ , C/C++, 32 linesATen/ core/ interned_strings_class.h - aten/
src/ , C++, 1,252 linesATen/ core/ ivalue.cpp - aten/
src/ , C/C++, 1,630 linesATen/ core/ ivalue.h - aten/
src/ , C/C++, 2,535 linesATen/ core/ ivalue_inl.h - aten/
src/ , C/C++, 36 linesATen/ core/ ivalue_to.h - aten/
src/ , C/C++, 2,421 linesATen/ core/ jit_type.h - aten/
src/ , C/C++, 597 linesATen/ core/ jit_type_base.h - aten/
src/ , C++, 318 linesATen/ core/ library.cpp - aten/
src/ , C/C++, 81 linesATen/ core/ op_registration/ adaption.h - aten/
src/ , C++, 92 linesATen/ core/ op_registration/ infer_schema.cpp - aten/
src/ , C/C++, 157 linesATen/ core/ op_registration/ infer_schema.h - aten/
src/ , C/C++, 181 linesATen/ core/ op_registration/ op_allowlist.h - aten/
src/ , C++, 19 linesATen/ core/ op_registration/ op_allowlist_test.cpp - aten/
src/ , C++, 140 linesATen/ core/ op_registration/ op_registration.cpp - aten/
src/ , C/C++, 598 linesATen/ core/ op_registration/ op_registration.h - aten/
src/ , C++, 2,281 linesATen/ core/ op_registration/ op_registration_test.cpp - aten/
src/ , C++, 19 linesATen/ core/ operator_name.cpp - aten/
src/ , C/C++, 99 linesATen/ core/ operator_name.h - aten/
src/ , C/C++, 161 linesATen/ core/ qualified_name.h - aten/
src/ , C++, 69 linesATen/ core/ register_symbols.cpp - aten/
src/ , C/C++, 41 linesATen/ core/ rref_interface.h - aten/
src/ , C/C++, 233 linesATen/ core/ stack.h - aten/
src/ , C/C++, 144 linesATen/ core/ symbol.h - aten/
src/ , C++, 474 linesATen/ core/ tensor_type.cpp - aten/
src/ , C++, 1,070 linesATen/ core/ type.cpp - aten/
src/ , C++, 69 linesATen/ core/ type_factory.cpp - aten/
src/ , C/C++, 108 linesATen/ core/ type_factory.h - aten/
src/ , C/C++, 54 linesATen/ core/ type_ptr.h - aten/
src/ , C/C++, 1 lineATen/ core/ typeid.h - aten/
src/ , C++, 464 linesATen/ core/ union_type.cpp - aten/
src/ , C/C++, 110 linesATen/ cpp_custom_type_hack.h - aten/
src/ , C++, 76 linesATen/ cpu/ FlushDenormal.cpp - aten/
src/ , C/C++, 14 linesATen/ cpu/ FlushDenormal.h - aten/
src/ , C/C++, 61 linesATen/ cpu/ StatelessPhilox4x32.h - aten/
src/ , C++, 223 linesATen/ cpu/ Utils.cpp - aten/
src/ , C/C++, 23 linesATen/ cpu/ Utils.h - aten/
src/ , C/C++, 4 linesATen/ cpu/ vec/ functional.h - aten/
src/ , C/C++, 475 linesATen/ cpu/ vec/ functional_base.h - aten/
src/ , C/C++, 647 linesATen/ cpu/ vec/ functional_bfloat16.h - aten/
src/ , C/C++, 1 lineATen/ cpu/ vec/ intrinsics.h - aten/
src/ , C/C++, 80 linesATen/ cpu/ vec/ sve/ sve_helper.h - aten/
src/ , C/C++, 683 linesATen/ cpu/ vec/ sve/ vec_bfloat16.h - aten/
src/ , C/C++, 234 linesATen/ cpu/ vec/ sve/ vec_common_sve.h - aten/
src/ , C/C++, 633 linesATen/ cpu/ vec/ sve/ vec_double.h - aten/
src/ , C/C++, 902 linesATen/ cpu/ vec/ sve/ vec_float.h - aten/
src/ , C/C++, 515 linesATen/ cpu/ vec/ sve/ vec_int.h - aten/
src/ , C/C++, 591 linesATen/ cpu/ vec/ sve/ vec_qint.h - aten/
src/ , C/C++, 64 linesATen/ cpu/ vec/ vec.h - aten/
src/ , C/C++, 25 linesATen/ cpu/ vec/ vec128/ vec128.h - aten/
src/ , C/C++, 717 linesATen/ cpu/ vec/ vec128/ vec128_bfloat16_neon.h - aten/
src/ , C/C++, 96 linesATen/ cpu/ vec/ vec128/ vec128_cast_neon.h - aten/
src/ , C/C++, 519 linesATen/ cpu/ vec/ vec128/ vec128_convert.h - aten/
src/ , C/C++, 602 linesATen/ cpu/ vec/ vec128/ vec128_double_neon.h - aten/
src/ , C/C++, 869 linesATen/ cpu/ vec/ vec128/ vec128_float_neon.h - aten/
src/ , C/C++, 622 linesATen/ cpu/ vec/ vec128/ vec128_half_neon.h - aten/
src/ , C/C++, 794 linesATen/ cpu/ vec/ vec128/ vec128_int_aarch64.h - aten/
src/ , C/C++, 311 linesATen/ cpu/ vec/ vec128/ vec128_reduced_precision _common_neon.h - aten/
src/ , C/C++, 378 linesATen/ cpu/ vec/ vec128/ vec128_uint_aarch64.h - aten/
src/ , C/C++, 2,622 linesATen/ cpu/ vec/ vec128/ zarch/ vec128_zarch.h - aten/
src/ , C/C++, 413 linesATen/ cpu/ vec/ vec256/ vec256.h - aten/
src/ , C/C++, 837 linesATen/ cpu/ vec/ vec256/ vec256_16bit_float.h - aten/
src/ , C/C++, 280 linesATen/ cpu/ vec/ vec256/ vec256_bfloat16.h - aten/
src/ , C/C++, 531 linesATen/ cpu/ vec/ vec256/ vec256_complex_double.h - aten/
src/ , C/C++, 615 linesATen/ cpu/ vec/ vec256/ vec256_complex_float.h - aten/
src/ , C/C++, 410 linesATen/ cpu/ vec/ vec256/ vec256_convert.h - aten/
src/ , C/C++, 547 linesATen/ cpu/ vec/ vec256/ vec256_double.h - aten/
src/ , C/C++, 915 linesATen/ cpu/ vec/ vec256/ vec256_float.h - aten/
src/ , C/C++, 280 linesATen/ cpu/ vec/ vec256/ vec256_half.h - aten/
src/ , C/C++, 2,317 linesATen/ cpu/ vec/ vec256/ vec256_int.h - aten/
src/ , C/C++, 298 linesATen/ cpu/ vec/ vec256/ vec256_mask.h - aten/
src/ , C/C++, 1,441 linesATen/ cpu/ vec/ vec256/ vec256_qint.h - aten/
src/ , C/C++, 75 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_bfloat16_vsx.h - aten/
src/ , C/C++, 250 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_common_vsx.h - aten/
src/ , C/C++, 679 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_complex_double_vs x.h - aten/
src/ , C/C++, 771 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_complex_float_vsx .h - aten/
src/ , C/C++, 515 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_double_vsx.h - aten/
src/ , C/C++, 551 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_float_vsx.h - aten/
src/ , C/C++, 417 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_int16_vsx.h - aten/
src/ , C/C++, 347 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_int32_vsx.h - aten/
src/ , C/C++, 301 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_int64_vsx.h - aten/
src/ , C/C++, 69 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_mask_vsx.h - aten/
src/ , C/C++, 301 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_qint32_vsx.h - aten/
src/ , C/C++, 512 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_qint8_vsx.h - aten/
src/ , C/C++, 533 linesATen/ cpu/ vec/ vec256/ vsx/ vec256_quint8_vsx.h - aten/
src/ , C/C++, 576 linesATen/ cpu/ vec/ vec256/ vsx/ vsx_helpers.h - aten/
src/ , C/C++, 378 linesATen/ cpu/ vec/ vec512/ vec512.h - aten/
src/ , C/C++, 1,945 linesATen/ cpu/ vec/ vec512/ vec512_bfloat16.h - aten/
src/ , C/C++, 654 linesATen/ cpu/ vec/ vec512/ vec512_complex_double.h - aten/
src/ , C/C++, 1,222 linesATen/ cpu/ vec/ vec512/ vec512_complex_float.h - aten/
src/ , C/C++, 400 linesATen/ cpu/ vec/ vec512/ vec512_convert.h - aten/
src/ , C/C++, 580 linesATen/ cpu/ vec/ vec512/ vec512_double.h - aten/
src/ , C/C++, 1,015 linesATen/ cpu/ vec/ vec512/ vec512_float.h - aten/
src/ , C/C++, 694 linesATen/ cpu/ vec/ vec512/ vec512_float8.h - aten/
src/ , C/C++, 2,121 linesATen/ cpu/ vec/ vec512/ vec512_int.h - aten/
src/ , C/C++, 390 linesATen/ cpu/ vec/ vec512/ vec512_mask.h - aten/
src/ , C/C++, 1,543 linesATen/ cpu/ vec/ vec512/ vec512_qint.h - aten/
src/ , C/C++, 1,552 linesATen/ cpu/ vec/ vec_base.h - aten/
src/ , C/C++, 157 linesATen/ cpu/ vec/ vec_convert.h - aten/
src/ , C/C++, 118 linesATen/ cpu/ vec/ vec_half.h - aten/
src/ , C/C++, 319 linesATen/ cpu/ vec/ vec_mask.h - aten/
src/ , C/C++, 407 linesATen/ cpu/ vec/ vec_n.h - aten/
src/ , C/C++, 253 linesATen/ cpu/ vec/ vec_quant.h - aten/
src/ , C/C++, 179 linesATen/ cpu/ vml.h - aten/
src/ , C/C++, 9 linesATen/ cuda/ ATenCUDAGeneral.h - aten/
src/ , C++, 2,935 linesATen/ cuda/ CUDABlas.cpp - aten/
src/ , C/C++, 450 linesATen/ cuda/ CUDABlas.h - aten/
src/ , C++, 88 linesATen/ cuda/ CUDAContext.cpp - aten/
src/ , C/C++, 9 linesATen/ cuda/ CUDAContext.h - aten/
src/ , C/C++, 160 linesATen/ cuda/ CUDAContextLight.h - aten/
src/ , C/C++, 105 linesATen/ cuda/ CUDADataType.h - aten/
src/ , C/C++, 23 linesATen/ cuda/ CUDADevice.h - aten/
src/ , C/C++, 7 linesATen/ cuda/ CUDAEvent.h - aten/
src/ , C++, 583 linesATen/ cuda/ CUDAGeneratorImpl.cpp - aten/
src/ , C/C++, 205 linesATen/ cuda/ CUDAGeneratorImpl.h - aten/
src/ , C++, 706 linesATen/ cuda/ CUDAGraph.cpp - aten/
src/ , CUDA, 30 linesATen/ cuda/ CUDAGraph.cu - aten/
src/ , C/C++, 221 linesATen/ cuda/ CUDAGraph.h - aten/
src/ , C/C++, 24 linesATen/ cuda/ CUDASparse.h - aten/
src/ , C++, 885 linesATen/ cuda/ CUDASparseBlas.cpp - aten/
src/ , C/C++, 318 linesATen/ cuda/ CUDASparseBlas.h - aten/
src/ , C++, 209 linesATen/ cuda/ CUDASparseDescriptors.cp p - aten/
src/ , C/C++, 250 linesATen/ cuda/ CUDASparseDescriptors.h - aten/
src/ , C/C++, 20 linesATen/ cuda/ CUDAUtils.h - aten/
src/ , C++, 282 linesATen/ cuda/ CachingHostAllocator.cpp - aten/
src/ , C/C++, 70 linesATen/ cuda/ CachingHostAllocator.h - aten/
src/ , C++, 46 linesATen/ cuda/ CuSparseHandlePool.cpp - aten/
src/ , C++, 658 linesATen/ cuda/ CublasHandlePool.cpp - aten/
src/ , C++, 94 linesATen/ cuda/ EmptyTensor.cpp - aten/
src/ , C/C++, 44 linesATen/ cuda/ EmptyTensor.h - aten/
src/ , C++, 106 linesATen/ cuda/ Exceptions.cpp - aten/
src/ , C/C++, 238 linesATen/ cuda/ Exceptions.h - aten/
src/ , C++, 70 linesATen/ cuda/ MemPool.cpp - aten/
src/ , C/C++, 46 linesATen/ cuda/ MemPool.h - aten/
src/ , C++, 20 linesATen/ cuda/ PeerToPeerAccess.cpp - aten/
src/ , C/C++, 37 linesATen/ cuda/ PeerToPeerAccess.h - aten/
src/ , C/C++, 5 linesATen/ cuda/ PhiloxCudaState.h - aten/
src/ , C/C++, 10 linesATen/ cuda/ PinnedMemoryAllocator.h - aten/
src/ , CUDA, 69 linesATen/ cuda/ Sleep.cu - aten/
src/ , C/C++, 13 linesATen/ cuda/ Sleep.h - aten/
src/ , C/C++, 23 linesATen/ cuda/ ThrustAllocator.h - aten/
src/ , CUDA, 58 linesATen/ cuda/ cub-RadixSortKeys.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-f16-8 .cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-int32 -1.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-int32 -2.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-int32 -4.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-int64 -1.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-int64 -2.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-int64 -4.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-scala rs.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-uint1 6-8.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-uint3 2-8.cu - aten/
src/ , CUDA, 7 linesATen/ cuda/ cub-RadixSortPairs-uint6 4-8.cu - aten/
src/ , CUDA, 48 linesATen/ cuda/ cub.cu - aten/
src/ , C/C++, 93 linesATen/ cuda/ cub.h - aten/
src/ , CUDA, 34 linesATen/ cuda/ detail/ BLASConstants.cu - aten/
src/ , C/C++, 10 linesATen/ cuda/ detail/ BLASConstants.h - aten/
src/ , C++, 593 linesATen/ cuda/ detail/ CUDAHooks.cpp - aten/
src/ , C/C++, 74 linesATen/ cuda/ detail/ CUDAHooks.h - aten/
src/ , C/C++, 382 linesATen/ cuda/ detail/ CublasLtUtils.h - aten/
src/ , C/C++, 200 linesATen/ cuda/ detail/ DeviceThreadHandles.h - aten/
src/ , CUDA, 64 linesATen/ cuda/ detail/ IndexUtils.cu - aten/
src/ , C/C++, 37 linesATen/ cuda/ detail/ KernelUtils.h - aten/
src/ , C++, 289 linesATen/ cuda/ detail/ LazyNVRTC.cpp - aten/
src/ , C/C++, 11 linesATen/ cuda/ detail/ LazyNVRTC.h - aten/
src/ , CUDA, 371 linesATen/ cuda/ jiterator.cu - aten/
src/ , C/C++, 40 linesATen/ cuda/ jiterator.h - aten/
src/ , C/C++, 252 linesATen/ cuda/ jiterator_impl.h - aten/
src/ , C++, 323 linesATen/ cuda/ llvm_basic.cpp - aten/
src/ , C++, 1,155 linesATen/ cuda/ llvm_complex.cpp - aten/
src/ , C/C++, 14 linesATen/ cuda/ llvm_jit_strings.h - aten/
src/ , C++, 12 linesATen/ cuda/ nvrtc_stub/ ATenNVRTC.cpp - aten/
src/ , C/C++, 131 linesATen/ cuda/ nvrtc_stub/ ATenNVRTC.h - aten/
src/ , C/C++, 884 linesATen/ cuda/ tunable/ GemmCommon.h - aten/
src/ , C/C++, 1,024 linesATen/ cuda/ tunable/ GemmCublaslt.h - aten/
src/ , C/C++, 715 linesATen/ cuda/ tunable/ GemmHipblaslt.h - aten/
src/ , C/C++, 289 linesATen/ cuda/ tunable/ GemmRocblas.h - aten/
src/ , C++, 70 linesATen/ cuda/ tunable/ StreamTimer.cpp - aten/
src/ , C/C++, 58 linesATen/ cuda/ tunable/ StreamTimer.h - aten/
src/ , C++, 1,086 linesATen/ cuda/ tunable/ Tunable.cpp - aten/
src/ , C/C++, 369 linesATen/ cuda/ tunable/ Tunable.h - aten/
src/ , C/C++, 379 linesATen/ cuda/ tunable/ TunableGemm.h - aten/
src/ , C/C++, 517 linesATen/ cuda/ tunable/ TunableOp.h - aten/
src/ , C++, 127 linesATen/ cudnn/ AutocastRNN.cpp - aten/
src/ , C++, 197 linesATen/ cudnn/ Descriptors.cpp - aten/
src/ , C/C++, 410 linesATen/ cudnn/ Descriptors.h - aten/
src/ , C++, 57 linesATen/ cudnn/ Handle.cpp - aten/
src/ , C/C++, 9 linesATen/ cudnn/ Handle.h - aten/
src/ , C/C++, 2 linesATen/ cudnn/ Handles.h - aten/
src/ , C++, 38 linesATen/ cudnn/ Types.cpp - aten/
src/ , C/C++, 14 linesATen/ cudnn/ Types.h - aten/
src/ , C/C++, 22 linesATen/ cudnn/ Utils.h - aten/
src/ , C/C++, 16 linesATen/ cudnn/ cudnn-wrapper.h - aten/
src/ , C/C++, 93 linesATen/ detail/ AcceleratorHooksInterfac e.h - aten/
src/ , C++, 7 linesATen/ detail/ CPUGuardImpl.cpp - aten/
src/ , C++, 46 linesATen/ detail/ CUDAHooksInterface.cpp - aten/
src/ , C/C++, 258 linesATen/ detail/ CUDAHooksInterface.h - aten/
src/ , C/C++, 103 linesATen/ detail/ FunctionTraits.h - aten/
src/ , C++, 24 linesATen/ detail/ HIPHooksInterface.cpp - aten/
src/ , C/C++, 65 linesATen/ detail/ HIPHooksInterface.h - aten/
src/ , C++, 22 linesATen/ detail/ HPUHooksInterface.cpp - aten/
src/ , C/C++, 57 linesATen/ detail/ HPUHooksInterface.h - aten/
src/ , C++, 22 linesATen/ detail/ IPUHooksInterface.cpp - aten/
src/ , C/C++, 44 linesATen/ detail/ IPUHooksInterface.h - aten/
src/ , C++, 23 linesATen/ detail/ MAIAHooksInterface.cpp - aten/
src/ , C/C++, 41 linesATen/ detail/ MAIAHooksInterface.h - aten/
src/ , C++, 25 linesATen/ detail/ MPSHooksInterface.cpp - aten/
src/ , C/C++, 129 linesATen/ detail/ MPSHooksInterface.h - aten/
src/ , C++, 30 linesATen/ detail/ MTIAHooksInterface.cpp - aten/
src/ , C/C++, 219 linesATen/ detail/ MTIAHooksInterface.h - aten/
src/ , C++, 8 linesATen/ detail/ MetaGuardImpl.cpp - aten/
src/ , C++, 29 linesATen/ detail/ PrivateUse1HooksInterfac e.cpp - aten/
src/ , C/C++, 86 linesATen/ detail/ PrivateUse1HooksInterfac e.h - aten/
src/ , C++, 23 linesATen/ detail/ XLAHooksInterface.cpp - aten/
src/ , C/C++, 78 linesATen/ detail/ XLAHooksInterface.h - aten/
src/ , C++, 21 linesATen/ detail/ XPUHooksInterface.cpp - aten/
src/ , C/C++, 93 linesATen/ detail/ XPUHooksInterface.h - aten/
src/ , C/C++, 15 linesATen/ div_rtn.h - aten/
src/ , C/C++, 641 linesATen/ dlpack.h - aten/
src/ , C++, 240 linesATen/ functorch/ ADInterpreters.cpp - aten/
src/ , C/C++, 38 linesATen/ functorch/ ADInterpreters.h - aten/
src/ , C++, 54 linesATen/ functorch/ BatchRulesActivation.cpp - aten/
src/ , C++, 675 linesATen/ functorch/ BatchRulesBinaryOps.cpp - aten/
src/ , C++, 461 linesATen/ functorch/ BatchRulesConvolution.cp p - aten/
src/ , C++, 394 linesATen/ functorch/ BatchRulesDecompositions .cpp - aten/
src/ , C++, 78 linesATen/ functorch/ BatchRulesDynamic.cpp - aten/
src/ , C++, 247 linesATen/ functorch/ BatchRulesFactory.cpp - aten/
src/ , C++, 205 linesATen/ functorch/ BatchRulesHelper.cpp - aten/
src/ , C/C++, 483 linesATen/ functorch/ BatchRulesHelper.h - aten/
src/ , C++, 23 linesATen/ functorch/ BatchRulesIndexing.cpp - aten/
src/ , C++, 887 linesATen/ functorch/ BatchRulesLinearAlgebra. cpp - aten/
src/ , C++, 183 linesATen/ functorch/ BatchRulesLoss.cpp - aten/
src/ , C++, 395 linesATen/ functorch/ BatchRulesModules.cpp - aten/
src/ , C++, 884 linesATen/ functorch/ BatchRulesNorm.cpp - aten/
src/ , C++, 74 linesATen/ functorch/ BatchRulesPooling.cpp - aten/
src/ , C++, 500 linesATen/ functorch/ BatchRulesRandomness.cpp - aten/
src/ , C++, 536 linesATen/ functorch/ BatchRulesReduceOps.cpp - aten/
src/ , C++, 1,416 linesATen/ functorch/ BatchRulesScatterOps.cpp - aten/
src/ , C++, 182 linesATen/ functorch/ BatchRulesUnaryOps.cpp - aten/
src/ , C++, 694 linesATen/ functorch/ BatchRulesViews.cpp - aten/
src/ , C++, 515 linesATen/ functorch/ BatchedFallback.cpp - aten/
src/ , C/C++, 81 linesATen/ functorch/ BatchedFallback.h - aten/
src/ , C++, 186 linesATen/ functorch/ BatchedTensorImpl.cpp - aten/
src/ , C/C++, 177 linesATen/ functorch/ BatchedTensorImpl.h - aten/
src/ , C/C++, 126 linesATen/ functorch/ BatchingMetaprogramming. h - aten/
src/ , C++, 495 linesATen/ functorch/ DynamicLayer.cpp - aten/
src/ , C/C++, 126 linesATen/ functorch/ DynamicLayer.h - aten/
src/ , C++, 67 linesATen/ functorch/ FunctionalizeInterpreter .cpp - aten/
src/ , C/C++, 22 linesATen/ functorch/ FunctionalizeInterpreter .h - aten/
src/ , C++, 130 linesATen/ functorch/ Interpreter.cpp - aten/
src/ , C/C++, 361 linesATen/ functorch/ Interpreter.h - aten/
src/ , C++, 766 linesATen/ functorch/ LegacyBatchingRegistrati ons.cpp - aten/
src/ , C++, 229 linesATen/ functorch/ LegacyVmapTransforms.cpp - aten/
src/ , C/C++, 187 linesATen/ functorch/ LegacyVmapTransforms.h - aten/
src/ , C/C++, 3 linesATen/ functorch/ Macros.h - aten/
src/ , C++, 94 linesATen/ functorch/ PlumbingHelper.cpp - aten/
src/ , C/C++, 63 linesATen/ functorch/ PlumbingHelper.h - aten/
src/ , C++, 276 linesATen/ functorch/ PyTorchOperatorHacks.cpp - aten/
src/ , C++, 214 linesATen/ functorch/ TensorWrapper.cpp - aten/
src/ , C/C++, 103 linesATen/ functorch/ TensorWrapper.h - aten/
src/ , C++, 24 linesATen/ functorch/ VmapInterpreter.cpp - aten/
src/ , C/C++, 25 linesATen/ functorch/ VmapInterpreter.h - aten/
src/ , C++, 65 linesATen/ functorch/ VmapModeRegistrations.cp p - aten/
src/ , C/C++, 242 linesATen/ hip/ impl/ HIPAllocatorMasquerading AsCUDA.h - aten/
src/ , C++, 16 linesATen/ hip/ impl/ HIPCachingAllocatorMasqu eradingAsCUDA.cpp - aten/
src/ , C/C++, 55 linesATen/ hip/ impl/ HIPCachingAllocatorMasqu eradingAsCUDA.h - aten/
src/ , C/C++, 85 linesATen/ hip/ impl/ HIPGuardImplMasquerading AsCUDA.h - aten/
src/ , C/C++, 41 linesATen/ hip/ impl/ HIPStreamMasqueradingAsC UDA.h - aten/
src/ , C/C++, 14 linesATen/ jit_macros.h - aten/
src/ , C/C++, 38 linesATen/ jiterator_macros.h - aten/
src/ , C++, 28 linesATen/ metal/ Context.cpp - aten/
src/ , C/C++, 32 linesATen/ metal/ Context.h - aten/
src/ , C++, 64 linesATen/ miopen/ AutocastRNN.cpp - aten/
src/ , C++, 146 linesATen/ miopen/ Descriptors.cpp - aten/
src/ , C/C++, 207 linesATen/ miopen/ Descriptors.h - aten/
src/ , C/C++, 49 linesATen/ miopen/ Exceptions.h - aten/
src/ , C++, 58 linesATen/ miopen/ Handle.cpp - aten/
src/ , C/C++, 9 linesATen/ miopen/ Handle.h - aten/
src/ , C++, 26 linesATen/ miopen/ Types.cpp - aten/
src/ , C/C++, 13 linesATen/ miopen/ Types.h - aten/
src/ , C/C++, 18 linesATen/ miopen/ Utils.h - aten/
src/ , C/C++, 28 linesATen/ miopen/ miopen-wrapper.h - aten/
src/ , C/C++, 42 linesATen/ mkl/ Descriptors.h - aten/
src/ , C/C++, 65 linesATen/ mkl/ Exceptions.h - aten/
src/ , C/C++, 11 linesATen/ mkl/ Limits.h - aten/
src/ , C/C++, 9 linesATen/ mkl/ Sparse.h - aten/
src/ , C++, 370 linesATen/ mkl/ SparseBlas.cpp - aten/
src/ , C/C++, 226 linesATen/ mkl/ SparseBlas.h - aten/
src/ , C/C++, 146 linesATen/ mkl/ SparseDescriptors.h - aten/
src/ , C/C++, 17 linesATen/ mkl/ Utils.h - aten/
src/ , C++, 138 linesATen/ mps/ EmptyTensor.cpp - aten/
src/ , C/C++, 28 linesATen/ mps/ EmptyTensor.h - aten/
src/ , C/C++, 502 linesATen/ mps/ MPSAllocator.h - aten/
src/ , C/C++, 88 linesATen/ mps/ MPSAllocatorInterface.h - aten/
src/ , C/C++, 117 linesATen/ mps/ MPSDevice.h - aten/
src/ , C/C++, 153 linesATen/ mps/ MPSEvent.h - aten/
src/ , C/C++, 61 linesATen/ mps/ MPSGeneratorImpl.h - aten/
src/ , C/C++, 196 linesATen/ mps/ MPSGuardImpl.h - aten/
src/ , C/C++, 75 linesATen/ mps/ MPSHooks.h - aten/
src/ , C/C++, 502 linesATen/ mps/ MPSProfiler.h - aten/
src/ , C/C++, 212 linesATen/ mps/ MPSStream.h - aten/
src/ , C++, 847 linesATen/ native/ Activation.cpp - aten/
src/ , C/C++, 73 linesATen/ native/ Activation.h - aten/
src/ , C++, 150 linesATen/ native/ AdaptiveAveragePooling.c pp - aten/
src/ , C++, 349 linesATen/ native/ AdaptiveAveragePooling3d .cpp - aten/
src/ , C++, 91 linesATen/ native/ AdaptiveMaxPooling2d.cpp - aten/
src/ , C++, 446 linesATen/ native/ AdaptiveMaxPooling3d.cpp - aten/
src/ , C/C++, 40 linesATen/ native/ AdaptivePooling.h - aten/
src/ , C++, 146 linesATen/ native/ AffineGridGenerator.cpp - aten/
src/ , C++, 38 linesATen/ native/ AmpKernels.cpp - aten/
src/ , C/C++, 28 linesATen/ native/ AmpKernels.h - aten/
src/ , C++, 108 linesATen/ native/ AutogradComposite.cpp - aten/
src/ , C++, 221 linesATen/ native/ AveragePool2d.cpp - aten/
src/ , C++, 511 linesATen/ native/ AveragePool3d.cpp - aten/
src/ , C++, 4,043 linesATen/ native/ BatchLinearAlgebra.cpp - aten/
src/ , C/C++, 332 linesATen/ native/ BatchLinearAlgebra.h - aten/
src/ , C++, 1,244 linesATen/ native/ BatchLinearAlgebraKernel .cpp - aten/
src/ , C++, 1,659 linesATen/ native/ BinaryOps.cpp - aten/
src/ , C/C++, 122 linesATen/ native/ BinaryOps.h - aten/
src/ , C++, 228 linesATen/ native/ Blas.cpp - aten/
src/ , C++, 786 linesATen/ native/ BlasKernel.cpp - aten/
src/ , C++, 246 linesATen/ native/ Bucketization.cpp - aten/
src/ , C/C++, 173 linesATen/ native/ BucketizationUtils.h - aten/
src/ , C++, 1,515 linesATen/ native/ CPUBlas.cpp - aten/
src/ , C/C++, 344 linesATen/ native/ CPUBlas.h - aten/
src/ , C++, 343 linesATen/ native/ CPUFallback.cpp - aten/
src/ , C/C++, 46 linesATen/ native/ CPUFallback.h - aten/
src/ , C/C++, 13 linesATen/ native/ CanUse32BitIndexMath.h - aten/
src/ , C++, 106 linesATen/ native/ ChanelShuffle.cpp - aten/
src/ , C++, 215 linesATen/ native/ Col2Im.cpp - aten/
src/ , C++, 72 linesATen/ native/ ComparisonUtils.cpp - aten/
src/ , C/C++, 105 linesATen/ native/ ComplexHelper.h - aten/
src/ , C/C++, 34 linesATen/ native/ CompositeRandomAccessor. h - aten/
src/ , C/C++, 263 linesATen/ native/ CompositeRandomAccessorC ommon.h - aten/
src/ , C++, 87 linesATen/ native/ Constraints.cpp - aten/
src/ , C/C++, 446 linesATen/ native/ ConvUtils.h - aten/
src/ , C++, 2,400 linesATen/ native/ Convolution.cpp - aten/
src/ , C++, 753 linesATen/ native/ ConvolutionMM2d.cpp - aten/
src/ , C++, 863 linesATen/ native/ ConvolutionMM3d.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ ConvolutionMM3d.h - aten/
src/ , C++, 121 linesATen/ native/ ConvolutionTBC.cpp - aten/
src/ , C++, 388 linesATen/ native/ Copy.cpp - aten/
src/ , C/C++, 20 linesATen/ native/ Copy.h - aten/
src/ , C++, 189 linesATen/ native/ Correlation.cpp - aten/
src/ , C++, 90 linesATen/ native/ Cross.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ Cross.h - aten/
src/ , C/C++, 230 linesATen/ native/ DilatedConvolutionUtils. h - aten/
src/ , C++, 210 linesATen/ native/ DilatedMaxPool2d.cpp - aten/
src/ , C++, 292 linesATen/ native/ DilatedMaxPool3d.cpp - aten/
src/ , C++, 430 linesATen/ native/ DispatchStub.cpp - aten/
src/ , C/C++, 507 linesATen/ native/ DispatchStub.h - aten/
src/ , C++, 340 linesATen/ native/ Distance.cpp - aten/
src/ , C/C++, 20 linesATen/ native/ Distance.h - aten/
src/ , C/C++, 400 linesATen/ native/ DistributionTemplates.h - aten/
src/ , C++, 645 linesATen/ native/ Distributions.cpp - aten/
src/ , C/C++, 559 linesATen/ native/ Distributions.h - aten/
src/ , C++, 196 linesATen/ native/ Dropout.cpp - aten/
src/ , C++, 214 linesATen/ native/ Embedding.cpp - aten/
src/ , C++, 1,778 linesATen/ native/ EmbeddingBag.cpp - aten/
src/ , C/C++, 154 linesATen/ native/ EmbeddingBag.h - aten/
src/ , C++, 164 linesATen/ native/ Fill.cpp - aten/
src/ , C/C++, 21 linesATen/ native/ Fill.h - aten/
src/ , C++, 573 linesATen/ native/ ForeachOpsKernels.cpp - aten/
src/ , C/C++, 392 linesATen/ native/ ForeachUtils.h - aten/
src/ , C++, 400 linesATen/ native/ FractionalMaxPool2d.cpp - aten/
src/ , C++, 428 linesATen/ native/ FractionalMaxPool3d.cpp - aten/
src/ , C/C++, 31 linesATen/ native/ FractionalMaxPooling.h - aten/
src/ , C++, 118 linesATen/ native/ FunctionOfAMatrixUtils.c pp - aten/
src/ , C/C++, 20 linesATen/ native/ FunctionOfAMatrixUtils.h - aten/
src/ , C++, 82 linesATen/ native/ FusedAdagrad.cpp - aten/
src/ , C/C++, 20 linesATen/ native/ FusedAdagrad.h - aten/
src/ , C++, 172 linesATen/ native/ FusedAdam.cpp - aten/
src/ , C/C++, 28 linesATen/ native/ FusedAdam.h - aten/
src/ , C++, 84 linesATen/ native/ FusedSGD.cpp - aten/
src/ , C/C++, 21 linesATen/ native/ FusedSGD.h - aten/
src/ , C++, 158 linesATen/ native/ GatedLinearUnit.cpp - aten/
src/ , C/C++, 33 linesATen/ native/ Gelu.h - aten/
src/ , C++, 1,233 linesATen/ native/ GridSampler.cpp - aten/
src/ , C/C++, 298 linesATen/ native/ GridSampler.h - aten/
src/ , C/C++, 164 linesATen/ native/ GridSamplerUtils.h - aten/
src/ , C/C++, 208 linesATen/ native/ GroupedMMUtils.h - aten/
src/ , C++, 478 linesATen/ native/ Histogram.cpp - aten/
src/ , C/C++, 16 linesATen/ native/ Histogram.h - aten/
src/ , C++, 154 linesATen/ native/ Im2Col.cpp - aten/
src/ , C/C++, 41 linesATen/ native/ IndexKernel.h - aten/
src/ , C++, 33 linesATen/ native/ IndexingUtils.cpp - aten/
src/ , C/C++, 186 linesATen/ native/ IndexingUtils.h - aten/
src/ , C++, 170 linesATen/ native/ Integration.cpp - aten/
src/ , C++, 91 linesATen/ native/ Itertools.cpp - aten/
src/ , C++, 122 linesATen/ native/ LegacyBatching.cpp - aten/
src/ , C++, 59 linesATen/ native/ Lerp.cpp - aten/
src/ , C/C++, 46 linesATen/ native/ Lerp.h - aten/
src/ , C++, 926 linesATen/ native/ Linear.cpp - aten/
src/ , C++, 3,801 linesATen/ native/ LinearAlgebra.cpp - aten/
src/ , C/C++, 17 linesATen/ native/ LinearAlgebra.h - aten/
src/ , C/C++, 656 linesATen/ native/ LinearAlgebraUtils.h - aten/
src/ , C++, 503 linesATen/ native/ Loss.cpp - aten/
src/ , C++, 595 linesATen/ native/ LossCTC.cpp - aten/
src/ , C/C++, 71 linesATen/ native/ LossMulti.h - aten/
src/ , C++, 329 linesATen/ native/ LossMultiLabelMargin.cpp - aten/
src/ , C++, 334 linesATen/ native/ LossMultiMargin.cpp - aten/
src/ , C++, 756 linesATen/ native/ LossNLL.cpp - aten/
src/ , C++, 490 linesATen/ native/ LossNLL2d.cpp - aten/
src/ , C/C++, 3,927 linesATen/ native/ Math.h - aten/
src/ , C/C++, 65 linesATen/ native/ MathBitFallThroughLists. h - aten/
src/ , C/C++, 157 linesATen/ native/ MathBitsFallback.h - aten/
src/ , C++, 93 linesATen/ native/ MaxPooling.cpp - aten/
src/ , C/C++, 97 linesATen/ native/ MaxPooling.h - aten/
src/ , C++, 111 linesATen/ native/ MaxUnpooling.cpp - aten/
src/ , C++, 75 linesATen/ native/ Memory.cpp - aten/
src/ , C++, 43 linesATen/ native/ MetaTensor.cpp - aten/
src/ , C++, 315 linesATen/ native/ NNPACK.cpp - aten/
src/ , C++, 875 linesATen/ native/ NaiveConvolutionTranspos e2d.cpp - aten/
src/ , C++, 952 linesATen/ native/ NaiveConvolutionTranspos e3d.cpp - aten/
src/ , C++, 747 linesATen/ native/ NaiveDilatedConvolution. cpp - aten/
src/ , C++, 50 linesATen/ native/ NegateFallback.cpp - aten/
src/ , C/C++, 27 linesATen/ native/ NonEmptyUtils.h - aten/
src/ , C/C++, 26 linesATen/ native/ NonSymbolicBC.h - aten/
src/ , C++, 1,040 linesATen/ native/ Normalization.cpp - aten/
src/ , C/C++, 19 linesATen/ native/ Normalization.h - aten/
src/ , C++, 75 linesATen/ native/ Onehot.cpp - aten/
src/ , C++, 243 linesATen/ native/ PackedSequence.cpp - aten/
src/ , C++, 271 linesATen/ native/ PadNd.cpp - aten/
src/ , C/C++, 148 linesATen/ native/ Padding.h - aten/
src/ , C++, 357 linesATen/ native/ PhiloxStatelessRNG.cpp - aten/
src/ , C/C++, 21 linesATen/ native/ PhiloxStatelessRNG.h - aten/
src/ , C++, 159 linesATen/ native/ PixelShuffle.cpp - aten/
src/ , C/C++, 48 linesATen/ native/ PixelShuffle.h - aten/
src/ , C++, 78 linesATen/ native/ PointwiseOps.cpp - aten/
src/ , C/C++, 28 linesATen/ native/ PointwiseOps.h - aten/
src/ , C/C++, 137 linesATen/ native/ Pool.h - aten/
src/ , C++, 182 linesATen/ native/ Pooling.cpp - aten/
src/ , C/C++, 331 linesATen/ native/ PoolingChecks.h - aten/
src/ , C++, 138 linesATen/ native/ Pow.cpp - aten/
src/ , C/C++, 69 linesATen/ native/ Pow.h - aten/
src/ , C++, 636 linesATen/ native/ QuantizedLinear.cpp - aten/
src/ , C++, 2,125 linesATen/ native/ RNN.cpp - aten/
src/ , C/C++, 60 linesATen/ native/ RNN.h - aten/
src/ , C++, 328 linesATen/ native/ RangeFactories.cpp - aten/
src/ , C/C++, 12 linesATen/ native/ RangeFactories.h - aten/
src/ , C/C++, 67 linesATen/ native/ RangeUtils.h - aten/
src/ , C++, 74 linesATen/ native/ ReduceAllOps.cpp - aten/
src/ , C/C++, 16 linesATen/ native/ ReduceAllOps.h - aten/
src/ , C++, 2,385 linesATen/ native/ ReduceOps.cpp - aten/
src/ , C/C++, 58 linesATen/ native/ ReduceOps.h - aten/
src/ , C/C++, 471 linesATen/ native/ ReduceOpsUtils.h - aten/
src/ , C/C++, 40 linesATen/ native/ ReductionType.h - aten/
src/ , C++, 187 linesATen/ native/ ReflectionPad.cpp - aten/
src/ , C++, 142 linesATen/ native/ Repeat.cpp - aten/
src/ , C/C++, 52 linesATen/ native/ Repeat.h - aten/
src/ , C++, 183 linesATen/ native/ ReplicationPadding.cpp - aten/
src/ , C++, 321 linesATen/ native/ Resize.cpp - aten/
src/ , C/C++, 275 linesATen/ native/ Resize.h - aten/
src/ , C/C++, 51 linesATen/ native/ ResizeCommon.h - aten/
src/ , C++, 115 linesATen/ native/ RowwisePrune.cpp - aten/
src/ , C++, 60 linesATen/ native/ Scalar.cpp - aten/
src/ , C++, 673 linesATen/ native/ ScaledBlas.cpp - aten/
src/ , C++, 629 linesATen/ native/ ScaledBlasUtils.cpp - aten/
src/ , C/C++, 218 linesATen/ native/ ScaledBlasUtils.h - aten/
src/ , C/C++, 128 linesATen/ native/ ScatterGatherChecks.h - aten/
src/ , C++, 520 linesATen/ native/ SegmentReduce.cpp - aten/
src/ , C/C++, 50 linesATen/ native/ SegmentReduce.h - aten/
src/ , C++, 64 linesATen/ native/ ShallowCopyData.cpp - aten/
src/ , C/C++, 546 linesATen/ native/ SharedReduceOps.h - aten/
src/ , C++, 238 linesATen/ native/ SobolEngineOps.cpp - aten/
src/ , C++, 17,029 linesATen/ native/ SobolEngineOpsUtils.cpp - aten/
src/ , C/C++, 55 linesATen/ native/ SobolEngineOpsUtils.h - aten/
src/ , C++, 613 linesATen/ native/ SoftMax.cpp - aten/
src/ , C++, 1,075 linesATen/ native/ Sorting.cpp - aten/
src/ , C/C++, 28 linesATen/ native/ Sorting.h - aten/
src/ , C/C++, 88 linesATen/ native/ SortingUtils.h - aten/
src/ , C++, 143 linesATen/ native/ SparseTensorUtils.cpp - aten/
src/ , C/C++, 190 linesATen/ native/ SparseTensorUtils.h - aten/
src/ , C++, 1,323 linesATen/ native/ SpectralOps.cpp - aten/
src/ , C/C++, 84 linesATen/ native/ SpectralOpsUtils.h - aten/
src/ , C/C++, 301 linesATen/ native/ StridedRandomAccessor.h - aten/
src/ , C++, 98 linesATen/ native/ SummaryOps.cpp - aten/
src/ , C++, 3,173 linesATen/ native/ TensorAdvancedIndexing.c pp - aten/
src/ , C/C++, 102 linesATen/ native/ TensorAdvancedIndexing.h - aten/
src/ , C/C++, 71 linesATen/ native/ TensorAdvancedIndexingUt ils.h - aten/
src/ , C++, 1,119 linesATen/ native/ TensorCompare.cpp - aten/
src/ , C/C++, 56 linesATen/ native/ TensorCompare.h - aten/
src/ , C++, 2,520 linesATen/ native/ TensorConversions.cpp - aten/
src/ , C/C++, 31 linesATen/ native/ TensorConversions.h - aten/
src/ , C/C++, 67 linesATen/ native/ TensorDimApply.h - aten/
src/ , C++, 2,278 linesATen/ native/ TensorFactories.cpp - aten/
src/ , C/C++, 170 linesATen/ native/ TensorFactories.h - aten/
src/ , C/C++, 2 linesATen/ native/ TensorIterator.h - aten/
src/ , C/C++, 52 linesATen/ native/ TensorIteratorDynamicCas ting.h - aten/
src/ , C++, 195 linesATen/ native/ TensorIteratorReduce.cpp - aten/
src/ , C++, 163 linesATen/ native/ TensorProperties.cpp - aten/
src/ , C/C++, 12 linesATen/ native/ TensorProperties.h - aten/
src/ , C++, 4,875 linesATen/ native/ TensorShape.cpp - aten/
src/ , C/C++, 144 linesATen/ native/ TensorShape.h - aten/
src/ , C++, 256 linesATen/ native/ TensorTransformations.cp p - aten/
src/ , C/C++, 35 linesATen/ native/ TensorTransformations.h - aten/
src/ , C++, 142 linesATen/ native/ TestOps.cpp - aten/
src/ , C/C++, 98 linesATen/ native/ TopKImpl.h - aten/
src/ , C/C++, 27 linesATen/ native/ TransposeType.h - aten/
src/ , C++, 202 linesATen/ native/ TriangularOps.cpp - aten/
src/ , C/C++, 57 linesATen/ native/ TriangularOpsUtils.h - aten/
src/ , C/C++, 86 linesATen/ native/ TunableOp.h - aten/
src/ , C++, 191 linesATen/ native/ TypeProperties.cpp - aten/
src/ , C/C++, 20 linesATen/ native/ TypeProperties.h - aten/
src/ , C++, 1,041 linesATen/ native/ UnaryOps.cpp - aten/
src/ , C/C++, 128 linesATen/ native/ UnaryOps.h - aten/
src/ , C++, 9 linesATen/ native/ Unfold2d.cpp - aten/
src/ , C/C++, 48 linesATen/ native/ Unfold2d.h - aten/
src/ , C++, 531 linesATen/ native/ Unfold3d.cpp - aten/
src/ , C/C++, 49 linesATen/ native/ Unfold3d.h - aten/
src/ , C++, 41 linesATen/ native/ UnfoldBackward.cpp - aten/
src/ , C/C++, 110 linesATen/ native/ UnfoldBackward.h - aten/
src/ , C++, 504 linesATen/ native/ Unique.cpp - aten/
src/ , C++, 34 linesATen/ native/ UpSample.cpp - aten/
src/ , C/C++, 512 linesATen/ native/ UpSample.h - aten/
src/ , C++, 303 linesATen/ native/ UpSampleBicubic2d.cpp - aten/
src/ , C++, 187 linesATen/ native/ UpSampleBilinear2d.cpp - aten/
src/ , C++, 102 linesATen/ native/ UpSampleLanczos2d.cpp - aten/
src/ , C++, 104 linesATen/ native/ UpSampleLinear1d.cpp - aten/
src/ , C++, 148 linesATen/ native/ UpSampleNearest1d.cpp - aten/
src/ , C++, 174 linesATen/ native/ UpSampleNearest2d.cpp - aten/
src/ , C++, 191 linesATen/ native/ UpSampleNearest3d.cpp - aten/
src/ , C++, 116 linesATen/ native/ UpSampleTrilinear3d.cpp - aten/
src/ , C++, 75 linesATen/ native/ VariableMethodStubs.cpp - aten/
src/ , C++, 159 linesATen/ native/ WeightNorm.cpp - aten/
src/ , C++, 27 linesATen/ native/ ao_sparse/ library.cpp - aten/
src/ , C++, 44 linesATen/ native/ ao_sparse/ quantized/ cpu/ fbgemm_utils.cpp - aten/
src/ , C/C++, 97 linesATen/ native/ ao_sparse/ quantized/ cpu/ fbgemm_utils.h - aten/
src/ , C/C++, 73 linesATen/ native/ ao_sparse/ quantized/ cpu/ packed_params.h - aten/
src/ , C++, 259 linesATen/ native/ ao_sparse/ quantized/ cpu/ qlinear.cpp - aten/
src/ , C++, 316 linesATen/ native/ ao_sparse/ quantized/ cpu/ qlinear_deserialize.cpp - aten/
src/ , C++, 195 linesATen/ native/ ao_sparse/ quantized/ cpu/ qlinear_dynamic.cpp - aten/
src/ , C++, 247 linesATen/ native/ ao_sparse/ quantized/ cpu/ qlinear_prepack.cpp - aten/
src/ , C++, 247 linesATen/ native/ ao_sparse/ quantized/ cpu/ qlinear_serialize.cpp - aten/
src/ , C++, 141 linesATen/ native/ ao_sparse/ quantized/ cpu/ qlinear_unpack.cpp - aten/
src/ , C/C++, 90 linesATen/ native/ ao_sparse/ quantized/ cpu/ qnnpack_utils.h - aten/
src/ , C/C++, 38 linesATen/ native/ batch_norm.h - aten/
src/ , C++, 1,344 linesATen/ native/ cpu/ Activation.cpp - aten/
src/ , C++, 862 linesATen/ native/ cpu/ AdaptiveAvgPoolKernel.cp p - aten/
src/ , C++, 988 linesATen/ native/ cpu/ AdaptiveMaxPoolKernel.cp p - aten/
src/ , C++, 198 linesATen/ native/ cpu/ AmpGradScalerKernels.cpp - aten/
src/ , C/C++, 36 linesATen/ native/ cpu/ AtomicAddFloat.h - aten/
src/ , C++, 1,138 linesATen/ native/ cpu/ AvgPoolKernel.cpp - aten/
src/ , C++, 1,516 linesATen/ native/ cpu/ BinaryOpsKernel.cpp - aten/
src/ , C++, 561 linesATen/ native/ cpu/ BlasKernel.cpp - aten/
src/ , C++, 83 linesATen/ native/ cpu/ CatKernel.cpp - aten/
src/ , C/C++, 12 linesATen/ native/ cpu/ CatKernel.h - aten/
src/ , C++, 116 linesATen/ native/ cpu/ ChannelShuffleKernel.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ cpu/ ChannelShuffleKernel.h - aten/
src/ , C++, 36 linesATen/ native/ cpu/ ComplexKernel.cpp - aten/
src/ , C++, 330 linesATen/ native/ cpu/ CopyKernel.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ cpu/ CopyKernel.h - aten/
src/ , C++, 81 linesATen/ native/ cpu/ CrossKernel.cpp - aten/
src/ , C++, 546 linesATen/ native/ cpu/ DepthwiseConvKernel.cpp - aten/
src/ , C/C++, 21 linesATen/ native/ cpu/ DepthwiseConvKernel.h - aten/
src/ , C++, 450 linesATen/ native/ cpu/ DistanceOpsKernel.cpp - aten/
src/ , C++, 253 linesATen/ native/ cpu/ DistributionKernels.cpp - aten/
src/ , C/C++, 411 linesATen/ native/ cpu/ DistributionTemplates.h - aten/
src/ , C/C++, 74 linesATen/ native/ cpu/ Elu.h - aten/
src/ , C++, 75 linesATen/ native/ cpu/ FillKernel.cpp - aten/
src/ , C++, 1,277 linesATen/ native/ cpu/ FlashAttentionKernel.cpp - aten/
src/ , C++, 57 linesATen/ native/ cpu/ FunctionOfAMatrixUtilsKe rnel.cpp - aten/
src/ , C++, 218 linesATen/ native/ cpu/ FusedAdagradKernel.cpp - aten/
src/ , C++, 368 linesATen/ native/ cpu/ FusedAdamKernel.cpp - aten/
src/ , C++, 273 linesATen/ native/ cpu/ FusedSGDKernel.cpp - aten/
src/ , C/C++, 83 linesATen/ native/ cpu/ Gelu.h - aten/
src/ , C++, 1,317 linesATen/ native/ cpu/ GridSamplerKernel.cpp - aten/
src/ , C/C++, 34 linesATen/ native/ cpu/ GridSamplerKernel.h - aten/
src/ , C++, 314 linesATen/ native/ cpu/ HistogramKernel.cpp - aten/
src/ , C++, 832 linesATen/ native/ cpu/ IndexKernel.cpp - aten/
src/ , C/C++, 104 linesATen/ native/ cpu/ IndexKernelUtils.h - aten/
src/ , C/C++, 33 linesATen/ native/ cpu/ Intrinsics.h - aten/
src/ , C/C++, 64 linesATen/ native/ cpu/ IsContiguous.h - aten/
src/ , C++, 165 linesATen/ native/ cpu/ LerpKernel.cpp - aten/
src/ , C++, 98 linesATen/ native/ cpu/ LinearAlgebraKernel.cpp - aten/
src/ , C/C++, 61 linesATen/ native/ cpu/ LogAddExp.h - aten/
src/ , C/C++, 337 linesATen/ native/ cpu/ LogSoftmaxKernelImpl.h - aten/
src/ , C/C++, 431 linesATen/ native/ cpu/ Loops.h - aten/
src/ , C++, 747 linesATen/ native/ cpu/ MaxPoolKernel.cpp - aten/
src/ , C++, 64 linesATen/ native/ cpu/ MaxPooling.cpp - aten/
src/ , C++, 278 linesATen/ native/ cpu/ MaxUnpoolKernel.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ cpu/ MaxUnpoolKernel.h - aten/
src/ , C++, 231 linesATen/ native/ cpu/ MultinomialKernel.cpp - aten/
src/ , C++, 111 linesATen/ native/ cpu/ NativeMultiheadAttnKerne l.cpp - aten/
src/ , C++, 729 linesATen/ native/ cpu/ PaddingKernel.cpp - aten/
src/ , C++, 253 linesATen/ native/ cpu/ PixelShuffleKernel.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ cpu/ PixelShuffleKernel.h - aten/
src/ , C++, 265 linesATen/ native/ cpu/ PointwiseOpsKernel.cpp - aten/
src/ , C++, 183 linesATen/ native/ cpu/ PowKernel.cpp - aten/
src/ , C++, 100 linesATen/ native/ cpu/ RangeFactoriesKernel.cpp - aten/
src/ , C/C++, 310 linesATen/ native/ cpu/ Reduce.h - aten/
src/ , C++, 227 linesATen/ native/ cpu/ ReduceAllOpsKernel.cpp - aten/
src/ , C++, 579 linesATen/ native/ cpu/ ReduceOpsKernel.cpp - aten/
src/ , C/C++, 237 linesATen/ native/ cpu/ ReduceUtils.h - aten/
src/ , C++, 502 linesATen/ native/ cpu/ ReducedPrecisionFloatGem vFastPathKernel.cpp - aten/
src/ , C/C++, 27 linesATen/ native/ cpu/ ReducedPrecisionFloatGem vFastPathKernel.h - aten/
src/ , C++, 38 linesATen/ native/ cpu/ RenormKernel.cpp - aten/
src/ , C++, 99 linesATen/ native/ cpu/ SampledAddmmKernel.cpp - aten/
src/ , C/C++, 12 linesATen/ native/ cpu/ SampledAddmmKernel.h - aten/
src/ , C++, 1,208 linesATen/ native/ cpu/ ScatterGatherKernel.cpp - aten/
src/ , C/C++, 146 linesATen/ native/ cpu/ SerialStackImpl.h - aten/
src/ , C++, 1,074 linesATen/ native/ cpu/ SoftMaxKernel.cpp - aten/
src/ , C/C++, 28 linesATen/ native/ cpu/ SoftmaxKernel.h - aten/
src/ , C++, 270 linesATen/ native/ cpu/ SortingKernel.cpp - aten/
src/ , C++, 65 linesATen/ native/ cpu/ SparseFactories.cpp - aten/
src/ , C++, 565 linesATen/ native/ cpu/ SpmmReduceKernel.cpp - aten/
src/ , C/C++, 22 linesATen/ native/ cpu/ SpmmReduceKernel.h - aten/
src/ , C++, 24 linesATen/ native/ cpu/ StackKernel.cpp - aten/
src/ , C/C++, 12 linesATen/ native/ cpu/ StackKernel.h - aten/
src/ , C++, 679 linesATen/ native/ cpu/ SumKernel.cpp - aten/
src/ , C++, 467 linesATen/ native/ cpu/ TensorCompareKernel.cpp - aten/
src/ , C++, 937 linesATen/ native/ cpu/ UnaryOpsKernel.cpp - aten/
src/ , C++, 445 linesATen/ native/ cpu/ Unfold2d.cpp - aten/
src/ , C++, 152 linesATen/ native/ cpu/ UnfoldBackwardKernel.cpp - aten/
src/ , C++, 2,226 linesATen/ native/ cpu/ UpSampleKernel.cpp - aten/
src/ , C/C++, 1,376 linesATen/ native/ cpu/ UpSampleKernelAVXAntiali as.h - aten/
src/ , C/C++, 381 linesATen/ native/ cpu/ UpSampleKernelNEONAntial ias.h - aten/
src/ , C++, 797 linesATen/ native/ cpu/ UpSampleMoreKernel.cpp - aten/
src/ , C++, 463 linesATen/ native/ cpu/ WeightNormKernel.cpp - aten/
src/ , C/C++, 20 linesATen/ native/ cpu/ WeightNormKernel.h - aten/
src/ , C++, 24 linesATen/ native/ cpu/ airy_ai.cpp - aten/
src/ , C/C++, 522 linesATen/ native/ cpu/ avx_mathfun.h - aten/
src/ , C++, 1,403 linesATen/ native/ cpu/ batch_norm_kernel.cpp - aten/
src/ , C++, 1,680 linesATen/ native/ cpu/ group_norm_kernel.cpp - aten/
src/ , C++, 1,383 linesATen/ native/ cpu/ int4mm_kernel.cpp - aten/
src/ , C++, 438 linesATen/ native/ cpu/ int8mm_kernel.cpp - aten/
src/ , C/C++, 38 linesATen/ native/ cpu/ int_mm_kernel.h - aten/
src/ , C++, 633 linesATen/ native/ cpu/ layer_norm_kernel.cpp - aten/
src/ , C/C++, 41 linesATen/ native/ cpu/ mixed_data_type.h - aten/
src/ , C/C++, 211 linesATen/ native/ cpu/ moments_utils.h - aten/
src/ , C++, 24 linesATen/ native/ cpu/ scaled_modified_bessel_k 0.cpp - aten/
src/ , C++, 24 linesATen/ native/ cpu/ scaled_modified_bessel_k 1.cpp - aten/
src/ , C++, 24 linesATen/ native/ cpu/ spherical_bessel_j0.cpp - aten/
src/ , C/C++, 219 linesATen/ native/ cpu/ utils.h - aten/
src/ , C/C++, 250 linesATen/ native/ cpu/ zmath.h - aten/
src/ , CUDA, 55 linesATen/ native/ cuda/ AbsKernel.cu - aten/
src/ , C++, 107 linesATen/ native/ cuda/ Activation.cpp - aten/
src/ , C/C++, 20 linesATen/ native/ cuda/ Activation.h - aten/
src/ , CUDA, 86 linesATen/ native/ cuda/ ActivationEluKernel.cu - aten/
src/ , CUDA, 88 linesATen/ native/ cuda/ ActivationGeluKernel.cu - aten/
src/ , CUDA, 141 linesATen/ native/ cuda/ ActivationGluKernel.cu - aten/
src/ , CUDA, 39 linesATen/ native/ cuda/ ActivationHardshrinkKern el.cu - aten/
src/ , CUDA, 74 linesATen/ native/ cuda/ ActivationHardsigmoidKer nel.cu - aten/
src/ , CUDA, 63 linesATen/ native/ cuda/ ActivationHardswishKerne l.cu - aten/
src/ , CUDA, 45 linesATen/ native/ cuda/ ActivationHardtanhKernel .cu - aten/
src/ , CUDA, 62 linesATen/ native/ cuda/ ActivationLeakyReluKerne l.cu - aten/
src/ , CUDA, 64 linesATen/ native/ cuda/ ActivationLogSigmoidKern el.cu - aten/
src/ , CUDA, 64 linesATen/ native/ cuda/ ActivationMishKernel.cu - aten/
src/ , CUDA, 48 linesATen/ native/ cuda/ ActivationPreluKernel.cu - aten/
src/ , CUDA, 60 linesATen/ native/ cuda/ ActivationSiluKernel.cu - aten/
src/ , CUDA, 74 linesATen/ native/ cuda/ ActivationSoftplusKernel .cu - aten/
src/ , CUDA, 59 linesATen/ native/ cuda/ ActivationSoftshrinkKern el.cu - aten/
src/ , CUDA, 52 linesATen/ native/ cuda/ ActivationThresholdKerne l.cu - aten/
src/ , CUDA, 840 linesATen/ native/ cuda/ AdaptiveAveragePooling.c u - aten/
src/ , CUDA, 551 linesATen/ native/ cuda/ AdaptiveAveragePooling3d .cu - aten/
src/ , CUDA, 478 linesATen/ native/ cuda/ AdaptiveMaxPooling2d.cu - aten/
src/ , CUDA, 489 linesATen/ native/ cuda/ AdaptiveMaxPooling3d.cu - aten/
src/ , CUDA, 252 linesATen/ native/ cuda/ AmpKernels.cu - aten/
src/ , CUDA, 544 linesATen/ native/ cuda/ AveragePool2d.cu - aten/
src/ , CUDA, 620 linesATen/ native/ cuda/ AveragePool3d.cu - aten/
src/ , CUDA, 119 linesATen/ native/ cuda/ BatchLinearAlgebraEig.cu - aten/
src/ , CUDA, 81 linesATen/ native/ cuda/ BinaryBitwiseOpsKernels. cu - aten/
src/ , CUDA, 82 linesATen/ native/ cuda/ BinaryDivFloorKernel.cu - aten/
src/ , CUDA, 65 linesATen/ native/ cuda/ BinaryDivTrueKernel.cu - aten/
src/ , CUDA, 53 linesATen/ native/ cuda/ BinaryDivTruncKernel.cu - aten/
src/ , CUDA, 39 linesATen/ native/ cuda/ BinaryGeometricKernels.c u - aten/
src/ , C/C++, 44 linesATen/ native/ cuda/ BinaryInternal.h - aten/
src/ , CUDA, 128 linesATen/ native/ cuda/ BinaryLogicalOpsKernels. cu - aten/
src/ , CUDA, 131 linesATen/ native/ cuda/ BinaryMiscBackwardOpsKer nels.cu - aten/
src/ , CUDA, 90 linesATen/ native/ cuda/ BinaryMiscOpsKernels.cu - aten/
src/ , CUDA, 48 linesATen/ native/ cuda/ BinaryMulKernel.cu - aten/
src/ , CUDA, 82 linesATen/ native/ cuda/ BinaryRemainderKernel.cu - aten/
src/ , CUDA, 44 linesATen/ native/ cuda/ BinaryShiftOpsKernels.cu - aten/
src/ , C++, 1,222 linesATen/ native/ cuda/ Blas.cpp - aten/
src/ , CUDA, 233 linesATen/ native/ cuda/ Bucketization.cu - aten/
src/ , CUDA, 84 linesATen/ native/ cuda/ CUDAScalar.cu - aten/
src/ , CUDA, 171 linesATen/ native/ cuda/ Col2Im.cu - aten/
src/ , CUDA, 50 linesATen/ native/ cuda/ CompareEQKernel.cu - aten/
src/ , CUDA, 103 linesATen/ native/ cuda/ CompareKernels.cu - aten/
src/ , CUDA, 46 linesATen/ native/ cuda/ ComplexKernel.cu - aten/
src/ , C/C++, 36 linesATen/ native/ cuda/ CompositeRandomAccessor. h - aten/
src/ , CUDA, 502 linesATen/ native/ cuda/ ConvolutionMM2d.cu - aten/
src/ , CUDA, 668 linesATen/ native/ cuda/ Copy.cu - aten/
src/ , C/C++, 11 linesATen/ native/ cuda/ Copy.h - aten/
src/ , CUDA, 60 linesATen/ native/ cuda/ CopysignKernel.cu - aten/
src/ , CUDA, 92 linesATen/ native/ cuda/ CrossKernel.cu - aten/
src/ , C/C++, 519 linesATen/ native/ cuda/ CuFFTPlanCache.h - aten/
src/ , C/C++, 75 linesATen/ native/ cuda/ CuFFTUtils.h - aten/
src/ , CUDA, 328 linesATen/ native/ cuda/ CublasGroupedArgs.cu - aten/
src/ , C/C++, 53 linesATen/ native/ cuda/ CublasGroupedArgs.h - aten/
src/ , CUDA, 29 linesATen/ native/ cuda/ CumminmaxKernel.cu - aten/
src/ , CUDA, 23 linesATen/ native/ cuda/ CumprodKernel.cu - aten/
src/ , CUDA, 25 linesATen/ native/ cuda/ CumsumKernel.cu - aten/
src/ , CUDA, 765 linesATen/ native/ cuda/ DepthwiseConv2d.cu - aten/
src/ , CUDA, 705 linesATen/ native/ cuda/ DepthwiseConv3d.cu - aten/
src/ , CUDA, 788 linesATen/ native/ cuda/ DilatedMaxPool2d.cu - aten/
src/ , CUDA, 653 linesATen/ native/ cuda/ DilatedMaxPool3d.cu - aten/
src/ , CUDA, 395 linesATen/ native/ cuda/ DistanceKernel.cu - aten/
src/ , CUDA, 39 linesATen/ native/ cuda/ DistributionBernoulli.cu - aten/
src/ , CUDA, 15 linesATen/ native/ cuda/ DistributionCauchyKernel .cu - aten/
src/ , CUDA, 15 linesATen/ native/ cuda/ DistributionExponentialK ernel.cu - aten/
src/ , CUDA, 15 linesATen/ native/ cuda/ DistributionGeometricKer nel.cu - aten/
src/ , CUDA, 15 linesATen/ native/ cuda/ DistributionLogNormalKer nel.cu - aten/
src/ , CUDA, 15 linesATen/ native/ cuda/ DistributionNormal.cu - aten/
src/ , CUDA, 27 linesATen/ native/ cuda/ DistributionRandomKernel .cu - aten/
src/ , C/C++, 697 linesATen/ native/ cuda/ DistributionTemplates.h - aten/
src/ , CUDA, 15 linesATen/ native/ cuda/ DistributionUniform.cu - aten/
src/ , C++, 96 linesATen/ native/ cuda/ Distributions.cpp - aten/
src/ , CUDA, 203 linesATen/ native/ cuda/ Distributions.cu - aten/
src/ , C/C++, 25 linesATen/ native/ cuda/ Distributions.h - aten/
src/ , CUDA, 464 linesATen/ native/ cuda/ Dropout.cu - aten/
src/ , CUDA, 421 linesATen/ native/ cuda/ Embedding.cu - aten/
src/ , CUDA, 495 linesATen/ native/ cuda/ EmbeddingBackwardKernel. cu - aten/
src/ , CUDA, 582 linesATen/ native/ cuda/ EmbeddingBag.cu - aten/
src/ , C++, 40 linesATen/ native/ cuda/ Equal.cpp - aten/
src/ , CUDA, 30 linesATen/ native/ cuda/ FillKernel.cu - aten/
src/ , CUDA, 28 linesATen/ native/ cuda/ FlattenIndicesKernel.cu - aten/
src/ , CUDA, 480 linesATen/ native/ cuda/ ForeachBinaryOpList.cu - aten/
src/ , CUDA, 281 linesATen/ native/ cuda/ ForeachBinaryOpScalar.cu - aten/
src/ , CUDA, 239 linesATen/ native/ cuda/ ForeachBinaryOpScalarLis t.cu - aten/
src/ , CUDA, 204 linesATen/ native/ cuda/ ForeachBinaryOpScalarTen sor.cu - aten/
src/ , CUDA, 395 linesATen/ native/ cuda/ ForeachPointwiseOp.cu - aten/
src/ , CUDA, 698 linesATen/ native/ cuda/ ForeachReduceOp.cu - aten/
src/ , CUDA, 228 linesATen/ native/ cuda/ ForeachTernaryOp.cu - aten/
src/ , CUDA, 478 linesATen/ native/ cuda/ ForeachUnaryOp.cu - aten/
src/ , CUDA, 274 linesATen/ native/ cuda/ FractionalMaxPool2d.cu - aten/
src/ , CUDA, 348 linesATen/ native/ cuda/ FractionalMaxPool3d.cu - aten/
src/ , CUDA, 114 linesATen/ native/ cuda/ FunctionOfAMatrixUtilsKe rnel.cu - aten/
src/ , CUDA, 99 linesATen/ native/ cuda/ FusedAdagradKernel.cu - aten/
src/ , CUDA, 186 linesATen/ native/ cuda/ FusedAdamKernel.cu - aten/
src/ , CUDA, 186 linesATen/ native/ cuda/ FusedAdamWKernel.cu - aten/
src/ , CUDA, 431 linesATen/ native/ cuda/ FusedSgdKernel.cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ GcdLcmKernel.cu - aten/
src/ , C++, 83 linesATen/ native/ cuda/ GridSampler.cpp - aten/
src/ , CUDA, 1,220 linesATen/ native/ cuda/ GridSampler.cu - aten/
src/ , C/C++, 31 linesATen/ native/ cuda/ GridSampler.h - aten/
src/ , CUDA, 439 linesATen/ native/ cuda/ GroupMM.cu - aten/
src/ , C/C++, 12 linesATen/ native/ cuda/ GroupMM.h - aten/
src/ , C++, 782 linesATen/ native/ cuda/ GroupedBlas.cpp - aten/
src/ , CUDA, 554 linesATen/ native/ cuda/ IGammaKernel.cu - aten/
src/ , CUDA, 165 linesATen/ native/ cuda/ Im2Col.cu - aten/
src/ , C++, 80 linesATen/ native/ cuda/ IndexKernel.cpp - aten/
src/ , CUDA, 523 linesATen/ native/ cuda/ IndexKernel.cu - aten/
src/ , C/C++, 15 linesATen/ native/ cuda/ IndexKernel.h - aten/
src/ , CUDA, 313 linesATen/ native/ cuda/ IndexKernelUtils.cu - aten/
src/ , C/C++, 69 linesATen/ native/ cuda/ IndexKernelUtils.h - aten/
src/ , CUDA, 2,046 linesATen/ native/ cuda/ Indexing.cu - aten/
src/ , C/C++, 16 linesATen/ native/ cuda/ LaunchUtils.h - aten/
src/ , CUDA, 149 linesATen/ native/ cuda/ Lerp.cu - aten/
src/ , CUDA, 144 linesATen/ native/ cuda/ LinearAlgebra.cu - aten/
src/ , C++, 192 linesATen/ native/ cuda/ LinearAlgebraStubs.cpp - aten/
src/ , CUDA, 305 linesATen/ native/ cuda/ LogAddExpKernel.cu - aten/
src/ , CUDA, 133 linesATen/ native/ cuda/ LogcumsumexpKernel.cu - aten/
src/ , CUDA, 632 linesATen/ native/ cuda/ Loss.cu - aten/
src/ , CUDA, 814 linesATen/ native/ cuda/ LossCTC.cu - aten/
src/ , CUDA, 99 linesATen/ native/ cuda/ MaxMinElementwiseKernel. cu - aten/
src/ , CUDA, 500 linesATen/ native/ cuda/ MaxUnpooling.cu - aten/
src/ , C/C++, 31 linesATen/ native/ cuda/ MiscUtils.h - aten/
src/ , CUDA, 353 linesATen/ native/ cuda/ MixedDtypesLinear.cu - aten/
src/ , CUDA, 442 linesATen/ native/ cuda/ MultiLabelMarginCriterio n.cu - aten/
src/ , CUDA, 378 linesATen/ native/ cuda/ MultiMarginLoss.cu - aten/
src/ , CUDA, 452 linesATen/ native/ cuda/ MultinomialKernel.cu - aten/
src/ , CUDA, 551 linesATen/ native/ cuda/ NLLLoss2d.cu - aten/
src/ , CUDA, 833 linesATen/ native/ cuda/ NaiveConvolutionTranspos e2d.cu - aten/
src/ , CUDA, 1,022 linesATen/ native/ cuda/ NaiveConvolutionTranspos e3d.cu - aten/
src/ , CUDA, 614 linesATen/ native/ cuda/ NaiveDilatedConvolution. cu - aten/
src/ , CUDA, 464 linesATen/ native/ cuda/ Nonzero.cu - aten/
src/ , CUDA, 842 linesATen/ native/ cuda/ Normalization.cu - aten/
src/ , CUDA, 400 linesATen/ native/ cuda/ PhiloxDistribution.cu - aten/
src/ , CUDA, 196 linesATen/ native/ cuda/ PhiloxKeySplit.cu - aten/
src/ , CUDA, 236 linesATen/ native/ cuda/ PointwiseOpsKernel.cu - aten/
src/ , CUDA, 215 linesATen/ native/ cuda/ PowKernel.cu - aten/
src/ , CUDA, 667 linesATen/ native/ cuda/ RNN.cu - aten/
src/ , CUDA, 133 linesATen/ native/ cuda/ Randperm.cu - aten/
src/ , CUDA, 297 linesATen/ native/ cuda/ RangeFactories.cu - aten/
src/ , CUDA, 16 linesATen/ native/ cuda/ RecordStream.cu - aten/
src/ , CUDA, 56 linesATen/ native/ cuda/ Reduce.cu - aten/
src/ , CUDA, 47 linesATen/ native/ cuda/ ReduceAMinMaxKernel.cu - aten/
src/ , CUDA, 44 linesATen/ native/ cuda/ ReduceArgMaxKernel.cu - aten/
src/ , CUDA, 44 linesATen/ native/ cuda/ ReduceArgMinKernel.cu - aten/
src/ , CUDA, 38 linesATen/ native/ cuda/ ReduceLogicKernel.cu - aten/
src/ , CUDA, 59 linesATen/ native/ cuda/ ReduceMaxValuesKernel.cu - aten/
src/ , CUDA, 55 linesATen/ native/ cuda/ ReduceMinValuesKernel.cu - aten/
src/ , CUDA, 74 linesATen/ native/ cuda/ ReduceMomentKernel.cu - aten/
src/ , CUDA, 77 linesATen/ native/ cuda/ ReduceNormKernel.cu - aten/
src/ , C++, 117 linesATen/ native/ cuda/ ReduceOps.cpp - aten/
src/ , C/C++, 21 linesATen/ native/ cuda/ ReduceOps.h - aten/
src/ , CUDA, 278 linesATen/ native/ cuda/ ReduceSumProdKernel.cu - aten/
src/ , CUDA, 976 linesATen/ native/ cuda/ ReflectionPad.cu - aten/
src/ , CUDA, 29 linesATen/ native/ cuda/ RenormKernel.cu - aten/
src/ , CUDA, 74 linesATen/ native/ cuda/ Repeat.cu - aten/
src/ , CUDA, 598 linesATen/ native/ cuda/ ReplicationPadding.cu - aten/
src/ , C++, 102 linesATen/ native/ cuda/ Resize.cpp - aten/
src/ , C/C++, 57 linesATen/ native/ cuda/ Resize.h - aten/
src/ , CUDA, 1,135 linesATen/ native/ cuda/ RowwiseScaledMM.cu - aten/
src/ , C/C++, 14 linesATen/ native/ cuda/ RowwiseScaledMM.h - aten/
src/ , CUDA, 190 linesATen/ native/ cuda/ RreluWithNoise.cu - aten/
src/ , C++, 1,854 linesATen/ native/ cuda/ ScaledBlas.cpp - aten/
src/ , C++, 37 linesATen/ native/ cuda/ ScaledBlasDeviceUtils.cp p - aten/
src/ , C/C++, 14 linesATen/ native/ cuda/ ScaledBlasDeviceUtils.h - aten/
src/ , CUDA, 528 linesATen/ native/ cuda/ ScaledGroupMM.cu - aten/
src/ , C/C++, 15 linesATen/ native/ cuda/ ScaledGroupMM.h - aten/
src/ , C++, 110 linesATen/ native/ cuda/ ScanKernels.cpp - aten/
src/ , C/C++, 18 linesATen/ native/ cuda/ ScanKernels.h - aten/
src/ , CUDA, 753 linesATen/ native/ cuda/ ScatterGatherKernel.cu - aten/
src/ , CUDA, 609 linesATen/ native/ cuda/ SegmentReduce.cu - aten/
src/ , CUDA, 698 linesATen/ native/ cuda/ Shape.cu - aten/
src/ , CUDA, 1,644 linesATen/ native/ cuda/ SoftMax.cu - aten/
src/ , C++, 129 linesATen/ native/ cuda/ Sort.cpp - aten/
src/ , CUDA, 398 linesATen/ native/ cuda/ Sort.cu - aten/
src/ , C/C++, 17 linesATen/ native/ cuda/ Sort.h - aten/
src/ , CUDA, 38 linesATen/ native/ cuda/ SortImpl.cu - aten/
src/ , CUDA, 291 linesATen/ native/ cuda/ SortStable.cu - aten/
src/ , C/C++, 17 linesATen/ native/ cuda/ SortStable.h - aten/
src/ , C++, 202 linesATen/ native/ cuda/ Sorting.cpp - aten/
src/ , CUDA, 300 linesATen/ native/ cuda/ Sorting.cu - aten/
src/ , C/C++, 17 linesATen/ native/ cuda/ Sorting.h - aten/
src/ , CUDA, 211 linesATen/ native/ cuda/ SparseBinaryOpIntersecti onKernel.cu - aten/
src/ , CUDA, 21 linesATen/ native/ cuda/ SparseMM.cu - aten/
src/ , C++, 578 linesATen/ native/ cuda/ SpectralOps.cpp - aten/
src/ , CUDA, 166 linesATen/ native/ cuda/ SpectralOps.cu - aten/
src/ , CUDA, 33 linesATen/ native/ cuda/ StepKernel.cu - aten/
src/ , CUDA, 433 linesATen/ native/ cuda/ SummaryOps.cu - aten/
src/ , C++, 23 linesATen/ native/ cuda/ TensorCompare.cpp - aten/
src/ , CUDA, 152 linesATen/ native/ cuda/ TensorCompare.cu - aten/
src/ , CUDA, 426 linesATen/ native/ cuda/ TensorFactories.cu - aten/
src/ , C++, 102 linesATen/ native/ cuda/ TensorModeKernel.cpp - aten/
src/ , CUDA, 289 linesATen/ native/ cuda/ TensorModeKernel.cu - aten/
src/ , C/C++, 18 linesATen/ native/ cuda/ TensorModeKernel.h - aten/
src/ , CUDA, 839 linesATen/ native/ cuda/ TensorShape.cu - aten/
src/ , C++, 41 linesATen/ native/ cuda/ TensorShapeCUDA.cpp - aten/
src/ , C++, 126 linesATen/ native/ cuda/ TensorTopK.cpp - aten/
src/ , CUDA, 1,628 linesATen/ native/ cuda/ TensorTopK.cu - aten/
src/ , C/C++, 13 linesATen/ native/ cuda/ TensorTopK.h - aten/
src/ , CUDA, 191 linesATen/ native/ cuda/ TensorTransformations.cu - aten/
src/ , CUDA, 250 linesATen/ native/ cuda/ TmaScatterKernel.cu - aten/
src/ , CUDA, 211 linesATen/ native/ cuda/ TriangularOps.cu - aten/
src/ , CUDA, 102 linesATen/ native/ cuda/ UnaryComplexKernels.cu - aten/
src/ , CUDA, 199 linesATen/ native/ cuda/ UnaryFractionKernels.cu - aten/
src/ , CUDA, 132 linesATen/ native/ cuda/ UnaryGammaKernels.cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ UnaryGeometricAcosKernel .cu - aten/
src/ , CUDA, 59 linesATen/ native/ cuda/ UnaryGeometricAcoshKerne l.cu - aten/
src/ , CUDA, 55 linesATen/ native/ cuda/ UnaryGeometricAsinKernel .cu - aten/
src/ , CUDA, 59 linesATen/ native/ cuda/ UnaryGeometricAsinhKerne l.cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ UnaryGeometricAtanKernel .cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ UnaryGeometricAtanhKerne l.cu - aten/
src/ , CUDA, 57 linesATen/ native/ cuda/ UnaryGeometricCosKernel. cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ UnaryGeometricCoshKernel .cu - aten/
src/ , CUDA, 57 linesATen/ native/ cuda/ UnaryGeometricSinKernel. cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ UnaryGeometricSinhKernel .cu - aten/
src/ , CUDA, 58 linesATen/ native/ cuda/ UnaryGeometricTanKernel. cu - aten/
src/ , CUDA, 59 linesATen/ native/ cuda/ UnaryGeometricTanhKernel .cu - aten/
src/ , CUDA, 121 linesATen/ native/ cuda/ UnaryLogKernels.cu - aten/
src/ , CUDA, 286 linesATen/ native/ cuda/ UnaryOpsKernel.cu - aten/
src/ , CUDA, 137 linesATen/ native/ cuda/ UnarySignKernels.cu - aten/
src/ , CUDA, 402 linesATen/ native/ cuda/ UnarySpecialOpsKernel.cu - aten/
src/ , CUDA, 162 linesATen/ native/ cuda/ UnfoldBackwardKernel.cu - aten/
src/ , CUDA, 235 linesATen/ native/ cuda/ Unique.cu - aten/
src/ , CUDA, 347 linesATen/ native/ cuda/ UniqueCub.cu - aten/
src/ , CUDA, 394 linesATen/ native/ cuda/ UpSampleBicubic2d.cu - aten/
src/ , CUDA, 1,050 linesATen/ native/ cuda/ UpSampleBilinear2d.cu - aten/
src/ , CUDA, 401 linesATen/ native/ cuda/ UpSampleLinear1d.cu - aten/
src/ , CUDA, 241 linesATen/ native/ cuda/ UpSampleNearest1d.cu - aten/
src/ , CUDA, 545 linesATen/ native/ cuda/ UpSampleNearest2d.cu - aten/
src/ , CUDA, 343 linesATen/ native/ cuda/ UpSampleNearest3d.cu - aten/
src/ , CUDA, 403 linesATen/ native/ cuda/ UpSampleTrilinear3d.cu - aten/
src/ , CUDA, 29 linesATen/ native/ cuda/ ValidateCompressedIndice sKernel.cu - aten/
src/ , CUDA, 525 linesATen/ native/ cuda/ WeightNorm.cu - aten/
src/ , CUDA, 39 linesATen/ native/ cuda/ ZetaKernel.cu - aten/
src/ , CUDA, 42 linesATen/ native/ cuda/ airy_ai.cu - aten/
src/ , CUDA, 42 linesATen/ native/ cuda/ bessel_j0.cu - aten/
src/ , CUDA, 42 linesATen/ native/ cuda/ bessel_j1.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ bessel_y0.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ bessel_y1.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ chebyshev_polynomial_t.c u - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ chebyshev_polynomial_u.c u - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ chebyshev_polynomial_v.c u - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ chebyshev_polynomial_w.c u - aten/
src/ , C/C++, 189 linesATen/ native/ cuda/ cuBlasCommonArgs.h - aten/
src/ , C/C++, 46 linesATen/ native/ cuda/ cutlass_extensions/ arch/ mma.h - aten/
src/ , C/C++, 76 linesATen/ native/ cuda/ cutlass_extensions/ epilogue/ thread/ ft_fused_activations.h - aten/
src/ , C/C++, 82 linesATen/ native/ cuda/ cutlass_extensions/ epilogue_helpers.h - aten/
src/ , C/C++, 58 linesATen/ native/ cuda/ cutlass_extensions/ ft_gemm_configs.h - aten/
src/ , C/C++, 123 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ kernel/ default_fpA_intB_traits. h - aten/
src/ , C/C++, 510 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ kernel/ fpA_intB_gemm.h - aten/
src/ , C/C++, 89 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ kernel/ mixed_gemm_B_layout.h - aten/
src/ , C/C++, 106 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ default_dq_mma.h - aten/
src/ , C/C++, 346 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ default_dq_mma_multistag e.h - aten/
src/ , C/C++, 315 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ default_dq_mma_pipelined .h - aten/
src/ , C/C++, 426 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ default_mma.h - aten/
src/ , C/C++, 527 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ default_mma_bf16.h - aten/
src/ , C/C++, 236 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ dq_mma_base.h - aten/
src/ , C/C++, 599 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ dq_mma_multistage.h - aten/
src/ , C/C++, 385 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ threadblock/ dq_mma_pipelined.h - aten/
src/ , C/C++, 127 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ warp/ default_mma_tensor_op.h - aten/
src/ , C/C++, 313 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ warp/ mma_tensorop_compute_B_w ith_f16.h - aten/
src/ , C/C++, 474 linesATen/ native/ cuda/ cutlass_extensions/ gemm/ warp/ mma_tensorop_dequantizer .h - aten/
src/ , C/C++, 429 linesATen/ native/ cuda/ cutlass_extensions/ interleaved_numeric_conv ersion.h - aten/
src/ , C/C++, 61 linesATen/ native/ cuda/ cutlass_extensions/ tile_interleaved_layout. h - aten/
src/ , CUDA, 94 linesATen/ native/ cuda/ fused_adagrad_impl.cu - aten/
src/ , CUDA, 74 linesATen/ native/ cuda/ fused_adam_amsgrad_impl. cu - aten/
src/ , CUDA, 72 linesATen/ native/ cuda/ fused_adam_impl.cu - aten/
src/ , CUDA, 74 linesATen/ native/ cuda/ fused_adamw_amsgrad_impl .cu - aten/
src/ , CUDA, 72 linesATen/ native/ cuda/ fused_adamw_impl.cu - aten/
src/ , CUDA, 1,417 linesATen/ native/ cuda/ group_norm_kernel.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ hermite_polynomial_h.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ hermite_polynomial_he.cu - aten/
src/ , CUDA, 1,374 linesATen/ native/ cuda/ int4mm.cu - aten/
src/ , CUDA, 95 linesATen/ native/ cuda/ int8mm.cu - aten/
src/ , C++, 1,742 linesATen/ native/ cuda/ jit_utils.cpp - aten/
src/ , C/C++, 254 linesATen/ native/ cuda/ jit_utils.h - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ laguerre_polynomial_l.cu - aten/
src/ , CUDA, 2,105 linesATen/ native/ cuda/ layer_norm_kernel.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ legendre_polynomial_p.cu - aten/
src/ , C++, 1,474 linesATen/ native/ cuda/ linalg/ BatchLinearAlgebra.cpp - aten/
src/ , C++, 1,763 linesATen/ native/ cuda/ linalg/ BatchLinearAlgebraLib.cp p - aten/
src/ , C/C++, 86 linesATen/ native/ cuda/ linalg/ BatchLinearAlgebraLib.h - aten/
src/ , C++, 298 linesATen/ native/ cuda/ linalg/ BatchLinearAlgebraLibBla s.cpp - aten/
src/ , C++, 2,397 linesATen/ native/ cuda/ linalg/ CUDASolver.cpp - aten/
src/ , C/C++, 793 linesATen/ native/ cuda/ linalg/ CUDASolver.h - aten/
src/ , C++, 51 linesATen/ native/ cuda/ linalg/ CudssHandlePool.cpp - aten/
src/ , C++, 53 linesATen/ native/ cuda/ linalg/ CusolverDnHandlePool.cpp - aten/
src/ , CUDA, 826 linesATen/ native/ cuda/ linalg/ LURecBlas3Kernel.cu - aten/
src/ , C/C++, 107 linesATen/ native/ cuda/ linalg/ MagmaUtils.h - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ modified_bessel_i0.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ modified_bessel_i1.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ modified_bessel_k0.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ modified_bessel_k1.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ scaled_modified_bessel_k 0.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ scaled_modified_bessel_k 1.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ shifted_chebyshev_polyno mial_t.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ shifted_chebyshev_polyno mial_u.cu - aten/
src/ , CUDA, 32 linesATen/ native/ cuda/ shifted_chebyshev_polyno mial_v.cu - aten/
src/ , CUDA, 31 linesATen/ native/ cuda/ shifted_chebyshev_polyno mial_w.cu - aten/
src/ , CUDA, 41 linesATen/ native/ cuda/ spherical_bessel_j0.cu - aten/
src/ , C/C++, 25 linesATen/ native/ cuda/ thread_constants.h - aten/
src/ , C/C++, 11 linesATen/ native/ cuda/ thrust_compat.h - aten/
src/ , C++, 123 linesATen/ native/ cudnn/ AffineGridGenerator.cpp - aten/
src/ , C++, 451 linesATen/ native/ cudnn/ BatchNorm.cpp - aten/
src/ , C/C++, 6 linesATen/ native/ cudnn/ BatchNorm.h - aten/
src/ , C++, 246 linesATen/ native/ cudnn/ ConvPlaceholders.cpp - aten/
src/ , C++, 831 linesATen/ native/ cudnn/ ConvShared.cpp - aten/
src/ , C/C++, 189 linesATen/ native/ cudnn/ ConvShared.h - aten/
src/ , C++, 1,300 linesATen/ native/ cudnn/ Conv_v7.cpp - aten/
src/ , C++, 1,648 linesATen/ native/ cudnn/ Conv_v8.cpp - aten/
src/ , C++, 193 linesATen/ native/ cudnn/ GridSampler.cpp - aten/
src/ , C++, 348 linesATen/ native/ cudnn/ LossCTC.cpp - aten/
src/ , C++, 2,277 linesATen/ native/ cudnn/ MHA.cpp - aten/
src/ , C/C++, 105 linesATen/ native/ cudnn/ MHA.h - aten/
src/ , C++, 2,837 linesATen/ native/ cudnn/ RNN.cpp - aten/
src/ , C/C++, 29 linesATen/ native/ cudnn/ RNNUtils.h - aten/
src/ , C++, 283 linesATen/ native/ group_norm.cpp - aten/
src/ , C/C++, 42 linesATen/ native/ group_norm.h - aten/
src/ , C/C++, 33 linesATen/ native/ hip/ bgemm_kernels/ bgemm_kernel_collection. h - aten/
src/ , C/C++, 159 linesATen/ native/ hip/ bgemm_kernels/ bgemm_kernel_template.h - aten/
src/ , C/C++, 16 linesATen/ native/ hip/ ck_bgemm.h - aten/
src/ , C/C++, 25 linesATen/ native/ hip/ ck_gemm.h - aten/
src/ , C/C++, 407 linesATen/ native/ hip/ ck_gemm_template.h - aten/
src/ , C/C++, 19 linesATen/ native/ hip/ ck_group_gemm.h - aten/
src/ , C/C++, 75 linesATen/ native/ hip/ ck_types.h - aten/
src/ , C/C++, 149 linesATen/ native/ im2col.h - aten/
src/ , C/C++, 240 linesATen/ native/ im2col_shape_check.h - aten/
src/ , C++, 483 linesATen/ native/ kleidiai/ kai_kernels.cpp - aten/
src/ , C/C++, 43 linesATen/ native/ kleidiai/ kai_kernels.h - aten/
src/ , C/C++, 109 linesATen/ native/ kleidiai/ kai_pack.h - aten/
src/ , C++, 106 linesATen/ native/ kleidiai/ kai_ukernel_interface.cp p - aten/
src/ , C/C++, 221 linesATen/ native/ kleidiai/ kai_ukernel_interface.h - aten/
src/ , C++, 375 linesATen/ native/ layer_norm.cpp - aten/
src/ , C/C++, 153 linesATen/ native/ layer_norm.h - aten/
src/ , C/C++, 25 linesATen/ native/ metal/ MetalCommandBuffer.h - aten/
src/ , C/C++, 23 linesATen/ native/ metal/ MetalContext.h - aten/
src/ , C/C++, 49 linesATen/ native/ metal/ MetalConvParams.h - aten/
src/ , C/C++, 52 linesATen/ native/ metal/ MetalDevice.h - aten/
src/ , C++, 62 linesATen/ native/ metal/ MetalGuardImpl.cpp - aten/
src/ , C/C++, 73 linesATen/ native/ metal/ MetalNeuronType.h - aten/
src/ , C/C++, 198 linesATen/ native/ metal/ MetalPrepackOpContext.h - aten/
src/ , C++, 130 linesATen/ native/ metal/ MetalPrepackOpRegister.c pp - aten/
src/ , C/C++, 868 linesATen/ native/ metal/ MetalShaders.h - aten/
src/ , C/C++, 51 linesATen/ native/ metal/ MetalTensorImpl.h - aten/
src/ , C/C++, 43 linesATen/ native/ metal/ MetalTensorImplStorage.h - aten/
src/ , C/C++, 71 linesATen/ native/ metal/ MetalTensorUtils.h - aten/
src/ , C/C++, 5 linesATen/ native/ metal/ mpscnn/ MPSCNNClampOp.h - aten/
src/ , C/C++, 25 linesATen/ native/ metal/ mpscnn/ MPSCNNConvOp.h - aten/
src/ , C/C++, 14 linesATen/ native/ metal/ mpscnn/ MPSCNNFullyConnectedOp.h - aten/
src/ , C/C++, 24 linesATen/ native/ metal/ mpscnn/ MPSCNNNeuronOp.h - aten/
src/ , C/C++, 24 linesATen/ native/ metal/ mpscnn/ MPSCNNOp.h - aten/
src/ , C/C++, 82 linesATen/ native/ metal/ mpscnn/ MPSCNNUtils.h - aten/
src/ , C/C++, 17 linesATen/ native/ metal/ mpscnn/ MPSImage+Tensor.h - aten/
src/ , C/C++, 78 linesATen/ native/ metal/ mpscnn/ MPSImageUtils.h - aten/
src/ , C/C++, 45 linesATen/ native/ metal/ mpscnn/ MPSImageWrapper.h - aten/
src/ , C/C++, 63 linesATen/ native/ metal/ mpscnn/ tests/ MPSCNNTests.h - aten/
src/ , C/C++, 20 linesATen/ native/ metal/ mpscnn/ tests/ MetalOpTestRunner.h - aten/
src/ , C/C++, 20 linesATen/ native/ metal/ ops/ MetalConvolution.h - aten/
src/ , C/C++, 12 linesATen/ native/ metal/ ops/ MetalCopy.h - aten/
src/ , C++, 242 linesATen/ native/ miopen/ BatchNorm_miopen.cpp - aten/
src/ , C++, 1,936 linesATen/ native/ miopen/ Conv_miopen.cpp - aten/
src/ , C++, 274 linesATen/ native/ miopen/ LossCTC_miopen.cpp - aten/
src/ , C++, 1,171 linesATen/ native/ miopen/ RNN_miopen.cpp - aten/
src/ , C++, 157 linesATen/ native/ mkl/ LinearAlgebra.cpp - aten/
src/ , C/C++, 50 linesATen/ native/ mkl/ LinearAlgebra.h - aten/
src/ , C++, 29 linesATen/ native/ mkl/ MklAllocationHelper.cpp - aten/
src/ , C++, 726 linesATen/ native/ mkl/ SparseBlasImpl.cpp - aten/
src/ , C/C++, 35 linesATen/ native/ mkl/ SparseBlasImpl.h - aten/
src/ , C++, 260 linesATen/ native/ mkl/ SparseCsrLinearAlgebra.c pp - aten/
src/ , C/C++, 13 linesATen/ native/ mkl/ SparseCsrLinearAlgebra.h - aten/
src/ , C++, 625 linesATen/ native/ mkl/ SpectralOps.cpp - aten/
src/ , C++, 159 linesATen/ native/ mkldnn/ BinaryOps.cpp - aten/
src/ , C/C++, 43 linesATen/ native/ mkldnn/ Common.h - aten/
src/ , C++, 1,238 linesATen/ native/ mkldnn/ Conv.cpp - aten/
src/ , C/C++, 68 linesATen/ native/ mkldnn/ Conv.h - aten/
src/ , C++, 281 linesATen/ native/ mkldnn/ ConvPrepack.cpp - aten/
src/ , C/C++, 41 linesATen/ native/ mkldnn/ ConvPrepack.h - aten/
src/ , C++, 47 linesATen/ native/ mkldnn/ Copy.cpp - aten/
src/ , C++, 64 linesATen/ native/ mkldnn/ Gelu.cpp - aten/
src/ , C++, 32 linesATen/ native/ mkldnn/ IDeepRegistration.cpp - aten/
src/ , C/C++, 7 linesATen/ native/ mkldnn/ IDeepRegistration.h - aten/
src/ , C++, 629 linesATen/ native/ mkldnn/ Linear.cpp - aten/
src/ , C/C++, 49 linesATen/ native/ mkldnn/ Linear.h - aten/
src/ , C++, 235 linesATen/ native/ mkldnn/ MKLDNNCommon.cpp - aten/
src/ , C/C++, 68 linesATen/ native/ mkldnn/ MKLDNNCommon.h - aten/
src/ , C++, 650 linesATen/ native/ mkldnn/ MKLDNNConversions.cpp - aten/
src/ , C++, 649 linesATen/ native/ mkldnn/ Matmul.cpp - aten/
src/ , C/C++, 90 linesATen/ native/ mkldnn/ Matmul.h - aten/
src/ , C++, 54 linesATen/ native/ mkldnn/ MkldnnTensorMath.cpp - aten/
src/ , C++, 281 linesATen/ native/ mkldnn/ Normalization.cpp - aten/
src/ , C++, 43 linesATen/ native/ mkldnn/ OpContext.cpp - aten/
src/ , C/C++, 95 linesATen/ native/ mkldnn/ OpContext.h - aten/
src/ , C++, 666 linesATen/ native/ mkldnn/ Pooling.cpp - aten/
src/ , C++, 72 linesATen/ native/ mkldnn/ Prelu.cpp - aten/
src/ , C++, 586 linesATen/ native/ mkldnn/ RNN.cpp - aten/
src/ , C++, 106 linesATen/ native/ mkldnn/ RegisterMkldnnOpContextC lass.cpp - aten/
src/ , C++, 76 linesATen/ native/ mkldnn/ Relu.cpp - aten/
src/ , C++, 49 linesATen/ native/ mkldnn/ SoftMax.cpp - aten/
src/ , C++, 35 linesATen/ native/ mkldnn/ TensorFactories.cpp - aten/
src/ , C++, 99 linesATen/ native/ mkldnn/ TensorShape.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ mkldnn/ TensorShape.h - aten/
src/ , C++, 76 linesATen/ native/ mkldnn/ UnaryOps.cpp - aten/
src/ , C++, 178 linesATen/ native/ mkldnn/ Utils.cpp - aten/
src/ , C/C++, 160 linesATen/ native/ mkldnn/ Utils.h - aten/
src/ , C++, 388 linesATen/ native/ mkldnn/ xpu/ Attention.cpp - aten/
src/ , C++, 840 linesATen/ native/ mkldnn/ xpu/ Blas.cpp - aten/
src/ , C++, 754 linesATen/ native/ mkldnn/ xpu/ Conv.cpp - aten/
src/ , C/C++, 54 linesATen/ native/ mkldnn/ xpu/ Conv.h - aten/
src/ , C++, 147 linesATen/ native/ mkldnn/ xpu/ FusionUtils.cpp - aten/
src/ , C/C++, 43 linesATen/ native/ mkldnn/ xpu/ FusionUtils.h - aten/
src/ , C++, 113 linesATen/ native/ mkldnn/ xpu/ Linear.cpp - aten/
src/ , C++, 252 linesATen/ native/ mkldnn/ xpu/ RNN.cpp - aten/
src/ , C++, 1,424 linesATen/ native/ mkldnn/ xpu/ ScaledBlas.cpp - aten/
src/ , C++, 1,269 linesATen/ native/ mkldnn/ xpu/ detail/ Attention.cpp - aten/
src/ , C/C++, 464 linesATen/ native/ mkldnn/ xpu/ detail/ Attr.h - aten/
src/ , C++, 382 linesATen/ native/ mkldnn/ xpu/ detail/ Conv.cpp - aten/
src/ , C++, 448 linesATen/ native/ mkldnn/ xpu/ detail/ Deconv.cpp - aten/
src/ , C/C++, 456 linesATen/ native/ mkldnn/ xpu/ detail/ DnnlExt.h - aten/
src/ , C/C++, 110 linesATen/ native/ mkldnn/ xpu/ detail/ LRUCache.h - aten/
src/ , C++, 266 linesATen/ native/ mkldnn/ xpu/ detail/ Matmul.cpp - aten/
src/ , C++, 246 linesATen/ native/ mkldnn/ xpu/ detail/ QConv.cpp - aten/
src/ , C++, 713 linesATen/ native/ mkldnn/ xpu/ detail/ QMatmul.cpp - aten/
src/ , C++, 591 linesATen/ native/ mkldnn/ xpu/ detail/ Utils.cpp - aten/
src/ , C/C++, 175 linesATen/ native/ mkldnn/ xpu/ detail/ Utils.h - aten/
src/ , C++, 322 linesATen/ native/ mkldnn/ xpu/ detail/ WoQMatmul.cpp - aten/
src/ , C/C++, 226 linesATen/ native/ mkldnn/ xpu/ detail/ oneDNN.h - aten/
src/ , C++, 59 linesATen/ native/ mkldnn/ xpu/ detail/ oneDNNContext.cpp - aten/
src/ , C/C++, 95 linesATen/ native/ mkldnn/ xpu/ detail/ oneDNNContext.h - aten/
src/ , C++, 320 linesATen/ native/ mkldnn/ xpu/ qconv.cpp - aten/
src/ , C/C++, 111 linesATen/ native/ mkldnn/ xpu/ qconv.h - aten/
src/ , C++, 277 linesATen/ native/ mkldnn/ xpu/ qlinear.cpp - aten/
src/ , C/C++, 91 linesATen/ native/ mkldnn/ xpu/ qlinear.h - aten/
src/ , C/C++, 29 linesATen/ native/ mps/ Copy.h - aten/
src/ , C/C++, 52 linesATen/ native/ mps/ MPSGraphSequoiaOps.h - aten/
src/ , C/C++, 264 linesATen/ native/ mps/ MetalShaderLibrary.h - aten/
src/ , C/C++, 890 linesATen/ native/ mps/ OperationUtils.h - aten/
src/ , C++, 163 linesATen/ native/ mps/ TensorFactory.cpp - aten/
src/ , C/C++, 14 linesATen/ native/ mps/ TensorFactory.h - aten/
src/ , C/C++, 23 linesATen/ native/ mps/ kernels/ Activation.h - aten/
src/ , C/C++, 50 linesATen/ native/ mps/ kernels/ Convolution.h - aten/
src/ , C/C++, 17 linesATen/ native/ mps/ kernels/ Copy.h - aten/
src/ , C/C++, 488 linesATen/ native/ mps/ kernels/ DecodeAttention.h - aten/
src/ , C/C++, 9 linesATen/ native/ mps/ kernels/ Distance.h - aten/
src/ , C/C++, 14 linesATen/ native/ mps/ kernels/ Embedding.h - aten/
src/ , C/C++, 49 linesATen/ native/ mps/ kernels/ EmbeddingBag.h - aten/
src/ , C/C++, 486 linesATen/ native/ mps/ kernels/ Gemv.h - aten/
src/ , C/C++, 98 linesATen/ native/ mps/ kernels/ GridSampler.h - aten/
src/ , C/C++, 18 linesATen/ native/ mps/ kernels/ GroupNorm.h - aten/
src/ , C/C++, 32 linesATen/ native/ mps/ kernels/ GroupedMM.h - aten/
src/ , C/C++, 21 linesATen/ native/ mps/ kernels/ Indexing.h - aten/
src/ , C/C++, 117 linesATen/ native/ mps/ kernels/ LinearAlgebra.h - aten/
src/ , C/C++, 55 linesATen/ native/ mps/ kernels/ LossOps.h - aten/
src/ , C/C++, 1,061 linesATen/ native/ mps/ kernels/ MppAttention.h - aten/
src/ , C/C++, 28 linesATen/ native/ mps/ kernels/ Pad.h - aten/
src/ , C/C++, 62 linesATen/ native/ mps/ kernels/ Pooling.h - aten/
src/ , C/C++, 1,031 linesATen/ native/ mps/ kernels/ PrefillAttention.h - aten/
src/ , C/C++, 71 linesATen/ native/ mps/ kernels/ ReduceOps.h - aten/
src/ , C/C++, 42 linesATen/ native/ mps/ kernels/ SamplingHelpers.h - aten/
src/ , C/C++, 33 linesATen/ native/ mps/ kernels/ ScaledMM.h - aten/
src/ , C/C++, 18 linesATen/ native/ mps/ kernels/ Shape.h - aten/
src/ , C/C++, 18 linesATen/ native/ mps/ kernels/ SoftMax.h - aten/
src/ , C/C++, 568 linesATen/ native/ mps/ kernels/ SortMerge.h - aten/
src/ , C/C++, 689 linesATen/ native/ mps/ kernels/ SortRadix.h - aten/
src/ , C/C++, 492 linesATen/ native/ mps/ kernels/ SortTopK.h - aten/
src/ , C/C++, 7 linesATen/ native/ mps/ kernels/ TensorCompare.h - aten/
src/ , C/C++, 12 linesATen/ native/ mps/ kernels/ UnaryKernel.h - aten/
src/ , C/C++, 46 linesATen/ native/ mps/ kernels/ UpSample.h - aten/
src/ , C++, 114 linesATen/ native/ mps/ operations/ ConvolutionHeuristics.cp p - aten/
src/ , C/C++, 49 linesATen/ native/ mps/ operations/ ConvolutionHeuristics.h - aten/
src/ , C++, 29 linesATen/ native/ mps/ operations/ Equal.cpp - aten/
src/ , C/C++, 38 linesATen/ native/ mps/ operations/ FusedAdamAmsgradKernelIm pl.h - aten/
src/ , C/C++, 35 linesATen/ native/ mps/ operations/ FusedAdamKernelImpl.h - aten/
src/ , C/C++, 37 linesATen/ native/ mps/ operations/ FusedAdamWAmsgradKernelI mpl.h - aten/
src/ , C/C++, 36 linesATen/ native/ mps/ operations/ FusedAdamWKernelImpl.h - aten/
src/ , C++, 110 linesATen/ native/ mps/ operations/ GemmHeuristics.cpp - aten/
src/ , C/C++, 67 linesATen/ native/ mps/ operations/ GemmHeuristics.h - aten/
src/ , C/C++, 398 linesATen/ native/ mps/ operations/ MultiTensorApply.h - aten/
src/ , C/C++, 11 linesATen/ native/ mps/ operations/ ScanKernel.h - aten/
src/ , C++, 14 linesATen/ native/ nested/ NestedTensorAliases.cpp - aten/
src/ , C++, 286 linesATen/ native/ nested/ NestedTensorBackward.cpp - aten/
src/ , C++, 332 linesATen/ native/ nested/ NestedTensorBinaryOps.cp p - aten/
src/ , C/C++, 18 linesATen/ native/ nested/ NestedTensorBinaryOps.h - aten/
src/ , C++, 246 linesATen/ native/ nested/ NestedTensorFactories.cp p - aten/
src/ , C++, 1,100 linesATen/ native/ nested/ NestedTensorMath.cpp - aten/
src/ , C/C++, 79 linesATen/ native/ nested/ NestedTensorMath.h - aten/
src/ , C++, 325 linesATen/ native/ nested/ NestedTensorMatmul.cpp - aten/
src/ , C++, 359 linesATen/ native/ nested/ NestedTensorTransformerF unctions.cpp - aten/
src/ , C/C++, 103 linesATen/ native/ nested/ NestedTensorTransformerF unctions.h - aten/
src/ , C/C++, 40 linesATen/ native/ nested/ NestedTensorTransformerU tils.h - aten/
src/ , C++, 194 linesATen/ native/ nested/ NestedTensorUnaryOps.cpp - aten/
src/ , C++, 181 linesATen/ native/ nested/ NestedTensorUtils.cpp - aten/
src/ , C/C++, 461 linesATen/ native/ nested/ NestedTensorUtils.h - aten/
src/ , CUDA, 120 linesATen/ native/ nested/ cuda/ NestedTensorBinaryOps.cu - aten/
src/ , CUDA, 475 linesATen/ native/ nested/ cuda/ NestedTensorMatmul.cu - aten/
src/ , C++, 467 linesATen/ native/ nested/ cuda/ NestedTensorTransformerF unctions.cpp - aten/
src/ , CUDA, 1,539 linesATen/ native/ nested/ cuda/ NestedTensorTransformerF unctions.cu - aten/
src/ , C++, 540 linesATen/ native/ nested/ cuda/ NestedTensorTransformerU tils.cpp - aten/
src/ , C++, 46 linesATen/ native/ prim_native_functions.cp p - aten/
src/ , C++, 296 linesATen/ native/ quantized/ AffineQuantizer.cpp - aten/
src/ , C/C++, 128 linesATen/ native/ quantized/ AffineQuantizer.h - aten/
src/ , C++, 288 linesATen/ native/ quantized/ AffineQuantizerBase.cpp - aten/
src/ , C/C++, 45 linesATen/ native/ quantized/ AffineQuantizerBase.h - aten/
src/ , C/C++, 62 linesATen/ native/ quantized/ ConvUtils.h - aten/
src/ , C++, 39 linesATen/ native/ quantized/ Copy.cpp - aten/
src/ , C/C++, 8 linesATen/ native/ quantized/ Copy.h - aten/
src/ , C/C++, 67 linesATen/ native/ quantized/ FakeQuantAffine.h - aten/
src/ , C++, 265 linesATen/ native/ quantized/ FakeQuantPerChannelAffin e.cpp - aten/
src/ , C++, 237 linesATen/ native/ quantized/ FakeQuantPerTensorAffine .cpp - aten/
src/ , C/C++, 13 linesATen/ native/ quantized/ IndexKernel.h - aten/
src/ , C/C++, 134 linesATen/ native/ quantized/ PackedParams.h - aten/
src/ , C++, 401 linesATen/ native/ quantized/ QTensor.cpp - aten/
src/ , C++, 195 linesATen/ native/ quantized/ TensorAdvancedIndexing.c pp - aten/
src/ , C++, 61 linesATen/ native/ quantized/ TensorCompare.cpp - aten/
src/ , C++, 175 linesATen/ native/ quantized/ TensorFactories.cpp - aten/
src/ , C++, 359 linesATen/ native/ quantized/ cpu/ ACLUtils.cpp - aten/
src/ , C/C++, 257 linesATen/ native/ quantized/ cpu/ ACLUtils.h - aten/
src/ , C++, 343 linesATen/ native/ quantized/ cpu/ AdaptiveAveragePooling.c pp - aten/
src/ , C++, 386 linesATen/ native/ quantized/ cpu/ AveragePool2d.cpp - aten/
src/ , C++, 182 linesATen/ native/ quantized/ cpu/ AveragePool3d.cpp - aten/
src/ , C++, 621 linesATen/ native/ quantized/ cpu/ BinaryOps.cpp - aten/
src/ , C/C++, 6 linesATen/ native/ quantized/ cpu/ BinaryOps.h - aten/
src/ , C++, 123 linesATen/ native/ quantized/ cpu/ ChannelShuffle.cpp - aten/
src/ , C/C++, 29 linesATen/ native/ quantized/ cpu/ EmbeddingPackedParams.h - aten/
src/ , C++, 54 linesATen/ native/ quantized/ cpu/ IntReprQuant.cpp - aten/
src/ , C++, 141 linesATen/ native/ quantized/ cpu/ LinearUnpackImpl.cpp - aten/
src/ , C++, 40 linesATen/ native/ quantized/ cpu/ MakePerTensorQuantizedTe nsor.cpp - aten/
src/ , C++, 496 linesATen/ native/ quantized/ cpu/ Normalization.cpp - aten/
src/ , C/C++, 539 linesATen/ native/ quantized/ cpu/ OnednnUtils.h - aten/
src/ , C++, 759 linesATen/ native/ quantized/ cpu/ Pooling.cpp - aten/
src/ , C/C++, 509 linesATen/ native/ quantized/ cpu/ QnnpackUtils.h - aten/
src/ , C/C++, 240 linesATen/ native/ quantized/ cpu/ QuantUtils.h - aten/
src/ , C/C++, 282 linesATen/ native/ quantized/ cpu/ QuantizedOps.h - aten/
src/ , C++, 229 linesATen/ native/ quantized/ cpu/ ReduceOps.cpp - aten/
src/ , C++, 33 linesATen/ native/ quantized/ cpu/ RuyUtils.cpp - aten/
src/ , C/C++, 17 linesATen/ native/ quantized/ cpu/ RuyUtils.h - aten/
src/ , C++, 66 linesATen/ native/ quantized/ cpu/ Sorting.cpp - aten/
src/ , C++, 101 linesATen/ native/ quantized/ cpu/ TensorOperators.cpp - aten/
src/ , C++, 189 linesATen/ native/ quantized/ cpu/ TensorShape.cpp - aten/
src/ , C++, 214 linesATen/ native/ quantized/ cpu/ UpSampleBilinear2d.cpp - aten/
src/ , C++, 219 linesATen/ native/ quantized/ cpu/ UpSampleNearest2d.cpp - aten/
src/ , C++, 236 linesATen/ native/ quantized/ cpu/ UpSampleNearest3d.cpp - aten/
src/ , C++, 85 linesATen/ native/ quantized/ cpu/ XnnpackUtils.cpp - aten/
src/ , C/C++, 345 linesATen/ native/ quantized/ cpu/ XnnpackUtils.h - aten/
src/ , C/C++, 418 linesATen/ native/ quantized/ cpu/ conv_serialization.h - aten/
src/ , C++, 554 linesATen/ native/ quantized/ cpu/ fbgemm_utils.cpp - aten/
src/ , C/C++, 410 linesATen/ native/ quantized/ cpu/ fbgemm_utils.h - aten/
src/ , C++, 254 linesATen/ native/ quantized/ cpu/ fused_obs_fake_quant.cpp - aten/
src/ , C++, 18 linesATen/ native/ quantized/ cpu/ init_qnnpack.cpp - aten/
src/ , C/C++, 11 linesATen/ native/ quantized/ cpu/ init_qnnpack.h - aten/
src/ , C++, 4,725 linesATen/ native/ quantized/ cpu/ kernels/ QuantizedOpKernels.cpp - aten/
src/ , C++, 172 linesATen/ native/ quantized/ cpu/ qclamp.cpp - aten/
src/ , C++, 2,268 linesATen/ native/ quantized/ cpu/ qconv.cpp - aten/
src/ , C/C++, 100 linesATen/ native/ quantized/ cpu/ qconv.h - aten/
src/ , C++, 239 linesATen/ native/ quantized/ cpu/ qconv_dynamic.cpp - aten/
src/ , C++, 876 linesATen/ native/ quantized/ cpu/ qconv_prepack.cpp - aten/
src/ , C++, 137 linesATen/ native/ quantized/ cpu/ qconv_unpack_impl.cpp - aten/
src/ , C++, 20 linesATen/ native/ quantized/ cpu/ qdropout.cpp - aten/
src/ , C++, 36 linesATen/ native/ quantized/ cpu/ qelu.cpp - aten/
src/ , C++, 1,560 linesATen/ native/ quantized/ cpu/ qembeddingbag.cpp - aten/
src/ , C/C++, 32 linesATen/ native/ quantized/ cpu/ qembeddingbag.h - aten/
src/ , C++, 670 linesATen/ native/ quantized/ cpu/ qembeddingbag_prepack.cp p - aten/
src/ , C/C++, 15 linesATen/ native/ quantized/ cpu/ qembeddingbag_prepack.h - aten/
src/ , C++, 291 linesATen/ native/ quantized/ cpu/ qembeddingbag_unpack.cpp - aten/
src/ , C++, 28 linesATen/ native/ quantized/ cpu/ qgelu.cpp - aten/
src/ , C++, 111 linesATen/ native/ quantized/ cpu/ qhardsigmoid.cpp - aten/
src/ , C++, 104 linesATen/ native/ quantized/ cpu/ qhardswish.cpp - aten/
src/ , C++, 1,734 linesATen/ native/ quantized/ cpu/ qlinear.cpp - aten/
src/ , C/C++, 51 linesATen/ native/ quantized/ cpu/ qlinear.h - aten/
src/ , C++, 1,101 linesATen/ native/ quantized/ cpu/ qlinear_dynamic.cpp - aten/
src/ , C++, 764 linesATen/ native/ quantized/ cpu/ qlinear_prepack.cpp - aten/
src/ , C++, 186 linesATen/ native/ quantized/ cpu/ qmatmul.cpp - aten/
src/ , C++, 455 linesATen/ native/ quantized/ cpu/ qmul.cpp - aten/
src/ , C++, 172 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ add.cc - aten/
src/ , C++, 192 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ average-pooling.cc - aten/
src/ , C++, 249 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ channel-shuffle.cc - aten/
src/ , C++, 1,019 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ convolution.cc - aten/
src/ , C++, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ global-average-pooling.c c - aten/
src/ , C++, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ hardsigmoid.cc - aten/
src/ , C++, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ hardswish.cc - aten/
src/ , C++, 378 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ hgemm.cc - aten/
src/ , C++, 166 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ max-pooling.cc - aten/
src/ , C++, 1,068 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ q8gemm.cc - aten/
src/ , C++, 657 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ q8gemm_sparse.cc - aten/
src/ , C++, 379 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ requantization.cc - aten/
src/ , C++, 624 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ sgemm.cc - aten/
src/ , C++, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ sigmoid.cc - aten/
src/ , C++, 101 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ softargmax.cc - aten/
src/ , C++, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ bench/ tanh.cc - aten/
src/ , Python, 281 linesATen/ native/ quantized/ cpu/ qnnpack/ configure.py - aten/
src/ , Python, 36 linesATen/ native/ quantized/ cpu/ qnnpack/ deps/ clog/ configure.py - aten/
src/ , C/C++, 123 linesATen/ native/ quantized/ cpu/ qnnpack/ deps/ clog/ include/ clog.h - aten/
src/ , C, 524 linesATen/ native/ quantized/ cpu/ qnnpack/ deps/ clog/ src/ clog.c - aten/
src/ , C++, 53 linesATen/ native/ quantized/ cpu/ qnnpack/ deps/ clog/ test/ clog.cc - aten/
src/ , Python, 136 linesATen/ native/ quantized/ cpu/ qnnpack/ generate-wrapper.py - aten/
src/ , C/C++, 383 linesATen/ native/ quantized/ cpu/ qnnpack/ include/ pack_block_sparse.h - aten/
src/ , C/C++, 456 linesATen/ native/ quantized/ cpu/ qnnpack/ include/ pytorch_qnnpack.h - aten/
src/ , C/C++, 166 linesATen/ native/ quantized/ cpu/ qnnpack/ include/ qnnpack_func.h - aten/
src/ , Shell, 68 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-android-arm64.sh - aten/
src/ , Shell, 68 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-android-armv7.sh - aten/
src/ , Shell, 68 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-android-x86.sh - aten/
src/ , Shell, 55 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-ios-arm64.sh - aten/
src/ , Shell, 55 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-ios-arm64e.sh - aten/
src/ , Shell, 55 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-ios-armv7.sh - aten/
src/ , Shell, 55 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-ios-armv7s.sh - aten/
src/ , Shell, 55 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-ios-i386.sh - aten/
src/ , Shell, 60 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-ios-x86_64.sh - aten/
src/ , Shell, 42 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ build-local.sh - aten/
src/ , Shell, 25 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ test-android-arm64.sh - aten/
src/ , Shell, 25 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ test-android-armv7.sh - aten/
src/ , Shell, 25 linesATen/ native/ quantized/ cpu/ qnnpack/ scripts/ test-android-x86.sh - aten/
src/ , C, 160 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ add.c - aten/
src/ , C, 290 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ average-pooling.c - aten/
src/ , C, 101 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ channel-shuffle.c - aten/
src/ , C, 102 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ clamp.c - aten/
src/ , C++, 284 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ conv-prepack.cc - aten/
src/ , C++, 682 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ conv-run.cc - aten/
src/ , C, 807 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ convolution.c - aten/
src/ , C++, 209 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ deconv-run.cc - aten/
src/ , C, 300 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ deconvolution.c - aten/
src/ , C++, 135 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ fc-dynamic-run.cc - aten/
src/ , C++, 58 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ fc-prepack.cc - aten/
src/ , C++, 122 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ fc-run.cc - aten/
src/ , C++, 70 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ fc-unpack.cc - aten/
src/ , C, 171 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ fully-connected-sparse.c - aten/
src/ , C, 160 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ fully-connected.c - aten/
src/ , C, 158 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ global-average-pooling.c - aten/
src/ , C, 164 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ hardsigmoid.c - aten/
src/ , C, 147 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ hardswish.c - aten/
src/ , C, 370 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ hgemm/ 8x8-neonfp16arith.c - aten/
src/ , C, 492 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ indirection.c - aten/
src/ , C, 392 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ init.c - aten/
src/ , C, 169 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ leaky-relu.c - aten/
src/ , C, 228 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ max-pooling.c - aten/
src/ , C, 27 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ operator-delete.c - aten/
src/ , C, 1,674 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ operator-run.c - aten/
src/ , C, 450 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8avgpool/ mp8x9p8q-neon.c - aten/
src/ , C, 471 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8avgpool/ mp8x9p8q-sse2.c - aten/
src/ , C, 242 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8avgpool/ up8x9-neon.c - aten/
src/ , C, 236 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8avgpool/ up8x9-sse2.c - aten/
src/ , C, 125 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8avgpool/ up8xm-neon.c - aten/
src/ , C, 102 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8avgpool/ up8xm-sse2.c - aten/
src/ , C, 360 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8conv/ 4x4c2-sse2.c - aten/
src/ , C, 756 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8conv/ 4x8-neon.c - aten/
src/ , C, 1,231 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8conv/ 8x8-neon.c - aten/
src/ , C, 950 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ mp8x25-neon-per-channel. c - aten/
src/ , C, 924 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ mp8x25-neon.c - aten/
src/ , C, 1,095 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ mp8x25-sse2-per-channel. c - aten/
src/ , C, 978 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ mp8x25-sse2.c - aten/
src/ , C, 1,044 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ mp8x27-neon.c - aten/
src/ , C, 159 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ mp8x27-sse2.c - aten/
src/ , C, 994 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ up8x9-neon-per-channel.c - aten/
src/ , C, 953 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ up8x9-neon.c - aten/
src/ , C, 440 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ up8x9-sse2-per-channel.c - aten/
src/ , C, 396 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8dwconv/ up8x9-sse2.c - aten/
src/ , C, 311 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gavgpool/ mp8x7p7q-neon.c - aten/
src/ , C, 323 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gavgpool/ mp8x7p7q-sse2.c - aten/
src/ , C, 210 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gavgpool/ up8x7-neon.c - aten/
src/ , C, 204 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gavgpool/ up8x7-sse2.c - aten/
src/ , C, 122 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gavgpool/ up8xm-neon.c - aten/
src/ , C, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gavgpool/ up8xm-sse2.c - aten/
src/ , C, 230 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 2x4c8-sse2.c - aten/
src/ , C, 154 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 4x-sumrows-neon.c - aten/
src/ , C, 328 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 4x4c2-dq-sse2.c - aten/
src/ , C, 354 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 4x4c2-sse2.c - aten/
src/ , C, 593 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 4x8-dq-neon.c - aten/
src/ , C, 743 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 4x8-neon.c - aten/
src/ , C, 543 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 4x8c2-xzp-neon.c - aten/
src/ , C, 577 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 6x4-neon.c - aten/
src/ , C, 1,231 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm/ 8x8-neon.c - aten/
src/ , C, 74 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm_sparse/ 8x4-packA-sse2.c - aten/
src/ , C, 17 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm_sparse/ 8x4c1x4-dq-packedA-sse2. c - aten/
src/ , C/C++, 435 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm_sparse/ 8x4c1x4-dq-packedA-sse2. h - aten/
src/ , C/C++, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8gemm_sparse/ 8x4c1x4-packed-sse2.h - aten/
src/ , C, 382 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8vadd/ neon.c - aten/
src/ , C, 224 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ q8vadd/ sse2.c - aten/
src/ , C/C++, 106 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ AlignedAllocator.h - aten/
src/ , C/C++, 33 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ assembly.h - aten/
src/ , C/C++, 94 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ common.h - aten/
src/ , C/C++, 37 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ hgemm.h - aten/
src/ , C/C++, 44 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ indirection.h - aten/
src/ , C/C++, 32 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ isa-checks.h - aten/
src/ , C/C++, 35 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ log.h - aten/
src/ , C/C++, 35 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ math.h - aten/
src/ , C/C++, 157 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ operator.h - aten/
src/ , C/C++, 776 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ pack.h - aten/
src/ , C/C++, 686 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ params.h - aten/
src/ , C/C++, 58 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8avgpool.h - aten/
src/ , C/C++, 42 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8conv.h - aten/
src/ , C/C++, 82 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8dwconv.h - aten/
src/ , C/C++, 54 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8gavgpool.h - aten/
src/ , C/C++, 96 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8gemm.h - aten/
src/ , C/C++, 150 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8gemm_sparse.h - aten/
src/ , C/C++, 34 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ q8vadd.h - aten/
src/ , C/C++, 74 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ requantization-stubs.h - aten/
src/ , C/C++, 546 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ requantization.h - aten/
src/ , C/C++, 119 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ scalar-utils.h - aten/
src/ , C/C++, 37 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ sconv.h - aten/
src/ , C/C++, 48 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ sdwconv.h - aten/
src/ , C/C++, 38 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ sgemm.h - aten/
src/ , C/C++, 33 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ u8clamp.h - aten/
src/ , C/C++, 29 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ u8lut32norm.h - aten/
src/ , C/C++, 39 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ u8maxpool.h - aten/
src/ , C/C++, 29 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ u8rmax.h - aten/
src/ , C/C++, 29 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ x8lut.h - aten/
src/ , C/C++, 40 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ qnnpack/ x8zip.h - aten/
src/ , C, 171 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ fp32-neon.c - aten/
src/ , C, 107 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ fp32-psimd.c - aten/
src/ , C, 121 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ fp32-scalar.c - aten/
src/ , C, 109 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ fp32-sse2.c - aten/
src/ , C, 113 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-neon.c - aten/
src/ , C, 81 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-scalar.c - aten/
src/ , C/C++, 48 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-scalar.h - aten/
src/ , C/C++, 123 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-sse.h - aten/
src/ , C, 73 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-sse2.c - aten/
src/ , C, 73 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-sse4.c - aten/
src/ , C, 73 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ gemmlowp-ssse3.c - aten/
src/ , C, 200 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ precise-neon.c - aten/
src/ , C, 165 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ precise-psimd.c - aten/
src/ , C, 365 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ precise-scalar.c - aten/
src/ , C, 164 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ precise-sse2.c - aten/
src/ , C, 134 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ precise-sse4.c - aten/
src/ , C, 150 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ precise-ssse3.c - aten/
src/ , C, 144 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ q31-neon.c - aten/
src/ , C, 163 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ q31-scalar.c - aten/
src/ , C, 241 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ q31-sse2.c - aten/
src/ , C, 162 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ q31-sse4.c - aten/
src/ , C, 238 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ q31-ssse3.c - aten/
src/ , C/C++, 29 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ runtime-assembly.h - aten/
src/ , C/C++, 22 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ runtime-neon.h - aten/
src/ , C/C++, 22 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ requantization/ runtime-sse2.h - aten/
src/ , C, 203 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ sconv/ 6x8-psimd.c - aten/
src/ , C, 156 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ sdwconv/ up4x9-psimd.c - aten/
src/ , C, 268 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ sgemm/ 5x8-neon.c - aten/
src/ , C, 307 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ sgemm/ 6x8-neon.c - aten/
src/ , C, 215 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ sgemm/ 6x8-psimd.c - aten/
src/ , C, 159 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ sigmoid.c - aten/
src/ , C, 139 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ softargmax.c - aten/
src/ , C, 161 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ tanh.c - aten/
src/ , C, 86 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8clamp/ neon.c - aten/
src/ , C, 81 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8clamp/ sse2.c - aten/
src/ , C, 49 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8lut32norm/ scalar.c - aten/
src/ , C, 251 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8maxpool/ 16x9p8q-neon.c - aten/
src/ , C, 254 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8maxpool/ 16x9p8q-sse2.c - aten/
src/ , C, 91 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8maxpool/ sub16-neon.c - aten/
src/ , C, 86 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8maxpool/ sub16-sse2.c - aten/
src/ , C, 48 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8rmax/ neon.c - aten/
src/ , C, 47 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ u8rmax/ sse2.c - aten/
src/ , C, 47 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8lut/ scalar.c - aten/
src/ , C, 46 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ x2-neon.c - aten/
src/ , C, 52 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ x2-sse2.c - aten/
src/ , C, 51 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ x3-neon.c - aten/
src/ , C, 205 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ x3-sse2.c - aten/
src/ , C, 57 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ x4-neon.c - aten/
src/ , C, 82 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ x4-sse2.c - aten/
src/ , C, 177 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ xm-neon.c - aten/
src/ , C, 208 linesATen/ native/ quantized/ cpu/ qnnpack/ src/ x8zip/ xm-sse2.c - aten/
src/ , C/C++, 281 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ add-operator-tester.h - aten/
src/ , C++, 397 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ add.cc - aten/
src/ , C/C++, 831 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ average-pooling-operator -tester.h - aten/
src/ , C++, 1,491 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ average-pooling.cc - aten/
src/ , C/C++, 429 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ avgpool-microkernel-test er.h - aten/
src/ , C/C++, 157 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ channel-shuffle-operator -tester.h - aten/
src/ , C++, 268 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ channel-shuffle.cc - aten/
src/ , C/C++, 118 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ clamp-microkernel-tester .h - aten/
src/ , C/C++, 177 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ clamp-operator-tester.h - aten/
src/ , C++, 107 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ clamp.cc - aten/
src/ , C/C++, 822 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ convolution-operator-tes ter.h - aten/
src/ , C++, 2,238 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ convolution.cc - aten/
src/ , C/C++, 654 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ deconvolution-operator-t ester.h - aten/
src/ , C++, 517 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ deconvolution.cc - aten/
src/ , C/C++, 550 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ dwconv-microkernel-teste r.h - aten/
src/ , C/C++, 391 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ fully-connected-operator -tester.h - aten/
src/ , C/C++, 613 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ fully-connected-sparse-o perator-tester.h - aten/
src/ , C++, 144 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ fully-connected-sparse.c c - aten/
src/ , C++, 700 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ fully-connected.cc - aten/
src/ , C/C++, 301 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ gavgpool-microkernel-tes ter.h - aten/
src/ , C/C++, 513 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ gemm-block-sparse-microk ernel-tester.h - aten/
src/ , C/C++, 1,169 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ gemm-microkernel-tester. h - aten/
src/ , C/C++, 253 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ global-average-pooling-o perator-tester.h - aten/
src/ , C++, 651 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ global-average-pooling.c c - aten/
src/ , C/C++, 217 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ hardsigmoid-operator-tes ter.h - aten/
src/ , C++, 229 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ hardsigmoid.cc - aten/
src/ , C/C++, 229 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ hardswish-operator-teste r.h - aten/
src/ , C++, 257 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ hardswish.cc - aten/
src/ , C++, 183 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ hgemm.cc - aten/
src/ , C/C++, 240 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ leaky-relu-operator-test er.h - aten/
src/ , C++, 161 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ leaky-relu.cc - aten/
src/ , C/C++, 90 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ lut-microkernel-tester.h - aten/
src/ , C/C++, 99 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ lut-norm-microkernel-tes ter.h - aten/
src/ , C/C++, 757 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ max-pooling-operator-tes ter.h - aten/
src/ , C++, 1,209 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ max-pooling.cc - aten/
src/ , C/C++, 256 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ maxpool-microkernel-test er.h - aten/
src/ , C++, 1,965 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8avgpool.cc - aten/
src/ , C++, 1,084 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8conv.cc - aten/
src/ , C++, 2,561 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8dwconv.cc - aten/
src/ , C++, 1,155 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8gavgpool.cc - aten/
src/ , C++, 3,870 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8gemm.cc - aten/
src/ , C++, 1,568 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8gemm_sparse.cc - aten/
src/ , C++, 297 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ q8vadd.cc - aten/
src/ , C/C++, 470 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ requantization-tester.h - aten/
src/ , C++, 1,077 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ requantization.cc - aten/
src/ , C/C++, 68 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ rmax-microkernel-tester. h - aten/
src/ , C++, 103 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ sconv.cc - aten/
src/ , C++, 502 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ sgemm.cc - aten/
src/ , C/C++, 214 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ sigmoid-operator-tester. h - aten/
src/ , C++, 229 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ sigmoid.cc - aten/
src/ , C/C++, 198 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ softargmax-operator-test er.h - aten/
src/ , C++, 135 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ softargmax.cc - aten/
src/ , C/C++, 214 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ tanh-operator-tester.h - aten/
src/ , C++, 229 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ tanh.cc - aten/
src/ , C/C++, 35 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ test_utils.h - aten/
src/ , C++, 127 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ u8clamp.cc - aten/
src/ , C++, 48 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ u8lut32norm.cc - aten/
src/ , C++, 1,539 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ u8maxpool.cc - aten/
src/ , C++, 71 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ u8rmax.cc - aten/
src/ , C/C++, 224 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ vadd-microkernel-tester. h - aten/
src/ , C++, 45 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ x8lut.cc - aten/
src/ , C++, 350 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ x8zip.cc - aten/
src/ , C/C++, 108 linesATen/ native/ quantized/ cpu/ qnnpack/ test/ zip-microkernel-tester.h - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8avgpool/ mp8x9p8q-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8avgpool/ mp8x9p8q-sse2.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8avgpool/ up8x9-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8avgpool/ up8x9-sse2.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8avgpool/ up8xm-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8avgpool/ up8xm-sse2.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8conv/ 4x4c2-sse2.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8conv/ 4x8-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8conv/ 8x8-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ mp8x25-neon-per-channel. c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ mp8x25-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ mp8x25-sse2-per-channel. c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ mp8x25-sse2.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ mp8x27-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ mp8x27-sse2.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ up8x9-neon-per-channel.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ up8x9-neon.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ up8x9-sse2-per-channel.c - aten/
src/ , C, 5 linesATen/ native/ quantized/ cpu/ qnnpack/ wrappers/ q8dwconv/ up8x9-sse2.c - repository limit reached (2,000 files or 30 MB): the rest is at the source (8517 files)
- LICENSE, License, 84 lines
- README.md, Text, 612 lines
The paper's code and data availability statement is in the Data section.
Tracing map
Proposed by the machine: these links were found in the paper and verified at the source, without human review. The map will receive a Zenodo DOI once one of the paper's authors has validated it with their ORCID.
What the map holds:
- 1 repository of the authors' code, each at its verified commit, with its license and how the link was found in the paper;
- 1,998 scripts, each with its path and the digest of its content;
- no match between paragraphs and code yet;
- neither the text of the paper nor the code itself.
Its JSON (tracing-map.json) is deposited on Zenodo with its DOI once the map is validated.
Data
No dataset and no data link were found in the paper.
Data, code, and materials availability
All data and code needed to evaluate and reproduce the results in the paper are present in the paper and/
Reproduced under the paper's license (CC BY), from the paper cited above.
Versions
The history of this record: each version stored by the harvester or made by a correction of its authors or of the maintainers of its code, and what changed in its facts. The texts of the paper (its abstract, its availability statements) are not part of it; versions that changed only those are not listed.
Version 1, 30 September 2026: the first record
Recorded: type, language, journal, volume, issue, pages, dates, 8 authors, 2 funders, 11 references.
Cite
This paper
Wang, Y., Yilmaz, Z., Li, Y., Liu, E., Ahlberg, E., Ghahari, F., Taciroglu, E., & Ozcan, A. (2026). Structural vibration monitoring with diffractive optical processors. Science advances, 12(10), eaea1712. https://
BibTeX
@article{wang2026structu
author = {Wang, Yuntian and Yilmaz, Zafer and Li, Yuhang and Liu, Edward and Ahlberg, Eric and Ghahari, Farid and Taciroglu, Ertugrul and Ozcan, Aydogan},
title = {{Structural vibration monitoring with diffractive optical processors}},
journal = {Science advances},
year = {2026},
month = mar,
volume = {12},
number = {10},
pages = {eaea1712},
publisher = {American Association for the Advancement of Science},
issn = {2375-2548},
doi = {10.1126/
url = {https://
pmid = {41779852},
pmcid = {PMC12959400}
}
RIS
TY - JOUR
AU - Wang, Yuntian
AU - Yilmaz, Zafer
AU - Li, Yuhang
AU - Liu, Edward
AU - Ahlberg, Eric
AU - Ghahari, Farid
AU - Taciroglu, Ertugrul
AU - Ozcan, Aydogan
TI - Structural vibration monitoring with diffractive optical processors
T2 - Science advances
J2 - Sci Adv
PY - 2026
DA - 2026/
VL - 12
IS - 10
SP - eaea1712
SN - 2375-2548
PB - American Association for the Advancement of Science
DO - 10.1126/
UR - https://
LA - en
ER -
CSL-JSON
{
"id": "10.1126/
"type": "article-journal",
"title": "Structural vibration monitoring with diffractive optical processors",
"container-title": "Science advances",
"author": [
{
"family": "Wang",
"given": "Yuntian"
},
{
"family": "Yilmaz",
"given": "Zafer"
},
{
"family": "Li",
"given": "Yuhang"
},
{
"family": "Liu",
"given": "Edward"
},
{
"family": "Ahlberg",
"given": "Eric"
},
{
"family": "Ghahari",
"given": "Farid"
},
{
"family": "Taciroglu",
"given": "Ertugrul"
},
{
"family": "Ozcan",
"given": "Aydogan"
}
],
"container-title-short":
"volume": "12",
"issue": "10",
"page": "eaea1712",
"DOI": "10.1126/
"PMID": "41779852",
"PMCID": "PMC12959400",
"ISSN": "2375-2548",
"publisher": "American Association for the Advancement of Science",
"URL": "https://
"language": "en",
"issued": {
"date-parts": [
[
2026,
3,
4
]
]
}
}
The tracing map gets a citation of its own once an author has validated it and it has a DOI.
Similar papers
The papers with a page that share the most with this one: the tools found in their code, their categories, datasets, cited references and authors, the rarest counting most.
- [1] doi:10.1038/s42003-026-10957-8 [code]
- Brain defence by the extracellular matrix protein Cochlin.Journal: Communications biologyIn common: JAX, Numba, NetworkX, 4 other tools
- [2] doi:10.1038/s41467-026-75959-w [code]
- Charting higher-order models of brain function beyond pairwise interactions.Journal: Nature communicationsIn common: JAX, Numba, NetworkX, 3 other tools
- [3] doi:10.1126/sciadv.aed4944 [code]
- Focused ultrasound blood-brain barrier opening reveals a paradoxical remote metabolic response in the primate brain.Journal: Science advancesIn common: JAX, Numba, PyTorch, 3 other tools
- [4] doi:10.1371/journal.pone.0346575 [code]
- Statistically valid explainable black-box machine learning: applications in sex classification across species using brain imaging.Journal: PloS oneIn common: JAX, Numba, PyTorch, 3 other tools
- [5] doi:10.1371/journal.pcbi.1014585 [code]
- Decoding behavior with minimal and interpretable agent models.Journal: PLoS computational biologyIn common: Numba, NetworkX, PyTorch, 3 other tools
- [6] doi:10.7554/elife.110588 [code]
- Opening the black box toward a modular approach to spike sorting.Journal: eLifeIn common: Numba, NetworkX, PyTorch, 3 other tools
- [7] doi:10.1126/sciadv.aed3650 [code]
- Truthful visualizations for mass spectrometry imaging enable high-spatial-resolution interactive &
lt;i& gt;m/ z& lt;/ i& gt; mapping and exploration. Journal: Science advancesIn common: Numba, NetworkX, PyTorch, 3 other tools - [8] doi:10.1371/journal.pcbi.1014571 [code]
- SynAPSeg: A novel dataset and image analysis framework for deep learning-based synapse detection and quantification.Journal: PLoS computational biologyIn common: Numba, NetworkX, PyTorch, 3 other tools
- [9] doi:10.1016/j.isci.2026.116825 [code]
- Social hierarchy shapes behavioral and transcriptional responses to chronic stress and ketamine in male mice.Journal: iScienceIn common: Numba, NetworkX, PyTorch, 3 other tools
- [10] doi:10.1093/nar/gkag706 [code]
- scDifformer: diffusion-based post-training for virtual cell modeling across large-scale single-cell data.Journal: Nucleic acids researchIn common: Numba, NetworkX, PyTorch, 3 other tools
Contribute
The authors of this paper can claim it, correct its record and validate its tracing map, and the maintainers of its code (its owner, or a public member of its organization) correct what it says of their repository; anyone signed in can ask for its removal. Every request goes to OSCR's own machine, which answers it; your account page follows them.
Sign in with ORCID to claim this paper as one of its authors, correct its record or validate its tracing map: when the paper's metadata lists your ORCID iD, you are recognized at once. Maintainers of its code: sign in with GitHub, then claim the repository on your account page.
Claim this paper
Correct its record
Say what each link of this record is, remove the ones that are not the paper's, add the ones that are missing. The correction becomes a new version of the record, in its Versions section.
Validate its tracing map
You validate the map as this page shows it: 1 repository of the authors' code, each at its verified commit and with its license, 1998 scripts, and 0 matches between paragraphs and code (see the Code and Map sections). It then receives a DOI on Zenodo, with you (your ORCID iD) and OSCR as its creators; the code itself is not deposited.
The map's fingerprint: sha256:21d4c8ec87379d27…
Add the badge to its README
The badge links the code to this page. Copy one of these into the README of the paper's code: only you decide where it goes, and nothing is changed for you.
Markdown
[, paste the snippet at the top, then “Commit changes…” and, to review it first, “Create a new branch and start a pull request”. You open the pull request; OSCR asks for no permission.
Request its removal
To ask OSCR to remove this record, the copies of its authors' scripts or its tracing map, use the removal request page: signed in, you say who you are, what to remove and why, then review and confirm the request. Published rules decide every request (how).
Discussion, reproductions, activity
Discussion: questions and error reports about this paper and its code, from signed-in readers and its authors. It opens with sign-in.
Reproductions: reports from readers who ran the authors' code: what they reproduced, with which environment, commit and data. It opens with sign-in.
Activity: what happens around this paper: new versions of its record, its map's validation, discussions and reproductions. It opens with sign-in.
