Depot 1758571 for SVFI - Advanced-GPU Optimization Pack
You own thisLast known name: SVFI - Targeted optimization of RTX 30 series graphics cards Depot
Windows 64-bitDLC 1758570
Depot ID | 1758571 |
Build ID | 17332513 |
Last public update | 12 February 2025 – 07:37:11 UTC ( |
Size on disk | 11.57 GiB |
Compressed size | 3.29 GiB (71.61% saving) |
Displaying manifest 5138775820992547222
dated 12 February 2025 – 07:36:14 UTC (
Name | Type | Size |
---|---|---|
Initializing table… | ||
alt_cuda_corr.pyd | pyd | 421.50 KiB |
bdb.py | py | 32.57 KiB |
nvinfer_10.dll | dll | 198.99 MiB |
nvinfer_builder_resource_10.dll | dll | 1.32 GiB |
nvinfer_dispatch_10.dll | dll | 599.04 KiB |
nvinfer_lean_10.dll | dll | 25.27 MiB |
nvinfer_plugin_10.dll | dll | 28.25 MiB |
nvinfer_vc_plugin_10.dll | dll | 507.04 KiB |
nvjitlink_120_0.dll | dll | 74.23 MiB |
nvonnxparser_10.dll | dll | 2.85 MiB |
pdb.py | py | 64.35 KiB |
softsplat_cuda.pyd | pyd | 961.50 KiB |
tensorrt | Folder | |
tensorrt/__init__.py | py | 6.06 KiB |
tensorrt/plugin | Folder | |
tensorrt/plugin/__init__.py | py | 1.52 KiB |
tensorrt/plugin/_autotune.py | py | 12.20 KiB |
tensorrt/plugin/_export.py | py | 1.16 KiB |
tensorrt/plugin/_lib.py | py | 20.88 KiB |
tensorrt/plugin/_plugin_class.py | py | 11.93 KiB |
tensorrt/plugin/_tensor.py | py | 28.26 KiB |
tensorrt/plugin/_top_level.py | py | 4.60 KiB |
tensorrt/plugin/_utils.py | py | 2.68 KiB |
tensorrt/plugin/_validate.py | py | 14.14 KiB |
tensorrt/tensorrt.cp311-win_amd64.pyd | pyd | 2.55 MiB |
tensorrt/tensorrt.pyd | pyd | 2.55 MiB |
torch | Folder | |
torch/_C | Folder | |
torch/_C.cp311-win_amd64.pyd | pyd | 10.50 KiB |
torch/_C/_VariableFunctions.pyi | pyi | 1.12 MiB |
torch/_C/__init__.pyi | pyi | 411.29 KiB |
torch/_C/_aoti.pyi | pyi | 704 B |
torch/_C/_autograd.pyi | pyi | 4.58 KiB |
torch/_C/_cpu.pyi | pyi | 447 B |
torch/_C/_cudnn.pyi | pyi | 387 B |
torch/_C/_cusparselt.pyi | pyi | 33 B |
torch/_C/_distributed_autograd.pyi | pyi | 945 B |
torch/_C/_distributed_c10d.pyi | pyi | 19.97 KiB |
torch/_C/_distributed_rpc.pyi | pyi | 6.12 KiB |
torch/_C/_distributed_rpc_testing.pyi | pyi | 1.02 KiB |
torch/_C/_export.pyi | pyi | 268 B |
torch/_C/_functions.pyi | pyi | 634 B |
torch/_C/_functorch.pyi | pyi | 3.26 KiB |
torch/_C/_instruction_counter.pyi | pyi | 113 B |
torch/_C/_itt.pyi | pyi | 174 B |
torch/_C/_lazy.pyi | pyi | 998 B |
torch/_C/_lazy_ts_backend.pyi | pyi | 338 B |
torch/_C/_monitor.pyi | pyi | 1.04 KiB |
torch/_C/_nn.pyi | pyi | 4.31 KiB |
torch/_C/_nvtx.pyi | pyi | 257 B |
torch/_C/_onnx.pyi | pyi | 749 B |
torch/_C/_profiler.pyi | pyi | 6.33 KiB |
torch/_C/_verbose.pyi | pyi | 137 B |
torch/_VF.py | py | 695 B |
torch/_VF.pyi | pyi | 1.12 MiB |
torch/__config__.py | py | 603 B |
torch/__future__.py | py | 3.18 KiB |
torch/__init__.py | py | 98.58 KiB |
torch/_appdirs.py | py | 26.21 KiB |
torch/_awaits | Folder | |
torch/_awaits/__init__.py | py | 1.67 KiB |
torch/_classes.py | py | 1.74 KiB |
torch/_compile.py | py | 1.31 KiB |
torch/_custom_op | Folder | |
torch/_custom_op/__init__.py | py | 0 B |
torch/_custom_op/autograd.py | py | 11.81 KiB |
torch/_custom_op/functional.py | py | 7.95 KiB |
torch/_custom_op/impl.py | py | 26.72 KiB |
torch/_custom_ops.py | py | 12.84 KiB |
torch/_decomp | Folder | |
torch/_decomp/__init__.py | py | 18.73 KiB |
torch/_decomp/decompositions.py | py | 178.32 KiB |
torch/_decomp/decompositions_for_jvp.py | py | 11.76 KiB |
torch/_decomp/decompositions_for_rng.py | py | 9.23 KiB |
torch/_deploy.py | py | 3.48 KiB |
torch/_dispatch | Folder | |
torch/_dispatch/__init__.py | py | 0 B |
torch/_dispatch/python.py | py | 6.40 KiB |
torch/_dynamo | Folder | |
torch/_dynamo/__init__.py | py | 4.40 KiB |
torch/_dynamo/_trace_wrapped_higher_order_op.py | py | 9.07 KiB |
torch/_dynamo/backends | Folder | |
torch/_dynamo/backends/__init__.py | py | 0 B |
torch/_dynamo/backends/common.py | py | 4.13 KiB |
torch/_dynamo/backends/cudagraphs.py | py | 8.42 KiB |
torch/_dynamo/backends/debugging.py | py | 13.53 KiB |
torch/_dynamo/backends/distributed.py | py | 24.56 KiB |
torch/_dynamo/backends/inductor.py | py | 303 B |
torch/_dynamo/backends/onnxrt.py | py | 1.54 KiB |
torch/_dynamo/backends/registry.py | py | 3.72 KiB |
torch/_dynamo/backends/tensorrt.py | py | 420 B |
torch/_dynamo/backends/torchxla.py | py | 1.27 KiB |
torch/_dynamo/backends/tvm.py | py | 6.98 KiB |
torch/_dynamo/bytecode_analysis.py | py | 8.79 KiB |
torch/_dynamo/bytecode_transformation.py | py | 56.33 KiB |
torch/_dynamo/cache_size.py | py | 7.89 KiB |
torch/_dynamo/callback.py | py | 2.63 KiB |
torch/_dynamo/code_context.py | py | 851 B |
torch/_dynamo/codegen.py | py | 21.56 KiB |
torch/_dynamo/compiled_autograd.py | py | 32.79 KiB |
torch/_dynamo/comptime.py | py | 13.26 KiB |
torch/_dynamo/config.py | py | 23.01 KiB |
torch/_dynamo/convert_frame.py | py | 53.27 KiB |
torch/_dynamo/create_parameter_op.py | py | 2.34 KiB |
torch/_dynamo/current_scope_id.py | py | 703 B |
torch/_dynamo/debug_utils.py | py | 28.90 KiB |
torch/_dynamo/decorators.py | py | 23.22 KiB |
torch/_dynamo/device_interface.py | py | 12.84 KiB |
torch/_dynamo/distributed.py | py | 619 B |
torch/_dynamo/eval_frame.py | py | 68.52 KiB |
torch/_dynamo/exc.py | py | 15.17 KiB |
torch/_dynamo/external_utils.py | py | 3.95 KiB |
torch/_dynamo/funcname_cache.py | py | 1.77 KiB |
torch/_dynamo/guards.py | py | 115.07 KiB |
torch/_dynamo/hooks.py | py | 304 B |
torch/_dynamo/logging.py | py | 1.79 KiB |
torch/_dynamo/metrics_context.py | py | 3.26 KiB |
torch/_dynamo/mutation_guard.py | py | 4.58 KiB |
torch/_dynamo/output_graph.py | py | 108.21 KiB |
torch/_dynamo/pgo.py | py | 26.00 KiB |
torch/_dynamo/polyfills | Folder | |
torch/_dynamo/polyfills/__init__.py | py | 5.60 KiB |
torch/_dynamo/polyfills/builtins.py | py | 1.36 KiB |
torch/_dynamo/polyfills/functools.py | py | 985 B |
torch/_dynamo/polyfills/itertools.py | py | 3.53 KiB |
torch/_dynamo/polyfills/loader.py | py | 1.20 KiB |
torch/_dynamo/polyfills/operator.py | py | 2.98 KiB |
torch/_dynamo/polyfills/os.py | py | 1014 B |
torch/_dynamo/polyfills/sys.py | py | 296 B |
torch/_dynamo/profiler.py | py | 5.15 KiB |
torch/_dynamo/replay_record.py | py | 3.67 KiB |
torch/_dynamo/repro | Folder | |
torch/_dynamo/repro/__init__.py | py | 0 B |
torch/_dynamo/repro/after_aot.py | py | 35.39 KiB |
torch/_dynamo/repro/after_dynamo.py | py | 19.84 KiB |
torch/_dynamo/repro/aoti.py | py | 15.23 KiB |
torch/_dynamo/resume_execution.py | py | 23.94 KiB |
torch/_dynamo/side_effects.py | py | 32.55 KiB |
torch/_dynamo/source.py | py | 27.83 KiB |
torch/_dynamo/symbolic_convert.py | py | 132.96 KiB |
torch/_dynamo/tensor_version_op.py | py | 2.14 KiB |
torch/_dynamo/test_case.py | py | 2.27 KiB |
torch/_dynamo/test_minifier_common.py | py | 10.68 KiB |
torch/_dynamo/testing.py | py | 14.62 KiB |
torch/_dynamo/trace_rules.py | py | 143.28 KiB |
torch/_dynamo/types.py | py | 2.23 KiB |
torch/_dynamo/utils.py | py | 119.48 KiB |
torch/_dynamo/variables | Folder | |
torch/_dynamo/variables/__init__.py | py | 4.94 KiB |
torch/_dynamo/variables/base.py | py | 16.52 KiB |
torch/_dynamo/variables/builder.py | py | 125.88 KiB |
torch/_dynamo/variables/builtin.py | py | 81.18 KiB |
torch/_dynamo/variables/constant.py | py | 9.38 KiB |
torch/_dynamo/variables/ctx_manager.py | py | 42.93 KiB |
torch/_dynamo/variables/dicts.py | py | 38.63 KiB |
torch/_dynamo/variables/distributed.py | py | 14.12 KiB |
torch/_dynamo/variables/functions.py | py | 45.67 KiB |
torch/_dynamo/variables/higher_order_ops.py | py | 105.99 KiB |
torch/_dynamo/variables/iter.py | py | 19.37 KiB |
torch/_dynamo/variables/lazy.py | py | 5.99 KiB |
torch/_dynamo/variables/lists.py | py | 37.53 KiB |
torch/_dynamo/variables/misc.py | py | 63.08 KiB |
torch/_dynamo/variables/nn_module.py | py | 49.63 KiB |
torch/_dynamo/variables/optimizer.py | py | 15.63 KiB |
torch/_dynamo/variables/script_object.py | py | 2.83 KiB |
torch/_dynamo/variables/sdpa.py | py | 2.48 KiB |
torch/_dynamo/variables/tensor.py | py | 54.97 KiB |
torch/_dynamo/variables/torch.py | py | 49.92 KiB |
torch/_dynamo/variables/torch_function.py | py | 24.27 KiB |
torch/_dynamo/variables/user_defined.py | py | 57.55 KiB |
torch/_environment.py | py | 44 B |
torch/_export | Folder | |
torch/_export/__init__.py | py | 14.38 KiB |
torch/_export/converter.py | py | 64.24 KiB |
torch/_export/db | Folder | |
torch/_export/db/__init__.py | py | 211 B |
torch/_export/db/case.py | py | 5.07 KiB |
torch/_export/db/examples | Folder | |
torch/_export/db/examples/__init__.py | py | 1.67 KiB |
torch/_export/db/examples/assume_constant_result.py | py | 530 B |
torch/_export/db/examples/autograd_function.py | py | 601 B |
torch/_export/db/examples/class_method.py | py | 521 B |
torch/_export/db/examples/cond_branch_class_method.py | py | 1.34 KiB |
torch/_export/db/examples/cond_branch_nested_function.py | py | 1.31 KiB |
torch/_export/db/examples/cond_branch_nonlocal_variables.py | py | 1.86 KiB |
torch/_export/db/examples/cond_closed_over_variable.py | py | 569 B |
torch/_export/db/examples/cond_operands.py | py | 882 B |
torch/_export/db/examples/cond_predicate.py | py | 688 B |
torch/_export/db/examples/constrain_as_size_example.py | py | 662 B |
torch/_export/db/examples/constrain_as_value_example.py | py | 717 B |
torch/_export/db/examples/decorator.py | py | 503 B |
torch/_export/db/examples/dictionary.py | py | 421 B |
torch/_export/db/examples/dynamic_shape_assert.py | py | 468 B |
torch/_export/db/examples/dynamic_shape_constructor.py | py | 411 B |
torch/_export/db/examples/dynamic_shape_if_guard.py | py | 579 B |
torch/_export/db/examples/dynamic_shape_map.py | py | 473 B |
torch/_export/db/examples/dynamic_shape_round.py | py | 546 B |
torch/_export/db/examples/dynamic_shape_slicing.py | py | 403 B |
torch/_export/db/examples/dynamic_shape_view.py | py | 461 B |
torch/_export/db/examples/fn_with_kwargs.py | py | 761 B |
torch/_export/db/examples/list_contains.py | py | 494 B |
torch/_export/db/examples/list_unpack.py | py | 614 B |
torch/_export/db/examples/model_attr_mutation.py | py | 688 B |
torch/_export/db/examples/nested_function.py | py | 514 B |
torch/_export/db/examples/null_context_manager.py | py | 499 B |
torch/_export/db/examples/optional_input.py | py | 475 B |
torch/_export/db/examples/pytree_flatten.py | py | 391 B |
torch/_export/db/examples/scalar_output.py | py | 566 B |
torch/_export/db/examples/specialized_attribute.py | py | 546 B |
torch/_export/db/examples/static_for_loop.py | py | 401 B |
torch/_export/db/examples/static_if.py | py | 415 B |
torch/_export/db/examples/tensor_setattr.py | py | 352 B |
torch/_export/db/examples/type_reflection_method.py | py | 483 B |
torch/_export/db/examples/unsupported_operator.py | py | 429 B |
torch/_export/db/examples/user_input_mutation.py | py | 319 B |
torch/_export/db/gen_example.py | py | 483 B |
torch/_export/db/logging.py | py | 1.62 KiB |
torch/_export/error.py | py | 1.78 KiB |
torch/_export/non_strict_utils.py | py | 21.99 KiB |
torch/_export/pass_base.py | py | 17.68 KiB |
torch/_export/pass_infra | Folder | |
torch/_export/pass_infra/__init__.py | py | 0 B |
torch/_export/pass_infra/node_metadata.py | py | 814 B |
torch/_export/pass_infra/proxy_value.py | py | 1.18 KiB |
torch/_export/passes | Folder | |
torch/_export/passes/__init__.py | py | 89 B |
torch/_export/passes/_node_metadata_hook.py | py | 2.45 KiB |
torch/_export/passes/add_runtime_assertions_for_constraints_pass.py | py | 9.74 KiB |
torch/_export/passes/collect_tracepoints_pass.py | py | 6.10 KiB |
torch/_export/passes/constant_folding.py | py | 11.08 KiB |
torch/_export/passes/functionalize_side_effectful_ops_pass.py | py | 3.29 KiB |
torch/_export/passes/lift_constants_pass.py | py | 14.13 KiB |
torch/_export/passes/remove_runtime_assertions.py | py | 1.07 KiB |
torch/_export/passes/replace_autocast_with_hop_pass.py | py | 6.72 KiB |
torch/_export/passes/replace_quantized_ops_with_standard_ops_pass.py | py | 25.73 KiB |
torch/_export/passes/replace_set_grad_with_hop_pass.py | py | 3.85 KiB |
torch/_export/passes/replace_view_ops_with_view_copy_ops_pass.py | py | 2.45 KiB |
torch/_export/passes/replace_with_hop_pass_util.py | py | 7.03 KiB |
torch/_export/serde | Folder | |
torch/_export/serde/__init__.py | py | 0 B |
torch/_export/serde/aoti_schema.py | py | 256 B |
torch/_export/serde/dynamic_shapes.py | py | 11.39 KiB |
torch/_export/serde/schema.py | py | 9.93 KiB |
torch/_export/serde/schema.yaml | yaml | 8.37 KiB |
torch/_export/serde/schema_check.py | py | 18.67 KiB |
torch/_export/serde/serialize.py | py | 125.41 KiB |
torch/_export/serde/union.py | py | 2.04 KiB |
torch/_export/tools.py | py | 4.59 KiB |
torch/_export/utils.py | py | 43.60 KiB |
torch/_export/verifier.py | py | 18.61 KiB |
torch/_export/wrappers.py | py | 4.19 KiB |
torch/_functorch | Folder | |
torch/_functorch/__init__.py | py | 211 B |
torch/_functorch/_aot_autograd | Folder | |
torch/_functorch/_aot_autograd/__init__.py | py | 211 B |
torch/_functorch/_aot_autograd/autograd_cache.py | py | 32.90 KiB |
torch/_functorch/_aot_autograd/collect_metadata_analysis.py | py | 40.52 KiB |
torch/_functorch/_aot_autograd/dispatch_and_compile_graph.py | py | 12.70 KiB |
torch/_functorch/_aot_autograd/functional_utils.py | py | 22.73 KiB |
torch/_functorch/_aot_autograd/input_output_analysis.py | py | 21.10 KiB |
torch/_functorch/_aot_autograd/jit_compile_runtime_wrappers.py | py | 36.24 KiB |
torch/_functorch/_aot_autograd/logging_utils.py | py | 4.69 KiB |
torch/_functorch/_aot_autograd/runtime_wrappers.py | py | 97.42 KiB |
torch/_functorch/_aot_autograd/schemas.py | py | 39.32 KiB |
torch/_functorch/_aot_autograd/subclass_parametrization.py | py | 2.49 KiB |
torch/_functorch/_aot_autograd/subclass_utils.py | py | 17.63 KiB |
torch/_functorch/_aot_autograd/traced_function_transforms.py | py | 43.57 KiB |
torch/_functorch/_aot_autograd/utils.py | py | 18.23 KiB |
torch/_functorch/aot_autograd.py | py | 67.82 KiB |
torch/_functorch/apis.py | py | 19.11 KiB |
torch/_functorch/autograd_function.py | py | 28.50 KiB |
torch/_functorch/batch_norm_replacement.py | py | 938 B |
torch/_functorch/benchmark_utils.py | py | 6.36 KiB |
torch/_functorch/compile_utils.py | py | 6.12 KiB |
torch/_functorch/compilers.py | py | 14.15 KiB |
torch/_functorch/config.py | py | 9.84 KiB |
torch/_functorch/deprecated.py | py | 5.26 KiB |
torch/_functorch/eager_transforms.py | py | 72.33 KiB |
torch/_functorch/functional_call.py | py | 10.58 KiB |
torch/_functorch/fx_minifier.py | py | 17.45 KiB |
torch/_functorch/make_functional.py | py | 22.86 KiB |
torch/_functorch/partitioners.py | py | 74.92 KiB |
torch/_functorch/pyfunctorch.py | py | 10.40 KiB |
torch/_functorch/python_key.py | py | 457 B |
torch/_functorch/pytree_hacks.py | py | 721 B |
torch/_functorch/top_operators_github_usage.py | py | 21.56 KiB |
torch/_functorch/utils.py | py | 1016 B |
torch/_functorch/vmap.py | py | 18.91 KiB |
torch/_guards.py | py | 33.49 KiB |
torch/_higher_order_ops | Folder | |
torch/_higher_order_ops/__init__.py | py | 641 B |
torch/_higher_order_ops/associative_scan.py | py | 16.80 KiB |
torch/_higher_order_ops/auto_functionalize.py | py | 29.20 KiB |
torch/_higher_order_ops/cond.py | py | 22.44 KiB |
torch/_higher_order_ops/effects.py | py | 9.58 KiB |
torch/_higher_order_ops/executorch_call_delegate.py | py | 5.99 KiB |
torch/_higher_order_ops/flex_attention.py | py | 39.24 KiB |
torch/_higher_order_ops/hints_wrap.py | py | 5.35 KiB |
torch/_higher_order_ops/invoke_subgraph.py | py | 11.11 KiB |
torch/_higher_order_ops/map.py | py | 8.98 KiB |
torch/_higher_order_ops/out_dtype.py | py | 5.61 KiB |
torch/_higher_order_ops/prim_hop_base.py | py | 7.34 KiB |
torch/_higher_order_ops/run_const_graph.py | py | 1.88 KiB |
torch/_higher_order_ops/scan.py | py | 18.77 KiB |
torch/_higher_order_ops/strict_mode.py | py | 3.68 KiB |
torch/_higher_order_ops/torchbind.py | py | 5.42 KiB |
torch/_higher_order_ops/triton_kernel_wrap.py | py | 55.07 KiB |
torch/_higher_order_ops/utils.py | py | 18.53 KiB |
torch/_higher_order_ops/while_loop.py | py | 11.40 KiB |
torch/_higher_order_ops/wrap.py | py | 9.80 KiB |
torch/_inductor | Folder | |
torch/_inductor/__init__.py | py | 11.78 KiB |
torch/_inductor/aoti_eager.py | py | 11.18 KiB |
torch/_inductor/async_compile.py | py | 11.92 KiB |
torch/_inductor/autoheuristic | Folder | |
torch/_inductor/autoheuristic/__init__.py | py | 0 B |
torch/_inductor/autoheuristic/artifacts | Folder | |
torch/_inductor/autoheuristic/artifacts/_MMRankingA100.py | py | 27.68 KiB |
torch/_inductor/autoheuristic/artifacts/_MMRankingH100.py | py | 30.26 KiB |
torch/_inductor/autoheuristic/artifacts/_MixedMMA100.py | py | 7.88 KiB |
torch/_inductor/autoheuristic/artifacts/_MixedMMH100.py | py | 7.84 KiB |
torch/_inductor/autoheuristic/artifacts/_PadMMA100.py | py | 4.92 KiB |
torch/_inductor/autoheuristic/artifacts/__init__.py | py | 0 B |
torch/_inductor/autoheuristic/autoheuristic.py | py | 11.97 KiB |
torch/_inductor/autoheuristic/autoheuristic_utils.py | py | 11.37 KiB |
torch/_inductor/autoheuristic/learned_heuristic_controller.py | py | 4.34 KiB |
torch/_inductor/autoheuristic/learnedheuristic_interface.py | py | 2.84 KiB |
torch/_inductor/autotune_process.py | py | 31.58 KiB |
torch/_inductor/bounds.py | py | 5.79 KiB |
torch/_inductor/choices.py | py | 3.68 KiB |
torch/_inductor/codecache.py | py | 139.01 KiB |
torch/_inductor/codegen | Folder | |
torch/_inductor/codegen/__init__.py | py | 0 B |
torch/_inductor/codegen/aoti_hipify_utils.py | py | 1.30 KiB |
torch/_inductor/codegen/aoti_runtime | Folder | |
torch/_inductor/codegen/aoti_runtime/implementation.cpp | cpp | 3.22 KiB |
torch/_inductor/codegen/aoti_runtime/interface.cpp | cpp | 13.16 KiB |
torch/_inductor/codegen/common.py | py | 93.84 KiB |
torch/_inductor/codegen/cpp.py | py | 199.12 KiB |
torch/_inductor/codegen/cpp_gemm_template.py | py | 50.27 KiB |
torch/_inductor/codegen/cpp_micro_gemm.py | py | 32.17 KiB |
torch/_inductor/codegen/cpp_prefix.h | h | 31.44 KiB |
torch/_inductor/codegen/cpp_template.py | py | 4.44 KiB |
torch/_inductor/codegen/cpp_template_kernel.py | py | 15.45 KiB |
torch/_inductor/codegen/cpp_utils.py | py | 35.71 KiB |
torch/_inductor/codegen/cpp_wrapper_cpu.py | py | 93.15 KiB |
torch/_inductor/codegen/cpp_wrapper_cpu_array_ref.py | py | 45.92 KiB |
torch/_inductor/codegen/cpp_wrapper_gpu.py | py | 19.58 KiB |
torch/_inductor/codegen/cpu_device_op_overrides.py | py | 606 B |
torch/_inductor/codegen/cuda | Folder | |
torch/_inductor/codegen/cuda/__init__.py | py | 0 B |
torch/_inductor/codegen/cuda/cuda_cpp_scheduling.py | py | 4.39 KiB |
torch/_inductor/codegen/cuda/cuda_env.py | py | 1.16 KiB |
torch/_inductor/codegen/cuda/cuda_kernel.py | py | 15.13 KiB |
torch/_inductor/codegen/cuda/cuda_template.py | py | 9.02 KiB |
torch/_inductor/codegen/cuda/cutlass_epilogue_gen.py | py | 14.51 KiB |
torch/_inductor/codegen/cuda/cutlass_lib_extensions | Folder | |
torch/_inductor/codegen/cuda/cutlass_lib_extensions/__init__.py | py | 0 B |
torch/_inductor/codegen/cuda/cutlass_lib_extensions/gemm_operation_extensions.py | py | 9.76 KiB |
torch/_inductor/codegen/cuda/cutlass_utils.py | py | 12.13 KiB |
torch/_inductor/codegen/cuda/device_op_overrides.py | py | 9.43 KiB |
torch/_inductor/codegen/cuda/gemm_template.py | py | 62.78 KiB |
torch/_inductor/codegen/cuda_combined_scheduling.py | py | 3.94 KiB |
torch/_inductor/codegen/debug_utils.py | py | 11.06 KiB |
torch/_inductor/codegen/halide.py | py | 61.99 KiB |
torch/_inductor/codegen/memory_planning.py | py | 25.10 KiB |
torch/_inductor/codegen/multi_kernel.py | py | 15.84 KiB |
torch/_inductor/codegen/rocm | Folder | |
torch/_inductor/codegen/rocm/__init__.py | py | 0 B |
torch/_inductor/codegen/rocm/ck_conv_template.py | py | 23.54 KiB |
torch/_inductor/codegen/rocm/ck_template.py | py | 3.15 KiB |
torch/_inductor/codegen/rocm/ck_universal_gemm_template.py | py | 29.07 KiB |
torch/_inductor/codegen/rocm/compile_command.py | py | 4.49 KiB |
torch/_inductor/codegen/rocm/rocm_benchmark_request.py | py | 5.05 KiB |
torch/_inductor/codegen/rocm/rocm_cpp_scheduling.py | py | 3.85 KiB |
torch/_inductor/codegen/rocm/rocm_kernel.py | py | 9.80 KiB |
torch/_inductor/codegen/rocm/rocm_template.py | py | 5.99 KiB |
torch/_inductor/codegen/rocm/rocm_template_buffer.py | py | 650 B |
torch/_inductor/codegen/simd.py | py | 65.60 KiB |
torch/_inductor/codegen/simd_kernel_features.py | py | 6.39 KiB |
torch/_inductor/codegen/triton.py | py | 145.90 KiB |
torch/_inductor/codegen/triton_combo_kernel.py | py | 46.82 KiB |
torch/_inductor/codegen/triton_split_scan.py | py | 6.71 KiB |
torch/_inductor/codegen/triton_utils.py | py | 6.12 KiB |
torch/_inductor/codegen/wrapper.py | py | 96.79 KiB |
torch/_inductor/codegen/xpu | Folder | |
torch/_inductor/codegen/xpu/__init__.py | py | 0 B |
torch/_inductor/codegen/xpu/device_op_overrides.py | py | 992 B |
torch/_inductor/comm_analysis.py | py | 8.35 KiB |
torch/_inductor/comm_lowering.py | py | 12.42 KiB |
torch/_inductor/comms.py | py | 32.58 KiB |
torch/_inductor/compile_fx.py | py | 71.17 KiB |
torch/_inductor/compile_worker | Folder | |
torch/_inductor/compile_worker/__init__.py | py | 0 B |
torch/_inductor/compile_worker/__main__.py | py | 1.28 KiB |
torch/_inductor/compile_worker/subproc_pool.py | py | 11.62 KiB |
torch/_inductor/compile_worker/watchdog.py | py | 1.25 KiB |
torch/_inductor/compiler_bisector.py | py | 21.88 KiB |
torch/_inductor/config.py | py | 53.95 KiB |
torch/_inductor/constant_folding.py | py | 13.86 KiB |
torch/_inductor/cpp_builder.py | py | 53.53 KiB |
torch/_inductor/cpu_vec_isa.py | py | 12.96 KiB |
torch/_inductor/cudagraph_trees.py | py | 99.69 KiB |
torch/_inductor/cudagraph_utils.py | py | 10.94 KiB |
torch/_inductor/custom_graph_pass.py | py | 2.54 KiB |
torch/_inductor/debug.py | py | 23.41 KiB |
torch/_inductor/decomposition.py | py | 32.98 KiB |
torch/_inductor/dependencies.py | py | 27.57 KiB |
torch/_inductor/exc.py | py | 3.05 KiB |
torch/_inductor/extern_node_serializer.py | py | 884 B |
torch/_inductor/freezing.py | py | 10.14 KiB |
torch/_inductor/fx_passes | Folder | |
torch/_inductor/fx_passes/__init__.py | py | 0 B |
torch/_inductor/fx_passes/b2b_gemm.py | py | 25.03 KiB |
torch/_inductor/fx_passes/binary_folding.py | py | 19.71 KiB |
torch/_inductor/fx_passes/ddp_fusion.py | py | 21.20 KiB |
torch/_inductor/fx_passes/decompose_mem_bound_mm.py | py | 5.23 KiB |
torch/_inductor/fx_passes/dedupe_symint_uses.py | py | 2.47 KiB |
torch/_inductor/fx_passes/efficient_conv_bn_eval.py | py | 14.12 KiB |
torch/_inductor/fx_passes/freezing_patterns.py | py | 6.80 KiB |
torch/_inductor/fx_passes/fuse_attention.py | py | 29.78 KiB |
torch/_inductor/fx_passes/group_batch_fusion.py | py | 55.99 KiB |
torch/_inductor/fx_passes/joint_graph.py | py | 26.75 KiB |
torch/_inductor/fx_passes/micro_pipeline_tp.py | py | 28.98 KiB |
torch/_inductor/fx_passes/misc_patterns.py | py | 4.79 KiB |
torch/_inductor/fx_passes/mkldnn_fusion.py | py | 49.76 KiB |
torch/_inductor/fx_passes/numeric_utils.py | py | 7.26 KiB |
torch/_inductor/fx_passes/pad_mm.py | py | 28.44 KiB |
torch/_inductor/fx_passes/post_grad.py | py | 42.44 KiB |
torch/_inductor/fx_passes/pre_grad.py | py | 30.17 KiB |
torch/_inductor/fx_passes/quantization.py | py | 97.14 KiB |
torch/_inductor/fx_passes/reinplace.py | py | 28.07 KiB |
torch/_inductor/fx_passes/replace_random.py | py | 4.10 KiB |
torch/_inductor/fx_passes/serialized_patterns | Folder | |
torch/_inductor/fx_passes/serialized_patterns/__init__.py | py | 0 B |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_1.py | py | 11.07 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_10.py | py | 14.07 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_11.py | py | 13.84 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_12.py | py | 15.10 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_13.py | py | 7.79 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_14.py | py | 14.18 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_15.py | py | 15.88 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_16.py | py | 43.14 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_17.py | py | 17.10 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_18.py | py | 32.39 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_19.py | py | 13.90 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_2.py | py | 11.08 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_3.py | py | 12.32 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_4.py | py | 12.29 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_5.py | py | 11.30 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_6.py | py | 12.52 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_7.py | py | 15.27 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_8.py | py | 14.05 KiB |
torch/_inductor/fx_passes/serialized_patterns/_sfdp_pattern_9.py | py | 15.28 KiB |
torch/_inductor/fx_passes/serialized_patterns/addmm_pattern.py | py | 1.85 KiB |
torch/_inductor/fx_passes/serialized_patterns/bmm_pattern.py | py | 1.27 KiB |
torch/_inductor/fx_passes/serialized_patterns/mm_pattern.py | py | 1.26 KiB |
torch/_inductor/fx_passes/split_cat.py | py | 103.89 KiB |
torch/_inductor/fx_utils.py | py | 9.04 KiB |
torch/_inductor/graph.py | py | 91.45 KiB |
torch/_inductor/hooks.py | py | 675 B |
torch/_inductor/index_propagation.py | py | 12.98 KiB |
torch/_inductor/inductor_prims.py | py | 6.07 KiB |
torch/_inductor/ir.py | py | 275.77 KiB |
torch/_inductor/jagged_lowerings.py | py | 9.06 KiB |
torch/_inductor/kernel | Folder | |
torch/_inductor/kernel/__init__.py | py | 58 B |
torch/_inductor/kernel/bmm.py | py | 7.28 KiB |
torch/_inductor/kernel/conv.py | py | 21.88 KiB |
torch/_inductor/kernel/flex_attention.py | py | 71.18 KiB |
torch/_inductor/kernel/flex_decoding.py | py | 21.49 KiB |
torch/_inductor/kernel/mm.py | py | 26.35 KiB |
torch/_inductor/kernel/mm_common.py | py | 18.44 KiB |
torch/_inductor/kernel/mm_plus_mm.py | py | 7.86 KiB |
torch/_inductor/kernel/mm_scaled.py | py | 10.86 KiB |
torch/_inductor/kernel/unpack_mixed_mm.py | py | 2.98 KiB |
torch/_inductor/loop_body.py | py | 24.06 KiB |
torch/_inductor/lowering.py | py | 212.35 KiB |
torch/_inductor/memory.py | py | 24.73 KiB |
torch/_inductor/metrics.py | py | 13.26 KiB |
torch/_inductor/mkldnn_ir.py | py | 39.00 KiB |
torch/_inductor/mkldnn_lowerings.py | py | 44.98 KiB |
torch/_inductor/mock_cache.py | py | 8.63 KiB |
torch/_inductor/ops_handler.py | py | 31.89 KiB |
torch/_inductor/optimize_indexing.py | py | 4.17 KiB |
torch/_inductor/package | Folder | |
torch/_inductor/package/__init__.py | py | 68 B |
torch/_inductor/package/build_package.py | py | 344 B |
torch/_inductor/package/package.py | py | 8.97 KiB |
torch/_inductor/package/pt2_archive_constants.py | py | 561 B |
torch/_inductor/pattern_matcher.py | py | 71.63 KiB |
torch/_inductor/quantized_lowerings.py | py | 3.19 KiB |
torch/_inductor/remote_cache.py | py | 12.09 KiB |
torch/_inductor/runtime | Folder | |
torch/_inductor/runtime/__init__.py | py | 0 B |
torch/_inductor/runtime/autotune_cache.py | py | 16.87 KiB |
torch/_inductor/runtime/benchmarking.py | py | 7.88 KiB |
torch/_inductor/runtime/cache_dir_utils.py | py | 831 B |
torch/_inductor/runtime/compile_tasks.py | py | 2.09 KiB |
torch/_inductor/runtime/coordinate_descent_tuner.py | py | 10.08 KiB |
torch/_inductor/runtime/halide_helpers.py | py | 3.57 KiB |
torch/_inductor/runtime/hints.py | py | 6.35 KiB |
torch/_inductor/runtime/runtime_utils.py | py | 4.50 KiB |
torch/_inductor/runtime/triton_helpers.py | py | 20.56 KiB |
torch/_inductor/runtime/triton_heuristics.py | py | 73.57 KiB |
torch/_inductor/scheduler.py | py | 149.52 KiB |
torch/_inductor/select_algorithm.py | py | 65.94 KiB |
torch/_inductor/sizevars.py | py | 35.36 KiB |
torch/_inductor/subgraph_lowering.py | py | 5.02 KiB |
torch/_inductor/test_case.py | py | 1.00 KiB |
torch/_inductor/test_operators.py | py | 843 B |
torch/_inductor/triton_bundler.py | py | 10.32 KiB |
torch/_inductor/utils.py | py | 69.02 KiB |
torch/_inductor/virtualized.py | py | 12.98 KiB |
torch/_inductor/wrapper_benchmark.py | py | 11.92 KiB |
torch/_jit_internal.py | py | 53.90 KiB |
torch/_lazy | Folder | |
torch/_lazy/__init__.py | py | 1.80 KiB |
torch/_lazy/closure.py | py | 5.46 KiB |
torch/_lazy/computation.py | py | 946 B |
torch/_lazy/config.py | py | 464 B |
torch/_lazy/debug.py | py | 760 B |
torch/_lazy/device_context.py | py | 683 B |
torch/_lazy/extract_compiled_graph.py | py | 8.46 KiB |
torch/_lazy/ir_cache.py | py | 362 B |
torch/_lazy/metrics.py | py | 567 B |
torch/_lazy/tensor_factory_functions.py | py | 1.38 KiB |
torch/_lazy/ts_backend.py | py | 170 B |
torch/_library | Folder | |
torch/_library/__init__.py | py | 262 B |
torch/_library/autograd.py | py | 8.84 KiB |
torch/_library/custom_ops.py | py | 32.38 KiB |
torch/_library/fake_class_registry.py | py | 12.57 KiB |
torch/_library/fake_impl.py | py | 8.21 KiB |
torch/_library/infer_schema.py | py | 10.88 KiB |
torch/_library/simple_registry.py | py | 2.66 KiB |
torch/_library/triton.py | py | 9.29 KiB |
torch/_library/utils.py | py | 16.67 KiB |
torch/_linalg_utils.py | py | 5.20 KiB |
torch/_lobpcg.py | py | 43.66 KiB |
torch/_logging | Folder | |
torch/_logging/__init__.py | py | 818 B |
torch/_logging/_internal.py | py | 44.92 KiB |
torch/_logging/_registrations.py | py | 6.56 KiB |
torch/_logging/scribe.py | py | 2.58 KiB |
torch/_logging/structured.py | py | 1.33 KiB |
torch/_lowrank.py | py | 10.60 KiB |
torch/_meta_registrations.py | py | 207.49 KiB |
torch/_namedtensor_internals.py | py | 5.32 KiB |
torch/_numpy | Folder | |
torch/_numpy/__init__.py | py | 590 B |
torch/_numpy/_binary_ufuncs_impl.py | py | 1.91 KiB |
torch/_numpy/_casting_dicts.py | py | 42.82 KiB |
torch/_numpy/_dtypes.py | py | 10.53 KiB |
torch/_numpy/_dtypes_impl.py | py | 5.98 KiB |
torch/_numpy/_funcs.py | py | 2.12 KiB |
torch/_numpy/_funcs_impl.py | py | 59.82 KiB |
torch/_numpy/_getlimits.py | py | 284 B |
torch/_numpy/_ndarray.py | py | 16.82 KiB |
torch/_numpy/_normalizations.py | py | 8.31 KiB |
torch/_numpy/_reductions_impl.py | py | 11.97 KiB |
torch/_numpy/_ufuncs.py | py | 8.50 KiB |
torch/_numpy/_unary_ufuncs_impl.py | py | 1.20 KiB |
torch/_numpy/_util.py | py | 7.63 KiB |
torch/_numpy/fft.py | py | 2.87 KiB |
torch/_numpy/linalg.py | py | 5.68 KiB |
torch/_numpy/random.py | py | 4.73 KiB |
torch/_numpy/testing | Folder | |
torch/_numpy/testing/__init__.py | py | 395 B |
torch/_numpy/testing/utils.py | py | 76.92 KiB |
torch/_ops.py | py | 55.63 KiB |
torch/_prims | Folder | |
torch/_prims/__init__.py | py | 81.86 KiB |
torch/_prims/context.py | py | 5.74 KiB |
torch/_prims/debug_prims.py | py | 1.90 KiB |
torch/_prims/executor.py | py | 1.68 KiB |
torch/_prims/rng_prims.py | py | 11.57 KiB |
torch/_prims_common | Folder | |
torch/_prims_common/__init__.py | py | 66.10 KiB |
torch/_prims_common/wrappers.py | py | 17.05 KiB |
torch/_python_dispatcher.py | py | 7.15 KiB |
torch/_refs | Folder | |
torch/_refs/__init__.py | py | 215.09 KiB |
torch/_refs/_conversions.py | py | 3.57 KiB |
torch/_refs/fft.py | py | 18.11 KiB |
torch/_refs/linalg | Folder | |
torch/_refs/linalg/__init__.py | py | 10.56 KiB |
torch/_refs/nn | Folder | |
torch/_refs/nn/__init__.py | py | 54 B |
torch/_refs/nn/functional | Folder | |
torch/_refs/nn/functional/__init__.py | py | 42.42 KiB |
torch/_refs/special | Folder | |
torch/_refs/special/__init__.py | py | 6.86 KiB |
torch/_size_docs.py | py | 947 B |
torch/_sources.py | py | 4.48 KiB |
torch/_storage_docs.py | py | 1.39 KiB |
torch/_streambase.py | py | 455 B |
torch/_strobelight | Folder | |
torch/_strobelight/__init__.py | py | 0 B |
torch/_strobelight/cli_function_profiler.py | py | 11.51 KiB |
torch/_strobelight/compile_time_profiler.py | py | 6.63 KiB |
torch/_subclasses | Folder | |
torch/_subclasses/__init__.py | py | 392 B |
torch/_subclasses/_fake_tensor_utils.py | py | 8.56 KiB |
torch/_subclasses/fake_impls.py | py | 34.73 KiB |
torch/_subclasses/fake_tensor.py | py | 113.27 KiB |
torch/_subclasses/fake_utils.py | py | 10.34 KiB |
torch/_subclasses/functional_tensor.py | py | 36.61 KiB |
torch/_subclasses/meta_utils.py | py | 82.54 KiB |
torch/_subclasses/schema_check_mode.py | py | 8.68 KiB |
torch/_tensor.py | py | 66.40 KiB |
torch/_tensor_docs.py | py | 146.07 KiB |
torch/_tensor_str.py | py | 27.04 KiB |
torch/_thread_safe_fork.py | py | 0 B |
torch/_torch_docs.py | py | 426.34 KiB |
torch/_utils.py | py | 37.80 KiB |
torch/_utils_internal.py | py | 8.43 KiB |
torch/_vendor | Folder | |
torch/_vendor/__init__.py | py | 0 B |
torch/_vendor/packaging | Folder | |
torch/_vendor/packaging/__init__.py | py | 511 B |
torch/_vendor/packaging/_structures.py | py | 1.46 KiB |
torch/_vendor/packaging/version.py | py | 16.41 KiB |
torch/_vmap_internals.py | py | 9.48 KiB |
torch/_weights_only_unpickler.py | py | 20.55 KiB |
torch/accelerator | Folder | |
torch/accelerator/__init__.py | py | 5.16 KiB |
torch/accelerator/_utils.py | py | 972 B |
torch/amp | Folder | |
torch/amp/__init__.py | py | 190 B |
torch/amp/autocast_mode.py | py | 23.99 KiB |
torch/amp/grad_scaler.py | py | 30.35 KiB |
torch/ao | Folder | |
torch/ao/__init__.py | py | 446 B |
torch/ao/nn | Folder | |
torch/ao/nn/__init__.py | py | 550 B |
torch/ao/nn/intrinsic | Folder | |
torch/ao/nn/intrinsic/__init__.py | py | 989 B |
torch/ao/nn/intrinsic/modules | Folder | |
torch/ao/nn/intrinsic/modules/__init__.py | py | 696 B |
torch/ao/nn/intrinsic/modules/fused.py | py | 9.80 KiB |
torch/ao/nn/intrinsic/qat | Folder | |
torch/ao/nn/intrinsic/qat/__init__.py | py | 38 B |
torch/ao/nn/intrinsic/qat/modules | Folder | |
torch/ao/nn/intrinsic/qat/modules/__init__.py | py | 579 B |
torch/ao/nn/intrinsic/qat/modules/conv_fused.py | py | 33.42 KiB |
torch/ao/nn/intrinsic/qat/modules/linear_fused.py | py | 6.46 KiB |
torch/ao/nn/intrinsic/qat/modules/linear_relu.py | py | 1.70 KiB |
torch/ao/nn/intrinsic/quantized | Folder | |
torch/ao/nn/intrinsic/quantized/__init__.py | py | 251 B |
torch/ao/nn/intrinsic/quantized/dynamic | Folder | |
torch/ao/nn/intrinsic/quantized/dynamic/__init__.py | py | 38 B |
torch/ao/nn/intrinsic/quantized/dynamic/modules | Folder | |
torch/ao/nn/intrinsic/quantized/dynamic/modules/__init__.py | py | 76 B |
torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.py | py | 2.01 KiB |
torch/ao/nn/intrinsic/quantized/modules | Folder | |
torch/ao/nn/intrinsic/quantized/modules/__init__.py | py | 427 B |
torch/ao/nn/intrinsic/quantized/modules/bn_relu.py | py | 3.26 KiB |
torch/ao/nn/intrinsic/quantized/modules/conv_add.py | py | 4.42 KiB |
torch/ao/nn/intrinsic/quantized/modules/conv_relu.py | py | 8.43 KiB |
torch/ao/nn/intrinsic/quantized/modules/linear_relu.py | py | 6.78 KiB |
torch/ao/nn/qat | Folder | |
torch/ao/nn/qat/__init__.py | py | 38 B |
torch/ao/nn/qat/dynamic | Folder | |
torch/ao/nn/qat/dynamic/__init__.py | py | 38 B |
torch/ao/nn/qat/dynamic/modules | Folder | |
torch/ao/nn/qat/dynamic/modules/__init__.py | py | 54 B |
torch/ao/nn/qat/dynamic/modules/linear.py | py | 1.02 KiB |
torch/ao/nn/qat/modules | Folder | |
torch/ao/nn/qat/modules/__init__.py | py | 241 B |
torch/ao/nn/qat/modules/conv.py | py | 9.91 KiB |
torch/ao/nn/qat/modules/embedding_ops.py | py | 7.87 KiB |
torch/ao/nn/qat/modules/linear.py | py | 3.07 KiB |
torch/ao/nn/quantizable | Folder | |
torch/ao/nn/quantizable/__init__.py | py | 38 B |
torch/ao/nn/quantizable/modules | Folder | |
torch/ao/nn/quantizable/modules/__init__.py | py | 154 B |
torch/ao/nn/quantizable/modules/activation.py | py | 23.14 KiB |
torch/ao/nn/quantizable/modules/rnn.py | py | 17.75 KiB |
torch/ao/nn/quantized | Folder | |
torch/ao/nn/quantized/__init__.py | py | 725 B |
torch/ao/nn/quantized/dynamic | Folder | |
torch/ao/nn/quantized/dynamic/__init__.py | py | 38 B |
torch/ao/nn/quantized/dynamic/modules | Folder | |
torch/ao/nn/quantized/dynamic/modules/__init__.py | py | 439 B |
torch/ao/nn/quantized/dynamic/modules/conv.py | py | 18.28 KiB |
torch/ao/nn/quantized/dynamic/modules/linear.py | py | 6.35 KiB |
torch/ao/nn/quantized/dynamic/modules/rnn.py | py | 51.46 KiB |
torch/ao/nn/quantized/functional.py | py | 29.66 KiB |
torch/ao/nn/quantized/modules | Folder | |
torch/ao/nn/quantized/modules/__init__.py | py | 4.57 KiB |
torch/ao/nn/quantized/modules/activation.py | py | 11.65 KiB |
torch/ao/nn/quantized/modules/batchnorm.py | py | 4.40 KiB |
torch/ao/nn/quantized/modules/conv.py | py | 43.34 KiB |
torch/ao/nn/quantized/modules/dropout.py | py | 836 B |
torch/ao/nn/quantized/modules/embedding_ops.py | py | 14.83 KiB |
torch/ao/nn/quantized/modules/functional_modules.py | py | 9.32 KiB |
torch/ao/nn/quantized/modules/linear.py | py | 13.66 KiB |
torch/ao/nn/quantized/modules/normalization.py | py | 9.65 KiB |
torch/ao/nn/quantized/modules/rnn.py | py | 1.83 KiB |
torch/ao/nn/quantized/modules/utils.py | py | 4.73 KiB |
torch/ao/nn/quantized/reference | Folder | |
torch/ao/nn/quantized/reference/__init__.py | py | 303 B |
torch/ao/nn/quantized/reference/modules | Folder | |
torch/ao/nn/quantized/reference/modules/__init__.py | py | 523 B |
torch/ao/nn/quantized/reference/modules/conv.py | py | 15.39 KiB |
torch/ao/nn/quantized/reference/modules/linear.py | py | 2.25 KiB |
torch/ao/nn/quantized/reference/modules/rnn.py | py | 29.72 KiB |
torch/ao/nn/quantized/reference/modules/sparse.py | py | 4.73 KiB |
torch/ao/nn/quantized/reference/modules/utils.py | py | 15.30 KiB |
torch/ao/nn/sparse | Folder | |
torch/ao/nn/sparse/__init__.py | py | 25 B |
torch/ao/nn/sparse/quantized | Folder | |
torch/ao/nn/sparse/quantized/__init__.py | py | 178 B |
torch/ao/nn/sparse/quantized/dynamic | Folder | |
torch/ao/nn/sparse/quantized/dynamic/__init__.py | py | 63 B |
torch/ao/nn/sparse/quantized/dynamic/linear.py | py | 6.36 KiB |
torch/ao/nn/sparse/quantized/linear.py | py | 9.09 KiB |
torch/ao/nn/sparse/quantized/utils.py | py | 1.88 KiB |
torch/ao/ns | Folder | |
torch/ao/ns/__init__.py | py | 0 B |
torch/ao/ns/_numeric_suite.py | py | 20.13 KiB |
torch/ao/ns/_numeric_suite_fx.py | py | 41.68 KiB |
torch/ao/ns/fx | Folder | |
torch/ao/ns/fx/__init__.py | py | 0 B |
torch/ao/ns/fx/graph_matcher.py | py | 19.29 KiB |
torch/ao/ns/fx/graph_passes.py | py | 44.49 KiB |
torch/ao/ns/fx/mappings.py | py | 18.65 KiB |
torch/ao/ns/fx/n_shadows_utils.py | py | 51.43 KiB |
torch/ao/ns/fx/ns_types.py | py | 2.35 KiB |
torch/ao/ns/fx/pattern_utils.py | py | 8.40 KiB |
torch/ao/ns/fx/qconfig_multi_mapping.py | py | 10.20 KiB |
torch/ao/ns/fx/utils.py | py | 20.66 KiB |
torch/ao/ns/fx/weight_utils.py | py | 11.20 KiB |
torch/ao/pruning | Folder | |
torch/ao/pruning/__init__.py | py | 663 B |
torch/ao/pruning/_experimental | Folder | |
torch/ao/pruning/_experimental/__init__.py | py | 0 B |
torch/ao/pruning/_experimental/activation_sparsifier | Folder | |
torch/ao/pruning/_experimental/activation_sparsifier/__init__.py | py | 0 B |
torch/ao/pruning/_experimental/activation_sparsifier/activation_sparsifier.py | py | 19.03 KiB |
torch/ao/pruning/_experimental/data_scheduler | Folder | |
torch/ao/pruning/_experimental/data_scheduler/__init__.py | py | 98 B |
torch/ao/pruning/_experimental/data_scheduler/base_data_scheduler.py | py | 7.62 KiB |
torch/ao/pruning/_experimental/data_sparsifier | Folder | |
torch/ao/pruning/_experimental/data_sparsifier/__init__.py | py | 182 B |
torch/ao/pruning/_experimental/data_sparsifier/base_data_sparsifier.py | py | 13.44 KiB |
torch/ao/pruning/_experimental/data_sparsifier/data_norm_sparsifier.py | py | 7.79 KiB |
torch/ao/pruning/_experimental/data_sparsifier/lightning | Folder | |
torch/ao/pruning/_experimental/data_sparsifier/lightning/__init__.py | py | 0 B |
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks | Folder | |
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/__init__.py | py | 0 B |
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/_data_sparstity_utils.py | py | 1.64 KiB |
torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/data_sparsity.py | py | 6.64 KiB |
torch/ao/pruning/_experimental/data_sparsifier/quantization_utils.py | py | 5.91 KiB |
torch/ao/pruning/_experimental/pruner | Folder | |
torch/ao/pruning/_experimental/pruner/FPGM_pruner.py | py | 3.47 KiB |
torch/ao/pruning/_experimental/pruner/__init__.py | py | 265 B |
torch/ao/pruning/_experimental/pruner/base_structured_sparsifier.py | py | 11.00 KiB |
torch/ao/pruning/_experimental/pruner/lstm_saliency_pruner.py | py | 2.14 KiB |
torch/ao/pruning/_experimental/pruner/match_utils.py | py | 2.01 KiB |
torch/ao/pruning/_experimental/pruner/parametrization.py | py | 1.86 KiB |
torch/ao/pruning/_experimental/pruner/prune_functions.py | py | 19.05 KiB |
torch/ao/pruning/_experimental/pruner/saliency_pruner.py | py | 1.40 KiB |
torch/ao/pruning/_mappings.py | py | 620 B |
torch/ao/pruning/scheduler | Folder | |
torch/ao/pruning/scheduler/__init__.py | py | 0 B |
torch/ao/pruning/scheduler/base_scheduler.py | py | 6.54 KiB |
torch/ao/pruning/scheduler/cubic_scheduler.py | py | 3.86 KiB |
torch/ao/pruning/scheduler/lambda_scheduler.py | py | 2.12 KiB |
torch/ao/pruning/sparsifier | Folder | |
torch/ao/pruning/sparsifier/__init__.py | py | 0 B |
torch/ao/pruning/sparsifier/base_sparsifier.py | py | 13.73 KiB |
torch/ao/pruning/sparsifier/nearly_diagonal_sparsifier.py | py | 2.27 KiB |
torch/ao/pruning/sparsifier/utils.py | py | 4.84 KiB |
torch/ao/pruning/sparsifier/weight_norm_sparsifier.py | py | 9.33 KiB |
torch/ao/quantization | Folder | |
torch/ao/quantization/__init__.py | py | 7.08 KiB |
torch/ao/quantization/_correct_bias.py | py | 5.49 KiB |
torch/ao/quantization/_equalize.py | py | 9.50 KiB |
torch/ao/quantization/_learnable_fake_quantize.py | py | 7.92 KiB |
torch/ao/quantization/backend_config | Folder | |
torch/ao/quantization/backend_config/__init__.py | py | 945 B |
torch/ao/quantization/backend_config/_common_operator_config_utils.py | py | 27.78 KiB |
torch/ao/quantization/backend_config/_qnnpack_pt2e.py | py | 6.48 KiB |
torch/ao/quantization/backend_config/backend_config.py | py | 31.48 KiB |
torch/ao/quantization/backend_config/executorch.py | py | 17.04 KiB |
torch/ao/quantization/backend_config/fbgemm.py | py | 4.24 KiB |
torch/ao/quantization/backend_config/native.py | py | 8.27 KiB |
torch/ao/quantization/backend_config/observation_type.py | py | 0 B |
torch/ao/quantization/backend_config/onednn.py | py | 19.44 KiB |
torch/ao/quantization/backend_config/qnnpack.py | py | 5.44 KiB |
torch/ao/quantization/backend_config/tensorrt.py | py | 3.05 KiB |
torch/ao/quantization/backend_config/utils.py | py | 12.47 KiB |
torch/ao/quantization/backend_config/x86.py | py | 3.90 KiB |
torch/ao/quantization/fake_quantize.py | py | 22.87 KiB |
torch/ao/quantization/fuse_modules.py | py | 6.88 KiB |
torch/ao/quantization/fuser_method_mappings.py | py | 10.44 KiB |
torch/ao/quantization/fx | Folder | |
torch/ao/quantization/fx/__init__.py | py | 84 B |
torch/ao/quantization/fx/_decomposed.py | py | 41.62 KiB |
torch/ao/quantization/fx/_equalize.py | py | 37.88 KiB |
torch/ao/quantization/fx/_lower_to_native_backend.py | py | 53.23 KiB |
torch/ao/quantization/fx/_model_report | Folder | |
torch/ao/quantization/fx/_model_report/__init__.py | py | 0 B |
torch/ao/quantization/fx/_model_report/detector.py | py | 76.32 KiB |
torch/ao/quantization/fx/_model_report/model_report.py | py | 29.46 KiB |
torch/ao/quantization/fx/_model_report/model_report_observer.py | py | 12.05 KiB |
torch/ao/quantization/fx/_model_report/model_report_visualizer.py | py | 32.62 KiB |
torch/ao/quantization/fx/convert.py | py | 57.24 KiB |
torch/ao/quantization/fx/custom_config.py | py | 21.85 KiB |
torch/ao/quantization/fx/fuse.py | py | 7.29 KiB |
torch/ao/quantization/fx/fuse_handler.py | py | 4.67 KiB |
torch/ao/quantization/fx/graph_module.py | py | 6.67 KiB |
torch/ao/quantization/fx/lower_to_fbgemm.py | py | 576 B |
torch/ao/quantization/fx/lower_to_qnnpack.py | py | 579 B |
torch/ao/quantization/fx/lstm_utils.py | py | 9.24 KiB |
torch/ao/quantization/fx/match_utils.py | py | 8.88 KiB |
torch/ao/quantization/fx/pattern_utils.py | py | 3.69 KiB |
torch/ao/quantization/fx/prepare.py | py | 87.77 KiB |
torch/ao/quantization/fx/qconfig_mapping_utils.py | py | 15.42 KiB |
torch/ao/quantization/fx/quantize_handler.py | py | 7.35 KiB |
torch/ao/quantization/fx/tracer.py | py | 1.70 KiB |
torch/ao/quantization/fx/utils.py | py | 37.66 KiB |
torch/ao/quantization/observer.py | py | 66.92 KiB |
torch/ao/quantization/pt2e | Folder | |
torch/ao/quantization/pt2e/__init__.py | py | 0 B |
torch/ao/quantization/pt2e/_numeric_debugger.py | py | 10.20 KiB |
torch/ao/quantization/pt2e/duplicate_dq_pass.py | py | 3.13 KiB |
torch/ao/quantization/pt2e/export_utils.py | py | 8.00 KiB |
torch/ao/quantization/pt2e/graph_utils.py | py | 3.93 KiB |
torch/ao/quantization/pt2e/port_metadata_pass.py | py | 9.08 KiB |
torch/ao/quantization/pt2e/prepare.py | py | 21.60 KiB |
torch/ao/quantization/pt2e/qat_utils.py | py | 36.22 KiB |
torch/ao/quantization/pt2e/representation | Folder | |
torch/ao/quantization/pt2e/representation/__init__.py | py | 116 B |
torch/ao/quantization/pt2e/representation/rewrite.py | py | 27.82 KiB |
torch/ao/quantization/pt2e/utils.py | py | 23.85 KiB |
torch/ao/quantization/qconfig.py | py | 24.41 KiB |
torch/ao/quantization/qconfig_mapping.py | py | 14.85 KiB |
torch/ao/quantization/quant_type.py | py | 795 B |
torch/ao/quantization/quantization_mappings.py | py | 13.89 KiB |
torch/ao/quantization/quantize.py | py | 30.64 KiB |
torch/ao/quantization/quantize_fx.py | py | 32.23 KiB |
torch/ao/quantization/quantize_jit.py | py | 14.67 KiB |
torch/ao/quantization/quantize_pt2e.py | py | 9.51 KiB |
torch/ao/quantization/quantizer | Folder | |
torch/ao/quantization/quantizer/__init__.py | py | 477 B |
torch/ao/quantization/quantizer/composable_quantizer.py | py | 3.00 KiB |
torch/ao/quantization/quantizer/embedding_quantizer.py | py | 3.50 KiB |
torch/ao/quantization/quantizer/quantizer.py | py | 6.66 KiB |
torch/ao/quantization/quantizer/utils.py | py | 3.25 KiB |
torch/ao/quantization/quantizer/x86_inductor_quantizer.py | py | 63.92 KiB |
torch/ao/quantization/quantizer/xnnpack_quantizer.py | py | 15.79 KiB |
torch/ao/quantization/quantizer/xnnpack_quantizer_utils.py | py | 40.19 KiB |
torch/ao/quantization/stubs.py | py | 2.06 KiB |
torch/ao/quantization/utils.py | py | 28.29 KiB |
torch/autograd | Folder | |
torch/autograd/__init__.py | py | 25.09 KiB |
torch/autograd/_functions | Folder | |
torch/autograd/_functions/__init__.py | py | 37 B |
torch/autograd/_functions/tensor.py | py | 2.20 KiB |
torch/autograd/_functions/utils.py | py | 2.07 KiB |
torch/autograd/anomaly_mode.py | py | 4.95 KiB |
torch/autograd/forward_ad.py | py | 7.73 KiB |
torch/autograd/function.py | py | 33.32 KiB |
torch/autograd/functional.py | py | 52.22 KiB |
torch/autograd/grad_mode.py | py | 13.08 KiB |
torch/autograd/gradcheck.py | py | 90.75 KiB |
torch/autograd/graph.py | py | 30.06 KiB |
torch/autograd/profiler.py | py | 48.08 KiB |
torch/autograd/profiler_legacy.py | py | 11.58 KiB |
torch/autograd/profiler_util.py | py | 40.07 KiB |
torch/autograd/variable.py | py | 406 B |
torch/backends | Folder | |
torch/backends/__init__.py | py | 1.78 KiB |
torch/backends/_coreml | Folder | |
torch/backends/_coreml/__init__.py | py | 0 B |
torch/backends/_coreml/preprocess.py | py | 4.29 KiB |
torch/backends/_nnapi | Folder | |
torch/backends/_nnapi/__init__.py | py | 0 B |
torch/backends/_nnapi/prepare.py | py | 6.61 KiB |
torch/backends/_nnapi/serializer.py | py | 83.14 KiB |
torch/backends/cpu | Folder | |
torch/backends/cpu/__init__.py | py | 335 B |
torch/backends/cuda | Folder | |
torch/backends/cuda/__init__.py | py | 16.65 KiB |
torch/backends/cudnn | Folder | |
torch/backends/cudnn/__init__.py | py | 6.65 KiB |
torch/backends/cudnn/rnn.py | py | 2.08 KiB |
torch/backends/cusparselt | Folder | |
torch/backends/cusparselt/__init__.py | py | 1.27 KiB |
torch/backends/mha | Folder | |
torch/backends/mha/__init__.py | py | 741 B |
torch/backends/mkl | Folder | |
torch/backends/mkl/__init__.py | py | 1.80 KiB |
torch/backends/mkldnn | Folder | |
torch/backends/mkldnn/__init__.py | py | 3.18 KiB |
torch/backends/mps | Folder | |
torch/backends/mps/__init__.py | py | 1.66 KiB |
torch/backends/nnpack | Folder | |
torch/backends/nnpack/__init__.py | py | 869 B |
torch/backends/openmp | Folder | |
torch/backends/openmp/__init__.py | py | 164 B |
torch/backends/opt_einsum | Folder | |
torch/backends/opt_einsum/__init__.py | py | 3.91 KiB |
torch/backends/quantized | Folder | |
torch/backends/quantized/__init__.py | py | 1.91 KiB |
torch/backends/xeon | Folder | |
torch/backends/xeon/__init__.py | py | 0 B |
torch/backends/xeon/run_cpu.py | py | 37.61 KiB |
torch/backends/xnnpack | Folder | |
torch/backends/xnnpack/__init__.py | py | 731 B |
torch/bin | Folder | |
torch/bin/FileStoreTest.exe | exe | 372.50 KiB |
torch/bin/TCPStoreTest.exe | exe | 397.00 KiB |
torch/bin/asmjit.dll | dll | 353.50 KiB |
torch/bin/fbgemm.dll | dll | 4.91 MiB |
torch/bin/protoc.exe | exe | 2.67 MiB |
torch/bin/test_api.exe | exe | 8.55 MiB |
torch/bin/test_edge_op_registration.exe | exe | 348.00 KiB |
torch/bin/test_jit.exe | exe | 4.97 MiB |
torch/bin/test_lazy.exe | exe | 2.22 MiB |
torch/bin/test_tensorexpr.exe | exe | 5.65 MiB |
torch/bin/tutorial_tensorexpr.exe | exe | 145.50 KiB |
torch/compiler | Folder | |
torch/compiler/__init__.py | py | 15.00 KiB |
torch/compiler/config.py | py | 2.50 KiB |
torch/contrib | Folder | |
torch/contrib/__init__.py | py | 0 B |
torch/contrib/_tensorboard_vis.py | py | 5.93 KiB |
torch/cpu | Folder | |
torch/cpu/__init__.py | py | 4.61 KiB |
torch/cpu/amp | Folder | |
torch/cpu/amp/__init__.py | py | 74 B |
torch/cpu/amp/autocast_mode.py | py | 1.54 KiB |
torch/cpu/amp/grad_scaler.py | py | 993 B |
torch/cuda | Folder | |
torch/cuda/__init__.py | py | 56.38 KiB |
torch/cuda/_gpu_trace.py | py | 2.40 KiB |
torch/cuda/_memory_viz.py | py | 24.82 KiB |
torch/cuda/_sanitizer.py | py | 24.27 KiB |
torch/cuda/_utils.py | py | 1.60 KiB |
torch/cuda/amp | Folder | |
torch/cuda/amp/__init__.py | py | 279 B |
torch/cuda/amp/autocast_mode.py | py | 2.84 KiB |
torch/cuda/amp/common.py | py | 241 B |
torch/cuda/amp/grad_scaler.py | py | 1.08 KiB |
torch/cuda/comm.py | py | 363 B |
torch/cuda/error.py | py | 0 B |
torch/cuda/gds.py | py | 4.46 KiB |
torch/cuda/graphs.py | py | 22.09 KiB |
torch/cuda/jiterator.py | py | 6.86 KiB |
torch/cuda/memory.py | py | 41.08 KiB |
torch/cuda/nccl.py | py | 4.59 KiB |
torch/cuda/nvtx.py | py | 2.48 KiB |
torch/cuda/profiler.py | py | 2.43 KiB |
torch/cuda/random.py | py | 5.31 KiB |
torch/cuda/sparse.py | py | 68 B |
torch/cuda/streams.py | py | 8.38 KiB |
torch/cuda/tunable.py | py | 13.49 KiB |
torch/distributed | Folder | |
torch/distributed/__init__.py | py | 5.00 KiB |
torch/distributed/_checkpointable.py | py | 1.29 KiB |
torch/distributed/_composable | Folder | |
torch/distributed/_composable/__init__.py | py | 166 B |
torch/distributed/_composable/checkpoint_activation.py | py | 4.57 KiB |
torch/distributed/_composable/contract.py | py | 9.54 KiB |
torch/distributed/_composable/fsdp | Folder | |
torch/distributed/_composable/fsdp/__init__.py | py | 158 B |
torch/distributed/_composable/fsdp/_fsdp_api.py | py | 3.63 KiB |
torch/distributed/_composable/fsdp/_fsdp_collectives.py | py | 22.51 KiB |
torch/distributed/_composable/fsdp/_fsdp_common.py | py | 5.79 KiB |
torch/distributed/_composable/fsdp/_fsdp_init.py | py | 6.76 KiB |
torch/distributed/_composable/fsdp/_fsdp_param.py | py | 41.15 KiB |
torch/distributed/_composable/fsdp/_fsdp_param_group.py | py | 33.00 KiB |
torch/distributed/_composable/fsdp/_fsdp_state.py | py | 17.10 KiB |
torch/distributed/_composable/fsdp/fully_shard.py | py | 21.95 KiB |
torch/distributed/_composable/fully_shard.py | py | 5.33 KiB |
torch/distributed/_composable/replicate.py | py | 9.31 KiB |
torch/distributed/_composable_state.py | py | 1.40 KiB |
torch/distributed/_functional_collectives.py | py | 44.31 KiB |
torch/distributed/_functional_collectives_impl.py | py | 3.27 KiB |
torch/distributed/_shard | Folder | |
torch/distributed/_shard/__init__.py | py | 88 B |
torch/distributed/_shard/_utils.py | py | 1.07 KiB |
torch/distributed/_shard/api.py | py | 12.40 KiB |
torch/distributed/_shard/checkpoint | Folder | |
torch/distributed/_shard/checkpoint/__init__.py | py | 603 B |
torch/distributed/_shard/common_op_utils.py | py | 2.19 KiB |
torch/distributed/_shard/metadata.py | py | 2.23 KiB |
torch/distributed/_shard/op_registry_utils.py | py | 1.05 KiB |
torch/distributed/_shard/sharded_optim | Folder | |
torch/distributed/_shard/sharded_optim/__init__.py | py | 1.86 KiB |
torch/distributed/_shard/sharded_optim/api.py | py | 4.26 KiB |
torch/distributed/_shard/sharded_tensor | Folder | |
torch/distributed/_shard/sharded_tensor/__init__.py | py | 19.29 KiB |
torch/distributed/_shard/sharded_tensor/_ops | Folder | |
torch/distributed/_shard/sharded_tensor/_ops/__init__.py | py | 511 B |
torch/distributed/_shard/sharded_tensor/_ops/_common.py | py | 4.21 KiB |
torch/distributed/_shard/sharded_tensor/_ops/binary_cmp.py | py | 2.75 KiB |
torch/distributed/_shard/sharded_tensor/_ops/init.py | py | 5.50 KiB |
torch/distributed/_shard/sharded_tensor/_ops/misc_ops.py | py | 509 B |
torch/distributed/_shard/sharded_tensor/_ops/tensor_ops.py | py | 7.74 KiB |
torch/distributed/_shard/sharded_tensor/api.py | py | 52.19 KiB |
torch/distributed/_shard/sharded_tensor/logger.py | py | 1.12 KiB |
torch/distributed/_shard/sharded_tensor/logging_handlers.py | py | 406 B |
torch/distributed/_shard/sharded_tensor/metadata.py | py | 3.04 KiB |
torch/distributed/_shard/sharded_tensor/reshard.py | py | 10.77 KiB |
torch/distributed/_shard/sharded_tensor/shard.py | py | 2.40 KiB |
torch/distributed/_shard/sharded_tensor/utils.py | py | 9.53 KiB |
torch/distributed/_shard/sharder.py | py | 930 B |
torch/distributed/_shard/sharding_plan | Folder | |
torch/distributed/_shard/sharding_plan/__init__.py | py | 48 B |
torch/distributed/_shard/sharding_plan/api.py | py | 3.66 KiB |
torch/distributed/_shard/sharding_spec | Folder | |
torch/distributed/_shard/sharding_spec/__init__.py | py | 301 B |
torch/distributed/_shard/sharding_spec/_internals.py | py | 8.02 KiB |
torch/distributed/_shard/sharding_spec/api.py | py | 9.86 KiB |
torch/distributed/_shard/sharding_spec/chunk_sharding_spec.py | py | 9.19 KiB |
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops | Folder | |
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/__init__.py | py | 0 B |
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/_common.py | py | 13.06 KiB |
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/embedding.py | py | 11.23 KiB |
torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/embedding_bag.py | py | 18.36 KiB |
torch/distributed/_sharded_tensor | Folder | |
torch/distributed/_sharded_tensor/__init__.py | py | 636 B |
torch/distributed/_sharding_spec | Folder | |
torch/distributed/_sharding_spec/__init__.py | py | 668 B |
torch/distributed/_state_dict_utils.py | py | 26.57 KiB |
torch/distributed/_symmetric_memory | Folder | |
torch/distributed/_symmetric_memory/__init__.py | py | 52.92 KiB |
torch/distributed/_tensor | Folder | |
torch/distributed/_tensor/__init__.py | py | 1012 B |
torch/distributed/_tensor/api.py | py | 309 B |
torch/distributed/_tensor/placement_types.py | py | 394 B |
torch/distributed/_tools | Folder | |
torch/distributed/_tools/__init__.py | py | 339 B |
torch/distributed/_tools/fsdp2_mem_tracker.py | py | 26.00 KiB |
torch/distributed/_tools/fsdp_ilp.py | py | 15.18 KiB |
torch/distributed/_tools/ilp_utils.py | py | 10.14 KiB |
torch/distributed/_tools/mem_tracker.py | py | 42.04 KiB |
torch/distributed/_tools/memory_tracker.py | py | 11.51 KiB |
torch/distributed/_tools/mod_tracker.py | py | 9.55 KiB |
torch/distributed/_tools/runtime_estimator.py | py | 21.19 KiB |
torch/distributed/_tools/sac_estimator.py | py | 43.51 KiB |
torch/distributed/_tools/sac_ilp.py | py | 11.36 KiB |
torch/distributed/algorithms | Folder | |
torch/distributed/algorithms/__init__.py | py | 44 B |
torch/distributed/algorithms/_checkpoint | Folder | |
torch/distributed/algorithms/_checkpoint/__init__.py | py | 0 B |
torch/distributed/algorithms/_checkpoint/checkpoint_wrapper.py | py | 12.31 KiB |
torch/distributed/algorithms/_comm_hooks | Folder | |
torch/distributed/algorithms/_comm_hooks/__init__.py | py | 138 B |
torch/distributed/algorithms/_comm_hooks/default_hooks.py | py | 7.66 KiB |
torch/distributed/algorithms/_optimizer_overlap | Folder | |
torch/distributed/algorithms/_optimizer_overlap/__init__.py | py | 53 B |
torch/distributed/algorithms/_optimizer_overlap/optimizer_overlap.py | py | 3.79 KiB |
torch/distributed/algorithms/_quantization | Folder | |
torch/distributed/algorithms/_quantization/__init__.py | py | 0 B |
torch/distributed/algorithms/_quantization/quantization.py | py | 5.63 KiB |
torch/distributed/algorithms/ddp_comm_hooks | Folder | |
torch/distributed/algorithms/ddp_comm_hooks/__init__.py | py | 3.62 KiB |
torch/distributed/algorithms/ddp_comm_hooks/ddp_zero_hook.py | py | 19.71 KiB |
torch/distributed/algorithms/ddp_comm_hooks/debugging_hooks.py | py | 1.12 KiB |
torch/distributed/algorithms/ddp_comm_hooks/default_hooks.py | py | 7.80 KiB |
torch/distributed/algorithms/ddp_comm_hooks/mixed_precision_hooks.py | py | 3.29 KiB |
torch/distributed/algorithms/ddp_comm_hooks/optimizer_overlap_hooks.py | py | 6.18 KiB |
torch/distributed/algorithms/ddp_comm_hooks/post_localSGD_hook.py | py | 5.13 KiB |
torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.py | py | 40.23 KiB |
torch/distributed/algorithms/ddp_comm_hooks/quantization_hooks.py | py | 8.25 KiB |
torch/distributed/algorithms/join.py | py | 13.43 KiB |
torch/distributed/algorithms/model_averaging | Folder | |
torch/distributed/algorithms/model_averaging/__init__.py | py | 0 B |
torch/distributed/algorithms/model_averaging/averagers.py | py | 5.23 KiB |
torch/distributed/algorithms/model_averaging/hierarchical_model_averager.py | py | 9.72 KiB |
torch/distributed/algorithms/model_averaging/utils.py | py | 3.15 KiB |
torch/distributed/argparse_util.py | py | 3.91 KiB |
torch/distributed/autograd | Folder | |
torch/distributed/autograd/__init__.py | py | 1.66 KiB |
torch/distributed/c10d_logger.py | py | 3.17 KiB |
torch/distributed/checkpoint | Folder | |
torch/distributed/checkpoint/__init__.py | py | 608 B |
torch/distributed/checkpoint/_checkpointer.py | py | 3.68 KiB |
torch/distributed/checkpoint/_dedup_save_plans.py | py | 2.34 KiB |
torch/distributed/checkpoint/_dedup_tensors.py | py | 2.02 KiB |
torch/distributed/checkpoint/_fsspec_filesystem.py | py | 4.28 KiB |
torch/distributed/checkpoint/_nested_dict.py | py | 2.32 KiB |
torch/distributed/checkpoint/_sharded_tensor_utils.py | py | 4.15 KiB |
torch/distributed/checkpoint/_storage_utils.py | py | 1.44 KiB |
torch/distributed/checkpoint/_traverse.py | py | 6.95 KiB |
torch/distributed/checkpoint/_version.py | py | 128 B |
torch/distributed/checkpoint/api.py | py | 1.46 KiB |
torch/distributed/checkpoint/default_planner.py | py | 20.26 KiB |
torch/distributed/checkpoint/filesystem.py | py | 26.05 KiB |
torch/distributed/checkpoint/format_utils.py | py | 10.28 KiB |
torch/distributed/checkpoint/logger.py | py | 3.24 KiB |
torch/distributed/checkpoint/logging_handlers.py | py | 259 B |
torch/distributed/checkpoint/metadata.py | py | 5.58 KiB |
torch/distributed/checkpoint/optimizer.py | py | 13.18 KiB |
torch/distributed/checkpoint/planner.py | py | 14.97 KiB |
torch/distributed/checkpoint/planner_helpers.py | py | 13.07 KiB |
torch/distributed/checkpoint/resharding.py | py | 2.34 KiB |
torch/distributed/checkpoint/staging.py | py | 4.90 KiB |
torch/distributed/checkpoint/state_dict.py | py | 52.49 KiB |
torch/distributed/checkpoint/state_dict_loader.py | py | 12.09 KiB |
torch/distributed/checkpoint/state_dict_saver.py | py | 12.76 KiB |
torch/distributed/checkpoint/stateful.py | py | 1.08 KiB |
torch/distributed/checkpoint/storage.py | py | 9.78 KiB |
torch/distributed/checkpoint/utils.py | py | 14.53 KiB |
torch/distributed/collective_utils.py | py | 7.33 KiB |
torch/distributed/constants.py | py | 1.23 KiB |
torch/distributed/device_mesh.py | py | 47.04 KiB |
torch/distributed/distributed_c10d.py | py | 207.49 KiB |
torch/distributed/elastic | Folder | |
torch/distributed/elastic/__init__.py | py | 3.64 KiB |
torch/distributed/elastic/agent | Folder | |
torch/distributed/elastic/agent/__init__.py | py | 0 B |
torch/distributed/elastic/agent/server | Folder | |
torch/distributed/elastic/agent/server/__init__.py | py | 1.41 KiB |
torch/distributed/elastic/agent/server/api.py | py | 37.03 KiB |
torch/distributed/elastic/agent/server/health_check_server.py | py | 1.70 KiB |
torch/distributed/elastic/agent/server/local_elastic_agent.py | py | 16.77 KiB |
torch/distributed/elastic/control_plane.py | py | 1.19 KiB |
torch/distributed/elastic/events | Folder | |
torch/distributed/elastic/events/__init__.py | py | 5.41 KiB |
torch/distributed/elastic/events/api.py | py | 3.29 KiB |
torch/distributed/elastic/events/handlers.py | py | 602 B |
torch/distributed/elastic/metrics | Folder | |
torch/distributed/elastic/metrics/__init__.py | py | 4.93 KiB |
torch/distributed/elastic/metrics/api.py | py | 5.76 KiB |
torch/distributed/elastic/multiprocessing | Folder | |
torch/distributed/elastic/multiprocessing/__init__.py | py | 7.47 KiB |
torch/distributed/elastic/multiprocessing/api.py | py | 33.86 KiB |
torch/distributed/elastic/multiprocessing/errors | Folder | |
torch/distributed/elastic/multiprocessing/errors/__init__.py | py | 14.39 KiB |
torch/distributed/elastic/multiprocessing/errors/error_handler.py | py | 6.62 KiB |
torch/distributed/elastic/multiprocessing/errors/handlers.py | py | 494 B |
torch/distributed/elastic/multiprocessing/redirects.py | py | 2.80 KiB |
torch/distributed/elastic/multiprocessing/subprocess_handler | Folder | |
torch/distributed/elastic/multiprocessing/subprocess_handler/__init__.py | py | 539 B |
torch/distributed/elastic/multiprocessing/subprocess_handler/handlers.py | py | 821 B |
torch/distributed/elastic/multiprocessing/subprocess_handler/subprocess_handler.py | py | 2.47 KiB |
torch/distributed/elastic/multiprocessing/tail_log.py | py | 5.00 KiB |
torch/distributed/elastic/rendezvous | Folder | |
torch/distributed/elastic/rendezvous/__init__.py | py | 6.32 KiB |
torch/distributed/elastic/rendezvous/api.py | py | 13.07 KiB |
torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.py | py | 10.92 KiB |
torch/distributed/elastic/rendezvous/dynamic_rendezvous.py | py | 48.37 KiB |
torch/distributed/elastic/rendezvous/etcd_rendezvous.py | py | 43.49 KiB |
torch/distributed/elastic/rendezvous/etcd_rendezvous_backend.py | py | 7.47 KiB |
torch/distributed/elastic/rendezvous/etcd_server.py | py | 8.48 KiB |
torch/distributed/elastic/rendezvous/etcd_store.py | py | 7.22 KiB |
torch/distributed/elastic/rendezvous/registry.py | py | 3.02 KiB |
torch/distributed/elastic/rendezvous/static_tcp_rendezvous.py | py | 3.70 KiB |
torch/distributed/elastic/rendezvous/utils.py | py | 8.48 KiB |
torch/distributed/elastic/timer | Folder | |
torch/distributed/elastic/timer/__init__.py | py | 1.76 KiB |
torch/distributed/elastic/timer/api.py | py | 9.66 KiB |
torch/distributed/elastic/timer/debug_info_logging.py | py | 665 B |
torch/distributed/elastic/timer/file_based_local_timer.py | py | 15.11 KiB |
torch/distributed/elastic/timer/local_timer.py | py | 4.33 KiB |
torch/distributed/elastic/utils | Folder | |
torch/distributed/elastic/utils/__init__.py | py | 327 B |
torch/distributed/elastic/utils/api.py | py | 1.73 KiB |
torch/distributed/elastic/utils/data | Folder | |
torch/distributed/elastic/utils/data/__init__.py | py | 382 B |
torch/distributed/elastic/utils/data/cycling_iterator.py | py | 1.41 KiB |
torch/distributed/elastic/utils/data/elastic_distributed_sampler.py | py | 2.49 KiB |
torch/distributed/elastic/utils/distributed.py | py | 5.96 KiB |
torch/distributed/elastic/utils/log_level.py | py | 353 B |
torch/distributed/elastic/utils/logging.py | py | 2.28 KiB |
torch/distributed/elastic/utils/store.py | py | 7.30 KiB |
torch/distributed/fsdp | Folder | |
torch/distributed/fsdp/__init__.py | py | 978 B |
torch/distributed/fsdp/_common_utils.py | py | 22.48 KiB |
torch/distributed/fsdp/_debug_utils.py | py | 5.73 KiB |
torch/distributed/fsdp/_dynamo_utils.py | py | 2.66 KiB |
torch/distributed/fsdp/_exec_order_utils.py | py | 16.10 KiB |
torch/distributed/fsdp/_flat_param.py | py | 123.44 KiB |
torch/distributed/fsdp/_fsdp_extensions.py | py | 5.03 KiB |
torch/distributed/fsdp/_init_utils.py | py | 46.36 KiB |
torch/distributed/fsdp/_limiter_utils.py | py | 1.09 KiB |
torch/distributed/fsdp/_optim_utils.py | py | 86.87 KiB |
torch/distributed/fsdp/_runtime_utils.py | py | 66.25 KiB |
torch/distributed/fsdp/_shard_utils.py | py | 4.65 KiB |
torch/distributed/fsdp/_state_dict_utils.py | py | 34.47 KiB |
torch/distributed/fsdp/_trace_utils.py | py | 10.76 KiB |
torch/distributed/fsdp/_traversal_utils.py | py | 4.64 KiB |
torch/distributed/fsdp/_unshard_param_utils.py | py | 11.56 KiB |
torch/distributed/fsdp/_wrap_utils.py | py | 10.92 KiB |
torch/distributed/fsdp/api.py | py | 18.84 KiB |
torch/distributed/fsdp/fully_sharded_data_parallel.py | py | 100.08 KiB |
torch/distributed/fsdp/sharded_grad_scaler.py | py | 16.12 KiB |
torch/distributed/fsdp/wrap.py | py | 22.70 KiB |
torch/distributed/launch.py | py | 7.63 KiB |
torch/distributed/launcher | Folder | |
torch/distributed/launcher/__init__.py | py | 363 B |
torch/distributed/launcher/api.py | py | 11.01 KiB |
torch/distributed/logging_handlers.py | py | 406 B |
torch/distributed/nn | Folder | |
torch/distributed/nn/__init__.py | py | 152 B |
torch/distributed/nn/api | Folder | |
torch/distributed/nn/api/__init__.py | py | 0 B |
torch/distributed/nn/api/remote_module.py | py | 31.31 KiB |
torch/distributed/nn/functional.py | py | 15.26 KiB |
torch/distributed/nn/jit | Folder | |
torch/distributed/nn/jit/__init__.py | py | 0 B |
torch/distributed/nn/jit/instantiator.py | py | 5.48 KiB |
torch/distributed/nn/jit/templates | Folder | |
torch/distributed/nn/jit/templates/__init__.py | py | 0 B |
torch/distributed/nn/jit/templates/remote_module_template.py | py | 3.49 KiB |
torch/distributed/optim | Folder | |
torch/distributed/optim/__init__.py | py | 1.45 KiB |
torch/distributed/optim/_deprecation_warning.py | py | 563 B |
torch/distributed/optim/apply_optimizer_in_backward.py | py | 5.20 KiB |
torch/distributed/optim/functional_adadelta.py | py | 3.98 KiB |
torch/distributed/optim/functional_adagrad.py | py | 4.33 KiB |
torch/distributed/optim/functional_adam.py | py | 7.47 KiB |
torch/distributed/optim/functional_adamax.py | py | 4.68 KiB |
torch/distributed/optim/functional_adamw.py | py | 7.58 KiB |
torch/distributed/optim/functional_rmsprop.py | py | 4.71 KiB |
torch/distributed/optim/functional_rprop.py | py | 3.87 KiB |
torch/distributed/optim/functional_sgd.py | py | 5.97 KiB |
torch/distributed/optim/named_optimizer.py | py | 13.99 KiB |
torch/distributed/optim/optimizer.py | py | 9.83 KiB |
torch/distributed/optim/post_localSGD_optimizer.py | py | 4.44 KiB |
torch/distributed/optim/utils.py | py | 2.28 KiB |
torch/distributed/optim/zero_redundancy_optimizer.py | py | 71.82 KiB |
torch/distributed/optim/zero_redundancy_optimizer.pyi | pyi | 2.84 KiB |
torch/distributed/pipelining | Folder | |
torch/distributed/pipelining/_IR.py | py | 48.70 KiB |
torch/distributed/pipelining/__init__.py | py | 611 B |
torch/distributed/pipelining/_backward.py | py | 15.70 KiB |
torch/distributed/pipelining/_debug.py | py | 578 B |
torch/distributed/pipelining/_unflatten.py | py | 936 B |
torch/distributed/pipelining/_utils.py | py | 2.65 KiB |
torch/distributed/pipelining/microbatch.py | py | 16.28 KiB |
torch/distributed/pipelining/schedules.py | py | 97.31 KiB |
torch/distributed/pipelining/stage.py | py | 60.84 KiB |
torch/distributed/remote_device.py | py | 4.61 KiB |
torch/distributed/rendezvous.py | py | 10.02 KiB |
torch/distributed/rpc | Folder | |
torch/distributed/rpc/__init__.py | py | 9.72 KiB |
torch/distributed/rpc/_testing | Folder | |
torch/distributed/rpc/_testing/__init__.py | py | 519 B |
torch/distributed/rpc/_testing/faulty_agent_backend_registry.py | py | 1.66 KiB |
torch/distributed/rpc/_utils.py | py | 1.65 KiB |
torch/distributed/rpc/api.py | py | 37.05 KiB |
torch/distributed/rpc/backend_registry.py | py | 16.35 KiB |
torch/distributed/rpc/constants.py | py | 853 B |
torch/distributed/rpc/functions.py | py | 7.27 KiB |
torch/distributed/rpc/internal.py | py | 11.13 KiB |
torch/distributed/rpc/options.py | py | 7.08 KiB |
torch/distributed/rpc/rref_proxy.py | py | 2.69 KiB |
torch/distributed/rpc/server_process_global_profiler.py | py | 8.38 KiB |
torch/distributed/run.py | py | 34.40 KiB |
torch/distributed/tensor | Folder | |
torch/distributed/tensor/__init__.py | py | 2.17 KiB |
torch/distributed/tensor/_api.py | py | 52.94 KiB |
torch/distributed/tensor/_collective_utils.py | py | 13.56 KiB |
torch/distributed/tensor/_dispatch.py | py | 21.51 KiB |
torch/distributed/tensor/_dtensor_spec.py | py | 10.33 KiB |
torch/distributed/tensor/_op_schema.py | py | 17.20 KiB |
torch/distributed/tensor/_ops | Folder | |
torch/distributed/tensor/_ops/__init__.py | py | 437 B |
torch/distributed/tensor/_ops/_common_rules.py | py | 12.07 KiB |
torch/distributed/tensor/_ops/_conv_ops.py | py | 3.34 KiB |
torch/distributed/tensor/_ops/_einsum_strategy.py | py | 6.46 KiB |
torch/distributed/tensor/_ops/_embedding_ops.py | py | 10.20 KiB |
torch/distributed/tensor/_ops/_experimental_ops.py | py | 1016 B |
torch/distributed/tensor/_ops/_math_ops.py | py | 40.67 KiB |
torch/distributed/tensor/_ops/_matrix_ops.py | py | 17.21 KiB |
torch/distributed/tensor/_ops/_pointwise_ops.py | py | 20.92 KiB |
torch/distributed/tensor/_ops/_random_ops.py | py | 1.22 KiB |
torch/distributed/tensor/_ops/_tensor_ops.py | py | 30.16 KiB |
torch/distributed/tensor/_ops/_view_ops.py | py | 22.94 KiB |
torch/distributed/tensor/_ops/utils.py | py | 10.28 KiB |
torch/distributed/tensor/_random.py | py | 15.88 KiB |
torch/distributed/tensor/_redistribute.py | py | 14.78 KiB |
torch/distributed/tensor/_sharding_prop.py | py | 21.77 KiB |
torch/distributed/tensor/_shards_wrapper.py | py | 11.67 KiB |
torch/distributed/tensor/_tp_conv.py | py | 10.15 KiB |
torch/distributed/tensor/_utils.py | py | 12.80 KiB |
torch/distributed/tensor/debug | Folder | |
torch/distributed/tensor/debug/__init__.py | py | 874 B |
torch/distributed/tensor/debug/_comm_mode.py | py | 28.81 KiB |
torch/distributed/tensor/debug/_op_coverage.py | py | 3.19 KiB |
torch/distributed/tensor/debug/_visualize_sharding.py | py | 6.52 KiB |
torch/distributed/tensor/device_mesh.py | py | 199 B |
torch/distributed/tensor/experimental | Folder | |
torch/distributed/tensor/experimental/__init__.py | py | 1.40 KiB |
torch/distributed/tensor/experimental/_attention.py | py | 45.25 KiB |
torch/distributed/tensor/experimental/_func_map.py | py | 10.91 KiB |
torch/distributed/tensor/experimental/_register_sharding.py | py | 5.67 KiB |
torch/distributed/tensor/experimental/_tp_transform.py | py | 20.46 KiB |
torch/distributed/tensor/parallel | Folder | |
torch/distributed/tensor/parallel/__init__.py | py | 604 B |
torch/distributed/tensor/parallel/_data_parallel_utils.py | py | 1.53 KiB |
torch/distributed/tensor/parallel/_utils.py | py | 2.33 KiB |
torch/distributed/tensor/parallel/api.py | py | 5.80 KiB |
torch/distributed/tensor/parallel/ddp.py | py | 3.77 KiB |
torch/distributed/tensor/parallel/fsdp.py | py | 13.71 KiB |
torch/distributed/tensor/parallel/input_reshard.py | py | 3.79 KiB |
torch/distributed/tensor/parallel/loss.py | py | 17.75 KiB |
torch/distributed/tensor/parallel/style.py | py | 28.46 KiB |
torch/distributed/tensor/placement_types.py | py | 25.57 KiB |
torch/distributed/utils.py | py | 13.53 KiB |
torch/distributions | Folder | |
torch/distributions/__init__.py | py | 6.06 KiB |
torch/distributions/bernoulli.py | py | 4.23 KiB |
torch/distributions/beta.py | py | 3.77 KiB |
torch/distributions/binomial.py | py | 6.03 KiB |
torch/distributions/categorical.py | py | 5.82 KiB |
torch/distributions/cauchy.py | py | 3.03 KiB |
torch/distributions/chi2.py | py | 1.01 KiB |
torch/distributions/constraint_registry.py | py | 10.37 KiB |
torch/distributions/constraints.py | py | 19.61 KiB |
torch/distributions/continuous_bernoulli.py | py | 8.80 KiB |
torch/distributions/dirichlet.py | py | 4.28 KiB |
torch/distributions/distribution.py | py | 12.42 KiB |
torch/distributions/exp_family.py | py | 2.39 KiB |
torch/distributions/exponential.py | py | 2.55 KiB |
torch/distributions/fishersnedecor.py | py | 3.53 KiB |
torch/distributions/gamma.py | py | 3.73 KiB |
torch/distributions/geometric.py | py | 4.70 KiB |
torch/distributions/gumbel.py | py | 2.81 KiB |
torch/distributions/half_cauchy.py | py | 2.45 KiB |
torch/distributions/half_normal.py | py | 2.21 KiB |
torch/distributions/independent.py | py | 4.70 KiB |
torch/distributions/inverse_gamma.py | py | 2.53 KiB |
torch/distributions/kl.py | py | 31.94 KiB |
torch/distributions/kumaraswamy.py | py | 3.48 KiB |
torch/distributions/laplace.py | py | 3.33 KiB |
torch/distributions/lkj_cholesky.py | py | 6.39 KiB |
torch/distributions/log_normal.py | py | 1.97 KiB |
torch/distributions/logistic_normal.py | py | 1.99 KiB |
torch/distributions/lowrank_multivariate_normal.py | py | 9.87 KiB |
torch/distributions/mixture_same_family.py | py | 8.50 KiB |
torch/distributions/multinomial.py | py | 5.43 KiB |
torch/distributions/multivariate_normal.py | py | 10.83 KiB |
torch/distributions/negative_binomial.py | py | 4.72 KiB |
torch/distributions/normal.py | py | 3.66 KiB |
torch/distributions/one_hot_categorical.py | py | 4.78 KiB |
torch/distributions/pareto.py | py | 2.21 KiB |
torch/distributions/poisson.py | py | 2.27 KiB |
torch/distributions/relaxed_bernoulli.py | py | 5.47 KiB |
torch/distributions/relaxed_categorical.py | py | 5.26 KiB |
torch/distributions/studentT.py | py | 3.96 KiB |
torch/distributions/transformed_distribution.py | py | 8.70 KiB |
torch/distributions/transforms.py | py | 41.22 KiB |
torch/distributions/uniform.py | py | 3.34 KiB |
torch/distributions/utils.py | py | 7.12 KiB |
torch/distributions/von_mises.py | py | 6.17 KiB |
torch/distributions/weibull.py | py | 3.11 KiB |
torch/distributions/wishart.py | py | 13.65 KiB |
torch/export | Folder | |
torch/export/__init__.py | py | 24.88 KiB |
torch/export/_draft_export.py | py | 12.35 KiB |
torch/export/_remove_auto_functionalized_pass.py | py | 1.96 KiB |
torch/export/_remove_effect_tokens_pass.py | py | 5.96 KiB |
torch/export/_safeguard.py | py | 1.95 KiB |
torch/export/_swap.py | py | 17.58 KiB |
torch/export/_trace.py | py | 78.17 KiB |
torch/export/_tree_utils.py | py | 2.28 KiB |
torch/export/_unlift.py | py | 15.10 KiB |
torch/export/custom_obj.py | py | 312 B |
torch/export/decomp_utils.py | py | 5.11 KiB |
torch/export/dynamic_shapes.py | py | 45.25 KiB |
torch/export/experimental | Folder | |
torch/export/experimental/__init__.py | py | 2.51 KiB |
torch/export/exported_program.py | py | 56.94 KiB |
torch/export/graph_signature.py | py | 21.98 KiB |
torch/export/passes | Folder | |
torch/export/passes/__init__.py | py | 2.28 KiB |
torch/export/unflatten.py | py | 63.88 KiB |
torch/fft | Folder | |
torch/fft/__init__.py | py | 55.10 KiB |
torch/func | Folder | |
torch/func/__init__.py | py | 410 B |
torch/functional.py | py | 87.56 KiB |
torch/futures | Folder | |
torch/futures/__init__.py | py | 14.39 KiB |
torch/fx | Folder | |
torch/fx/__init__.py | py | 4.14 KiB |
torch/fx/_compatibility.py | py | 1.10 KiB |
torch/fx/_lazy_graph_module.py | py | 7.17 KiB |
torch/fx/_pytree.py | py | 3.52 KiB |
torch/fx/_symbolic_trace.py | py | 49.97 KiB |
torch/fx/_utils.py | py | 1.66 KiB |
torch/fx/annotate.py | py | 1.26 KiB |
torch/fx/config.py | py | 334 B |
torch/fx/experimental | Folder | |
torch/fx/experimental/__init__.py | py | 0 B |
torch/fx/experimental/_backward_state.py | py | 994 B |
torch/fx/experimental/_config.py | py | 3.76 KiB |
torch/fx/experimental/accelerator_partitioner.py | py | 47.73 KiB |
torch/fx/experimental/const_fold.py | py | 12.37 KiB |
torch/fx/experimental/debug.py | py | 775 B |
torch/fx/experimental/graph_gradual_typechecker.py | py | 33.02 KiB |
torch/fx/experimental/merge_matmul.py | py | 6.13 KiB |
torch/fx/experimental/meta_tracer.py | py | 10.61 KiB |
torch/fx/experimental/migrate_gradual_types | Folder | |
torch/fx/experimental/migrate_gradual_types/__init__.py | py | 0 B |
torch/fx/experimental/migrate_gradual_types/constraint.py | py | 17.45 KiB |
torch/fx/experimental/migrate_gradual_types/constraint_generator.py | py | 49.86 KiB |
torch/fx/experimental/migrate_gradual_types/constraint_transformation.py | py | 41.34 KiB |
torch/fx/experimental/migrate_gradual_types/operation.py | py | 301 B |
torch/fx/experimental/migrate_gradual_types/transform_to_z3.py | py | 16.42 KiB |
torch/fx/experimental/migrate_gradual_types/util.py | py | 1.39 KiB |
torch/fx/experimental/migrate_gradual_types/z3_types.py | py | 837 B |
torch/fx/experimental/normalize.py | py | 5.54 KiB |
torch/fx/experimental/optimization.py | py | 17.13 KiB |
torch/fx/experimental/partitioner_utils.py | py | 12.33 KiB |
torch/fx/experimental/proxy_tensor.py | py | 85.71 KiB |
torch/fx/experimental/recording.py | py | 19.40 KiB |
torch/fx/experimental/refinement_types.py | py | 449 B |
torch/fx/experimental/rewriter.py | py | 5.49 KiB |
torch/fx/experimental/schema_type_annotation.py | py | 5.41 KiB |
torch/fx/experimental/sym_node.py | py | 55.57 KiB |
torch/fx/experimental/symbolic_shapes.py | py | 283.67 KiB |
torch/fx/experimental/unification | Folder | |
torch/fx/experimental/unification/__init__.py | py | 200 B |
torch/fx/experimental/unification/core.py | py | 2.83 KiB |
torch/fx/experimental/unification/dispatch.py | py | 201 B |
torch/fx/experimental/unification/match.py | py | 3.46 KiB |
torch/fx/experimental/unification/more.py | py | 3.01 KiB |
torch/fx/experimental/unification/multipledispatch | Folder | |
torch/fx/experimental/unification/multipledispatch/__init__.py | py | 146 B |
torch/fx/experimental/unification/multipledispatch/conflict.py | py | 4.25 KiB |
torch/fx/experimental/unification/multipledispatch/core.py | py | 2.73 KiB |
torch/fx/experimental/unification/multipledispatch/dispatcher.py | py | 13.98 KiB |
torch/fx/experimental/unification/multipledispatch/utils.py | py | 3.80 KiB |
torch/fx/experimental/unification/multipledispatch/variadic.py | py | 2.99 KiB |
torch/fx/experimental/unification/unification_tools.py | py | 10.73 KiB |
torch/fx/experimental/unification/utils.py | py | 2.97 KiB |
torch/fx/experimental/unification/variable.py | py | 2.10 KiB |
torch/fx/experimental/unify_refinements.py | py | 3.20 KiB |
torch/fx/experimental/validator.py | py | 32.71 KiB |
torch/fx/graph.py | py | 76.70 KiB |
torch/fx/graph_module.py | py | 38.76 KiB |
torch/fx/immutable_collections.py | py | 3.03 KiB |
torch/fx/interpreter.py | py | 22.61 KiB |
torch/fx/node.py | py | 35.04 KiB |
torch/fx/operator_schemas.py | py | 21.40 KiB |
torch/fx/passes | Folder | |
torch/fx/passes/__init__.py | py | 254 B |
torch/fx/passes/_tensorify_python_scalars.py | py | 14.09 KiB |
torch/fx/passes/annotate_getitem_nodes.py | py | 1.95 KiB |
torch/fx/passes/backends | Folder | |
torch/fx/passes/backends/__init__.py | py | 0 B |
torch/fx/passes/backends/cudagraphs.py | py | 2.09 KiB |
torch/fx/passes/dialect | Folder | |
torch/fx/passes/dialect/__init__.py | py | 0 B |
torch/fx/passes/dialect/common | Folder | |
torch/fx/passes/dialect/common/__init__.py | py | 0 B |
torch/fx/passes/dialect/common/cse_pass.py | py | 5.29 KiB |
torch/fx/passes/fake_tensor_prop.py | py | 2.83 KiB |
torch/fx/passes/graph_drawer.py | py | 18.75 KiB |
torch/fx/passes/graph_manipulation.py | py | 3.99 KiB |
torch/fx/passes/graph_transform_observer.py | py | 4.23 KiB |
torch/fx/passes/infra | Folder | |
torch/fx/passes/infra/__init__.py | py | 28 B |
torch/fx/passes/infra/partitioner.py | py | 17.00 KiB |
torch/fx/passes/infra/pass_base.py | py | 2.50 KiB |
torch/fx/passes/infra/pass_manager.py | py | 10.42 KiB |
torch/fx/passes/net_min_base.py | py | 35.31 KiB |
torch/fx/passes/operator_support.py | py | 7.68 KiB |
torch/fx/passes/param_fetch.py | py | 3.75 KiB |
torch/fx/passes/pass_manager.py | py | 7.14 KiB |
torch/fx/passes/reinplace.py | py | 34.36 KiB |
torch/fx/passes/runtime_assert.py | py | 28.70 KiB |
torch/fx/passes/shape_prop.py | py | 7.46 KiB |
torch/fx/passes/split_module.py | py | 25.70 KiB |
torch/fx/passes/split_utils.py | py | 11.29 KiB |
torch/fx/passes/splitter_base.py | py | 33.85 KiB |
torch/fx/passes/tests | Folder | |
torch/fx/passes/tests/__init__.py | py | 0 B |
torch/fx/passes/tests/test_pass_manager.py | py | 1.88 KiB |
torch/fx/passes/tools_common.py | py | 11.25 KiB |
torch/fx/passes/utils | Folder | |
torch/fx/passes/utils/__init__.py | py | 75 B |
torch/fx/passes/utils/common.py | py | 3.16 KiB |
torch/fx/passes/utils/fuser_utils.py | py | 9.54 KiB |
torch/fx/passes/utils/matcher_utils.py | py | 17.29 KiB |
torch/fx/passes/utils/matcher_with_name_node_map_utils.py | py | 4.24 KiB |
torch/fx/passes/utils/source_matcher_utils.py | py | 5.65 KiB |
torch/fx/proxy.py | py | 25.76 KiB |
torch/fx/subgraph_rewriter.py | py | 15.41 KiB |
torch/fx/tensor_type.py | py | 3.05 KiB |
torch/fx/traceback.py | py | 3.53 KiB |
torch/hub.py | py | 33.47 KiB |
torch/include | Folder | |
torch/include/ATen | Folder | |
torch/include/ATen/ATen.h | h | 1.12 KiB |
torch/include/ATen/AccumulateType.h | h | 6.14 KiB |
torch/include/ATen/ArrayRef.h | h | 46 B |
torch/include/ATen/Backend.h | h | 45 B |
torch/include/ATen/Backtrace.h | h | 48 B |
torch/include/ATen/BlasBackend.h | h | 703 B |
torch/include/ATen/CPUApplyUtils.h | h | 10.73 KiB |
torch/include/ATen/CPUFixedAllocator.h | h | 915 B |
torch/include/ATen/CPUFunctions.h | h | 1.94 KiB |
torch/include/ATen/CPUFunctions_inl.h | h | 27.02 KiB |
torch/include/ATen/CPUGeneratorImpl.h | h | 1.54 KiB |
torch/include/ATen/CUDAFunctions.h | h | 1.94 KiB |
torch/include/ATen/CUDAFunctions_inl.h | h | 32.25 KiB |
torch/include/ATen/CachedTensorUtils.h | h | 1.01 KiB |
torch/include/ATen/CollapseDims.h | h | 2.59 KiB |
torch/include/ATen/CompositeExplicitAutogradFunctions.h | h | 1.96 KiB |
torch/include/ATen/CompositeExplicitAutogradFunctions_inl.h | h | 39.82 KiB |
torch/include/ATen/CompositeExplicitAutogradNonFunctionalFunctions.h | h | 1.97 KiB |
torch/include/ATen/CompositeExplicitAutogradNonFunctionalFunctions_inl.h | h | 25.94 KiB |
torch/include/ATen/CompositeImplicitAutogradFunctions.h | h | 1.96 KiB |
torch/include/ATen/CompositeImplicitAutogradFunctions_inl.h | h | 34.69 KiB |
torch/include/ATen/CompositeImplicitAutogradNestedTensorFunctions.h | h | 1.97 KiB |
torch/include/ATen/CompositeImplicitAutogradNestedTensorFunctions_inl.h | h | 1.12 KiB |
torch/include/ATen/Config.h | h | 758 B |
torch/include/ATen/Context.h | h | 20.88 KiB |
torch/include/ATen/DLConvertor.h | h | 678 B |
torch/include/ATen/Device.h | h | 44 B |
torch/include/ATen/DeviceAccelerator.h | h | 938 B |
torch/include/ATen/DeviceGuard.h | h | 1.20 KiB |
torch/include/ATen/DimVector.h | h | 48 B |
torch/include/ATen/Dimname.h | h | 32 B |
torch/include/ATen/Dispatch.h | h | 40.03 KiB |
torch/include/ATen/Dispatch_v2.h | h | 59.10 KiB |
torch/include/ATen/DynamicLibrary.h | h | 715 B |
torch/include/ATen/EmptyTensor.h | h | 4.76 KiB |
torch/include/ATen/ExpandBase.h | h | 944 B |
torch/include/ATen/ExpandUtils.h | h | 16.47 KiB |
torch/include/ATen/Formatting.h | h | 35 B |
torch/include/ATen/FuncTorchTLS.h | h | 1.82 KiB |
torch/include/ATen/FunctionalStorageImpl.h | h | 7.72 KiB |
torch/include/ATen/FunctionalTensorWrapper.h | h | 17.79 KiB |
torch/include/ATen/Functions.h | h | 54.92 KiB |
torch/include/ATen/Generator.h | h | 48 B |
torch/include/ATen/InferSize.h | h | 2.67 KiB |
torch/include/ATen/InitialTensorOptions.h | h | 454 B |
torch/include/ATen/Layout.h | h | 44 B |
torch/include/ATen/LegacyBatchedFallback.h | h | 999 B |
torch/include/ATen/LegacyBatchedTensorImpl.h | h | 5.58 KiB |
torch/include/ATen/LegacyVmapMode.h | h | 953 B |
torch/include/ATen/LegacyVmapTransforms.h | h | 7.81 KiB |
torch/include/ATen/LinalgBackend.h | h | 750 B |
torch/include/ATen/MapAllocator.h | h | 3.70 KiB |
torch/include/ATen/MatrixRef.h | h | 2.92 KiB |
torch/include/ATen/MemoryOverlap.h | h | 1.30 KiB |
torch/include/ATen/MetaFunctions.h | h | 1.94 KiB |
torch/include/ATen/MetaFunctions_inl.h | h | 15.79 KiB |
torch/include/ATen/MethodOperators.h | h | 15.52 KiB |
torch/include/ATen/NamedTensor.h | h | 36 B |
torch/include/ATen/NamedTensorUtils.h | h | 6.84 KiB |
torch/include/ATen/NativeFunctions.h | h | 60.35 KiB |
torch/include/ATen/NativeMetaFunctions.h | h | 56.94 KiB |
torch/include/ATen/NestedTensorImpl.h | h | 10.26 KiB |
torch/include/ATen/NumericUtils.h | h | 5.21 KiB |
torch/include/ATen/OpMathType.h | h | 1.54 KiB |
torch/include/ATen/OpaqueTensorImpl.h | h | 6.23 KiB |
torch/include/ATen/Operators.h | h | 58.55 KiB |
torch/include/ATen/PTThreadPool.h | h | 408 B |
torch/include/ATen/PadNd.h | h | 206 B |
torch/include/ATen/Parallel-inl.h | h | 2.33 KiB |
torch/include/ATen/Parallel.h | h | 4.80 KiB |
torch/include/ATen/ParallelFuture.h | h | 312 B |
torch/include/ATen/ParallelNative.h | h | 307 B |
torch/include/ATen/ParallelOpenMP.h | h | 1.31 KiB |
torch/include/ATen/PythonTorchFunctionTLS.h | h | 1.20 KiB |
torch/include/ATen/RedispatchFunctions.h | h | 2.12 MiB |
torch/include/ATen/RegistrationDeclarations.h | h | 837.77 KiB |
torch/include/ATen/SavedTensorHooks.h | h | 2.41 KiB |
torch/include/ATen/Scalar.h | h | 47 B |
torch/include/ATen/ScalarOps.h | h | 1.61 KiB |
torch/include/ATen/ScalarType.h | h | 133 B |
torch/include/ATen/SequenceNumber.h | h | 346 B |
torch/include/ATen/SmallVector.h | h | 49 B |
torch/include/ATen/SparseCsrTensorImpl.h | h | 7.14 KiB |
torch/include/ATen/SparseCsrTensorUtils.h | h | 17.80 KiB |
torch/include/ATen/SparseTensorImpl.h | h | 15.36 KiB |
torch/include/ATen/Storage.h | h | 45 B |
torch/include/ATen/StorageUtils.h | h | 1.33 KiB |
torch/include/ATen/Tensor.h | h | 47 B |
torch/include/ATen/TensorAccessor.h | h | 53 B |
torch/include/ATen/TensorGeometry.h | h | 4.31 KiB |
torch/include/ATen/TensorIndexing.h | h | 24.20 KiB |
torch/include/ATen/TensorIterator.h | h | 38.92 KiB |
torch/include/ATen/TensorIteratorInternal.h | h | 1.96 KiB |
torch/include/ATen/TensorMeta.h | h | 5.05 KiB |
torch/include/ATen/TensorNames.h | h | 2.58 KiB |
torch/include/ATen/TensorOperators.h | h | 2.48 KiB |
torch/include/ATen/TensorOptions.h | h | 51 B |
torch/include/ATen/TensorSubclassLikeUtils.h | h | 3.24 KiB |
torch/include/ATen/TensorUtils.h | h | 6.00 KiB |
torch/include/ATen/ThreadLocalPythonObjects.h | h | 628 B |
torch/include/ATen/ThreadLocalState.h | h | 4.33 KiB |
torch/include/ATen/TracerMode.h | h | 5.51 KiB |
torch/include/ATen/TypeDefault.h | h | 696 B |
torch/include/ATen/Utils.h | h | 3.62 KiB |
torch/include/ATen/Version.h | h | 402 B |
torch/include/ATen/VmapGeneratedPlumbing.h | h | 1.68 MiB |
torch/include/ATen/WrapDimUtils.h | h | 4.91 KiB |
torch/include/ATen/WrapDimUtilsMulti.h | h | 1.09 KiB |
torch/include/ATen/autocast_mode.h | h | 38.92 KiB |
torch/include/ATen/ceil_div.h | h | 521 B |
torch/include/ATen/code_template.h | h | 7.03 KiB |
torch/include/ATen/core | Folder | |
torch/include/ATen/core/ATenGeneral.h | h | 48 B |
torch/include/ATen/core/ATenOpList.h | h | 259 B |
torch/include/ATen/core/ATen_fwd.h | h | 1.04 KiB |
torch/include/ATen/core/ATen_pch.h | h | 5.29 KiB |
torch/include/ATen/core/Array.h | h | 1.15 KiB |
torch/include/ATen/core/Backtrace.h | h | 61 B |
torch/include/ATen/core/CachingHostAllocator.h | h | 16.24 KiB |
torch/include/ATen/core/CheckMemoryFormat.h | h | 814 B |
torch/include/ATen/core/DeprecatedTypeProperties.h | h | 3.92 KiB |
torch/include/ATen/core/DeprecatedTypePropertiesRegistry.h | h | 877 B |
torch/include/ATen/core/Dict.h | h | 13.40 KiB |
torch/include/ATen/core/Dict_inl.h | h | 7.51 KiB |
torch/include/ATen/core/DimVector.h | h | 292 B |
torch/include/ATen/core/Dimname.h | h | 1.19 KiB |
torch/include/ATen/core/DistributionsHelper.h | h | 12.54 KiB |
torch/include/ATen/core/Formatting.h | h | 718 B |
torch/include/ATen/core/Generator.h | h | 6.44 KiB |
torch/include/ATen/core/GeneratorForPrivateuseone.h | h | 1.08 KiB |
torch/include/ATen/core/IListRef.h | h | 21.06 KiB |
torch/include/ATen/core/IListRef_inl.h | h | 6.16 KiB |
torch/include/ATen/core/LegacyTypeDispatch.h | h | 4.85 KiB |
torch/include/ATen/core/List.h | h | 16.18 KiB |
torch/include/ATen/core/List_inl.h | h | 10.85 KiB |
torch/include/ATen/core/MT19937RNGEngine.h | h | 6.55 KiB |
torch/include/ATen/core/NamedTensor.h | h | 5.25 KiB |
torch/include/ATen/core/NestedIntSymNodeImpl.h | h | 6.00 KiB |
torch/include/ATen/core/PhiloxRNGEngine.h | h | 7.80 KiB |
torch/include/ATen/core/PythonFallbackKernel.h | h | 1.10 KiB |
torch/include/ATen/core/PythonOpRegistrationTrampoline.h | h | 617 B |
torch/include/ATen/core/QuantizerBase.h | h | 2.71 KiB |
torch/include/ATen/core/Range.h | h | 443 B |
torch/include/ATen/core/Reduction.h | h | 413 B |
torch/include/ATen/core/Scalar.h | h | 30 B |
torch/include/ATen/core/ScalarType.h | h | 34 B |
torch/include/ATen/core/Tensor.h | h | 2.56 KiB |
torch/include/ATen/core/TensorAccessor.h | h | 10.55 KiB |
torch/include/ATen/core/TensorBase.h | h | 38.26 KiB |
torch/include/ATen/core/TensorBody.h | h | 289.11 KiB |
torch/include/ATen/core/TorchDispatchUtils.h | h | 501 B |
torch/include/ATen/core/TransformationHelper.h | h | 6.86 KiB |
torch/include/ATen/core/UndefinedTensorImpl.h | h | 43 B |
torch/include/ATen/core/UnsafeFromTH.h | h | 729 B |
torch/include/ATen/core/VariableHooksInterface.h | h | 3.54 KiB |
torch/include/ATen/core/Variadic.h | h | 2.41 KiB |
torch/include/ATen/core/Vitals.h | h | 2.47 KiB |
torch/include/ATen/core/alias_info.h | h | 4.21 KiB |
torch/include/ATen/core/aten_interned_strings.h | h | 56.78 KiB |
torch/include/ATen/core/blob.h | h | 5.32 KiB |
torch/include/ATen/core/boxing | Folder | |
torch/include/ATen/core/boxing/BoxedKernel.h | h | 7.91 KiB |
torch/include/ATen/core/boxing/BoxedKernel_impl.h | h | 3.28 KiB |
torch/include/ATen/core/boxing/KernelFunction.h | h | 8.59 KiB |
torch/include/ATen/core/boxing/KernelFunction_impl.h | h | 11.18 KiB |
torch/include/ATen/core/boxing/OperatorKernel.h | h | 719 B |
torch/include/ATen/core/boxing/impl | Folder | |
torch/include/ATen/core/boxing/impl/WrapFunctionIntoFunctor.h | h | 1.30 KiB |
torch/include/ATen/core/boxing/impl/WrapFunctionIntoRuntimeFunctor.h | h | 1.44 KiB |
torch/include/ATen/core/boxing/impl/boxing.h | h | 13.48 KiB |
torch/include/ATen/core/boxing/impl/make_boxed_from_unboxed_functor.h | h | 31.02 KiB |
torch/include/ATen/core/boxing/impl/test_helpers.h | h | 4.32 KiB |
torch/include/ATen/core/builtin_function.h | h | 2.08 KiB |
torch/include/ATen/core/class_type.h | h | 14.29 KiB |
torch/include/ATen/core/custom_class.h | h | 772 B |
torch/include/ATen/core/dispatch | Folder | |
torch/include/ATen/core/dispatch/CppSignature.h | h | 2.45 KiB |
torch/include/ATen/core/dispatch/DispatchKeyExtractor.h | h | 9.68 KiB |
torch/include/ATen/core/dispatch/Dispatcher.h | h | 32.99 KiB |
torch/include/ATen/core/dispatch/ObservedOperators.h | h | 346 B |
torch/include/ATen/core/dispatch/OperatorEntry.h | h | 12.89 KiB |
torch/include/ATen/core/dispatch/OperatorOptions.h | h | 953 B |
torch/include/ATen/core/dispatch/RegistrationHandleRAII.h | h | 894 B |
torch/include/ATen/core/dynamic_type.h | h | 10.66 KiB |
torch/include/ATen/core/enum_tag.h | h | 501 B |
torch/include/ATen/core/enum_type.h | h | 2.83 KiB |
torch/include/ATen/core/function.h | h | 3.49 KiB |
torch/include/ATen/core/function_schema.h | h | 24.09 KiB |
torch/include/ATen/core/function_schema_inl.h | h | 2.02 KiB |
torch/include/ATen/core/functional.h | h | 1.48 KiB |
torch/include/ATen/core/grad_mode.h | h | 220 B |
torch/include/ATen/core/interned_strings.h | h | 13.44 KiB |
torch/include/ATen/core/interned_strings_class.h | h | 754 B |
torch/include/ATen/core/ivalue.h | h | 51.85 KiB |
torch/include/ATen/core/ivalue_inl.h | h | 87.85 KiB |
torch/include/ATen/core/ivalue_to.h | h | 792 B |
torch/include/ATen/core/jit_type.h | h | 72.82 KiB |
torch/include/ATen/core/jit_type_base.h | h | 23.29 KiB |
torch/include/ATen/core/op_registration | Folder | |
torch/include/ATen/core/op_registration/adaption.h | h | 3.22 KiB |
torch/include/ATen/core/op_registration/infer_schema.h | h | 6.75 KiB |
torch/include/ATen/core/op_registration/op_allowlist.h | h | 6.91 KiB |
torch/include/ATen/core/op_registration/op_registration.h | h | 28.51 KiB |
torch/include/ATen/core/operator_name.h | h | 3.07 KiB |
torch/include/ATen/core/qualified_name.h | h | 4.43 KiB |
torch/include/ATen/core/rref_interface.h | h | 1.22 KiB |
torch/include/ATen/core/stack.h | h | 6.32 KiB |
torch/include/ATen/core/symbol.h | h | 5.88 KiB |
torch/include/ATen/core/type_factory.h | h | 3.28 KiB |
torch/include/ATen/core/type_ptr.h | h | 1.24 KiB |
torch/include/ATen/core/typeid.h | h | 30 B |
torch/include/ATen/cpp_custom_type_hack.h | h | 5.41 KiB |
torch/include/ATen/cpu | Folder | |
torch/include/ATen/cpu/FlushDenormal.h | h | 551 B |
torch/include/ATen/cpu/Utils.h | h | 836 B |
torch/include/ATen/cpu/vec | Folder | |
torch/include/ATen/cpu/vec/functional.h | h | 106 B |
torch/include/ATen/cpu/vec/functional_base.h | h | 12.78 KiB |
torch/include/ATen/cpu/vec/functional_bfloat16.h | h | 24.64 KiB |
torch/include/ATen/cpu/vec/intrinsics.h | h | 2.11 KiB |
torch/include/ATen/cpu/vec/sve | Folder | |
torch/include/ATen/cpu/vec/sve/sve_helper.h | h | 2.85 KiB |
torch/include/ATen/cpu/vec/sve/vec_common_sve.h | h | 7.01 KiB |
torch/include/ATen/cpu/vec/sve/vec_double.h | h | 17.45 KiB |
torch/include/ATen/cpu/vec/sve/vec_float.h | h | 20.22 KiB |
torch/include/ATen/cpu/vec/sve/vec_int.h | h | 28.19 KiB |
torch/include/ATen/cpu/vec/sve/vec_qint.h | h | 19.58 KiB |
torch/include/ATen/cpu/vec/vec.h | h | 1.34 KiB |
torch/include/ATen/cpu/vec/vec128 | Folder | |
torch/include/ATen/cpu/vec/vec128/vec128.h | h | 379 B |
torch/include/ATen/cpu/vec/vec128/vec128_bfloat16_neon.h | h | 23.33 KiB |
torch/include/ATen/cpu/vec/vec128/vec128_convert.h | h | 2.00 KiB |
torch/include/ATen/cpu/vec/vec128/vec128_float_neon.h | h | 19.86 KiB |
torch/include/ATen/cpu/vec/vec128/vec128_half_neon.h | h | 20.58 KiB |
torch/include/ATen/cpu/vec/vec128/vec128_reduced_precision_common_neon.h | h | 9.34 KiB |
torch/include/ATen/cpu/vec/vec256 | Folder | |
torch/include/ATen/cpu/vec/vec256/missing_vld1_neon.h | h | 13.68 KiB |
torch/include/ATen/cpu/vec/vec256/missing_vst1_neon.h | h | 290 B |
torch/include/ATen/cpu/vec/vec256/vec256.h | h | 11.95 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_bfloat16.h | h | 40.95 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_complex_double.h | h | 18.14 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_complex_float.h | h | 19.94 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_convert.h | h | 9.78 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_double.h | h | 14.79 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_float.h | h | 23.00 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_float_neon.h | h | 30.25 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_half_neon.h | h | 29.12 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_int.h | h | 60.74 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_mask.h | h | 8.86 KiB |
torch/include/ATen/cpu/vec/vec256/vec256_qint.h | h | 48.04 KiB |
torch/include/ATen/cpu/vec/vec256/vsx | Folder | |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_bfloat16_vsx.h | h | 2.13 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_common_vsx.h | h | 8.10 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_complex_double_vsx.h | h | 19.15 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_complex_float_vsx.h | h | 21.71 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_double_vsx.h | h | 15.96 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_float_vsx.h | h | 16.38 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_int16_vsx.h | h | 14.15 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_int32_vsx.h | h | 11.96 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_int64_vsx.h | h | 10.22 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_qint32_vsx.h | h | 9.82 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_qint8_vsx.h | h | 16.78 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vec256_quint8_vsx.h | h | 17.46 KiB |
torch/include/ATen/cpu/vec/vec256/vsx/vsx_helpers.h | h | 20.05 KiB |
torch/include/ATen/cpu/vec/vec256/zarch | Folder | |
torch/include/ATen/cpu/vec/vec256/zarch/vec256_zarch.h | h | 107.15 KiB |
torch/include/ATen/cpu/vec/vec512 | Folder | |
torch/include/ATen/cpu/vec/vec512/vec512.h | h | 10.52 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_bfloat16.h | h | 62.66 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_complex_double.h | h | 23.52 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_complex_float.h | h | 43.73 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_convert.h | h | 8.55 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_double.h | h | 16.70 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_float.h | h | 25.95 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_int.h | h | 54.65 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_mask.h | h | 12.42 KiB |
torch/include/ATen/cpu/vec/vec512/vec512_qint.h | h | 50.58 KiB |
torch/include/ATen/cpu/vec/vec_base.h | h | 41.84 KiB |
torch/include/ATen/cpu/vec/vec_convert.h | h | 1.86 KiB |
torch/include/ATen/cpu/vec/vec_half.h | h | 1.40 KiB |
torch/include/ATen/cpu/vec/vec_mask.h | h | 10.00 KiB |
torch/include/ATen/cpu/vec/vec_n.h | h | 13.50 KiB |
torch/include/ATen/cpu/vml.h | h | 6.10 KiB |
torch/include/ATen/cuda | Folder | |
torch/include/ATen/cuda/ATenCUDAGeneral.h | h | 199 B |
torch/include/ATen/cuda/ApplyGridUtils.cuh | cuh | 1.32 KiB |
torch/include/ATen/cuda/AsmUtils.cuh | cuh | 3.46 KiB |
torch/include/ATen/cuda/Atomic.cuh | cuh | 26.55 KiB |
torch/include/ATen/cuda/CUDAApplyUtils.cuh | cuh | 20.49 KiB |
torch/include/ATen/cuda/CUDABlas.h | h | 13.01 KiB |
torch/include/ATen/cuda/CUDAConfig.h | h | 799 B |
torch/include/ATen/cuda/CUDAContext.h | h | 247 B |
torch/include/ATen/cuda/CUDAContextLight.h | h | 2.77 KiB |
torch/include/ATen/cuda/CUDADataType.h | h | 2.91 KiB |
torch/include/ATen/cuda/CUDADevice.h | h | 555 B |
torch/include/ATen/cuda/CUDAEvent.h | h | 7.14 KiB |
torch/include/ATen/cuda/CUDAGeneratorImpl.h | h | 6.12 KiB |
torch/include/ATen/cuda/CUDAGraph.h | h | 3.35 KiB |
torch/include/ATen/cuda/CUDAGraphsUtils.cuh | cuh | 1.91 KiB |
torch/include/ATen/cuda/CUDASparse.h | h | 2.61 KiB |
torch/include/ATen/cuda/CUDASparseBlas.h | h | 12.79 KiB |
torch/include/ATen/cuda/CUDASparseDescriptors.h | h | 9.41 KiB |
torch/include/ATen/cuda/CUDATensorMethods.cuh | cuh | 285 B |
torch/include/ATen/cuda/CUDAUtils.h | h | 436 B |
torch/include/ATen/cuda/CachingHostAllocator.h | h | 1.33 KiB |
torch/include/ATen/cuda/DeviceUtils.cuh | cuh | 3.32 KiB |
torch/include/ATen/cuda/EmptyTensor.h | h | 1.22 KiB |
torch/include/ATen/cuda/Exceptions.h | h | 11.02 KiB |
torch/include/ATen/cuda/NumericLimits.cuh | cuh | 5.21 KiB |
torch/include/ATen/cuda/PeerToPeerAccess.h | h | 306 B |
torch/include/ATen/cuda/PhiloxCudaState.h | h | 90 B |
torch/include/ATen/cuda/PhiloxUtils.cuh | cuh | 99 B |
torch/include/ATen/cuda/PinnedMemoryAllocator.h | h | 256 B |
torch/include/ATen/cuda/ScanUtils.cuh | cuh | 2.06 KiB |
torch/include/ATen/cuda/Sleep.h | h | 332 B |
torch/include/ATen/cuda/ThrustAllocator.h | h | 528 B |
torch/include/ATen/cuda/cub.cuh | cuh | 21.31 KiB |
torch/include/ATen/cuda/cub.h | h | 3.37 KiB |
torch/include/ATen/cuda/cub_definitions.cuh | cuh | 1.47 KiB |
torch/include/ATen/cuda/detail | Folder | |
torch/include/ATen/cuda/detail/CUDAHooks.h | h | 2.44 KiB |
torch/include/ATen/cuda/detail/DeviceThreadHandles.h | h | 7.00 KiB |
torch/include/ATen/cuda/detail/IndexUtils.cuh | cuh | 903 B |
torch/include/ATen/cuda/detail/IntegerDivider.cuh | cuh | 4.05 KiB |
torch/include/ATen/cuda/detail/KernelUtils.h | h | 1.52 KiB |
torch/include/ATen/cuda/detail/LazyNVRTC.h | h | 231 B |
torch/include/ATen/cuda/detail/OffsetCalculator.cuh | cuh | 4.44 KiB |
torch/include/ATen/cuda/detail/PhiloxCudaStateRaw.cuh | cuh | 1.37 KiB |
torch/include/ATen/cuda/detail/TensorInfo.cuh | cuh | 3.28 KiB |
torch/include/ATen/cuda/detail/UnpackRaw.cuh | cuh | 1.46 KiB |
torch/include/ATen/cuda/jiterator.h | h | 1000 B |
torch/include/ATen/cuda/jiterator_impl.h | h | 7.19 KiB |
torch/include/ATen/cuda/llvm_jit_strings.h | h | 442 B |
torch/include/ATen/cuda/tunable | Folder | |
torch/include/ATen/cuda/tunable/GemmCommon.h | h | 13.39 KiB |
torch/include/ATen/cuda/tunable/GemmHipblaslt.h | h | 19.18 KiB |
torch/include/ATen/cuda/tunable/GemmRocblas.h | h | 10.27 KiB |
torch/include/ATen/cuda/tunable/StreamTimer.h | h | 811 B |
torch/include/ATen/cuda/tunable/Tunable.h | h | 8.07 KiB |
torch/include/ATen/cuda/tunable/TunableGemm.h | h | 9.06 KiB |
torch/include/ATen/cuda/tunable/TunableOp.h | h | 10.70 KiB |
torch/include/ATen/cudnn | Folder | |
torch/include/ATen/cudnn/Descriptors.h | h | 15.09 KiB |
torch/include/ATen/cudnn/Exceptions.h | h | 0 B |
torch/include/ATen/cudnn/Handle.h | h | 202 B |
torch/include/ATen/cudnn/Handles.h | h | 46 B |
torch/include/ATen/cudnn/Types.h | h | 324 B |
torch/include/ATen/cudnn/Utils.h | h | 617 B |
torch/include/ATen/cudnn/cudnn-wrapper.h | h | 556 B |
torch/include/ATen/detail | Folder | |
torch/include/ATen/detail/AcceleratorHooksInterface.h | h | 2.34 KiB |
torch/include/ATen/detail/CUDAHooksInterface.h | h | 7.25 KiB |
torch/include/ATen/detail/FunctionTraits.h | h | 3.10 KiB |
torch/include/ATen/detail/HIPHooksInterface.h | h | 2.03 KiB |
torch/include/ATen/detail/HPUHooksInterface.h | h | 1.66 KiB |
torch/include/ATen/detail/IPUHooksInterface.h | h | 1.27 KiB |
torch/include/ATen/detail/MAIAHooksInterface.h | h | 1.28 KiB |
torch/include/ATen/detail/MPSHooksInterface.h | h | 3.49 KiB |
torch/include/ATen/detail/MTIAHooksInterface.h | h | 3.35 KiB |
torch/include/ATen/detail/PrivateUse1HooksInterface.h | h | 2.06 KiB |
torch/include/ATen/detail/XPUHooksInterface.h | h | 2.49 KiB |
torch/include/ATen/div_rtn.h | h | 222 B |
torch/include/ATen/dlpack.h | h | 7.07 KiB |
torch/include/ATen/functorch | Folder | |
torch/include/ATen/functorch/ADInterpreters.h | h | 1.56 KiB |
torch/include/ATen/functorch/BatchRulesHelper.h | h | 18.83 KiB |
torch/include/ATen/functorch/BatchedFallback.h | h | 3.44 KiB |
torch/include/ATen/functorch/BatchedTensorImpl.h | h | 6.40 KiB |
torch/include/ATen/functorch/BatchingMetaprogramming.h | h | 5.00 KiB |
torch/include/ATen/functorch/DynamicLayer.h | h | 5.55 KiB |
torch/include/ATen/functorch/FunctionalizeInterpreter.h | h | 929 B |
torch/include/ATen/functorch/Interpreter.h | h | 7.91 KiB |
torch/include/ATen/functorch/LegacyVmapTransforms.h | h | 8.24 KiB |
torch/include/ATen/functorch/Macros.h | h | 53 B |
torch/include/ATen/functorch/PlumbingHelper.h | h | 2.86 KiB |
torch/include/ATen/functorch/TensorWrapper.h | h | 4.03 KiB |
torch/include/ATen/functorch/VmapInterpreter.h | h | 982 B |
torch/include/ATen/hip | Folder | |
torch/include/ATen/hip/impl | Folder | |
torch/include/ATen/hip/impl/HIPAllocatorMasqueradingAsCUDA.h | h | 1.00 KiB |
torch/include/ATen/hip/impl/HIPCachingAllocatorMasqueradingAsCUDA.h | h | 535 B |
torch/include/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.h | h | 15.25 KiB |
torch/include/ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h | h | 4.54 KiB |
torch/include/ATen/jit_macros.h | h | 237 B |
torch/include/ATen/jiterator_macros.h | h | 1.51 KiB |
torch/include/ATen/miopen | Folder | |
torch/include/ATen/miopen/Descriptors.h | h | 4.79 KiB |
torch/include/ATen/miopen/Exceptions.h | h | 1.09 KiB |
torch/include/ATen/miopen/Handle.h | h | 150 B |
torch/include/ATen/miopen/Types.h | h | 245 B |
torch/include/ATen/miopen/Utils.h | h | 419 B |
torch/include/ATen/miopen/miopen-wrapper.h | h | 44 B |
torch/include/ATen/mps | Folder | |
torch/include/ATen/mps/EmptyTensor.h | h | 783 B |
torch/include/ATen/mps/IndexKernels.h | h | 10.59 KiB |
torch/include/ATen/mps/MPSAllocator.h | h | 18.62 KiB |
torch/include/ATen/mps/MPSAllocatorInterface.h | h | 2.72 KiB |
torch/include/ATen/mps/MPSDevice.h | h | 1.99 KiB |
torch/include/ATen/mps/MPSEvent.h | h | 3.59 KiB |
torch/include/ATen/mps/MPSGeneratorImpl.h | h | 1.58 KiB |
torch/include/ATen/mps/MPSGuardImpl.h | h | 5.23 KiB |
torch/include/ATen/mps/MPSHooks.h | h | 2.11 KiB |
torch/include/ATen/mps/MPSProfiler.h | h | 17.25 KiB |
torch/include/ATen/mps/MPSStream.h | h | 4.66 KiB |
torch/include/ATen/native | Folder | |
torch/include/ATen/native/Activation.h | h | 3.50 KiB |
torch/include/ATen/native/AdaptivePooling.h | h | 2.42 KiB |
torch/include/ATen/native/AmpKernels.h | h | 645 B |
torch/include/ATen/native/BatchLinearAlgebra.h | h | 9.99 KiB |
torch/include/ATen/native/BinaryOps.h | h | 5.91 KiB |
torch/include/ATen/native/BucketizationUtils.h | h | 7.78 KiB |
torch/include/ATen/native/CPUBlas.h | h | 6.78 KiB |
torch/include/ATen/native/CPUFallback.h | h | 2.40 KiB |
torch/include/ATen/native/CanUse32BitIndexMath.h | h | 255 B |
torch/include/ATen/native/ComplexHelper.h | h | 4.05 KiB |
torch/include/ATen/native/CompositeRandomAccessor.h | h | 910 B |
torch/include/ATen/native/CompositeRandomAccessorCommon.h | h | 6.83 KiB |
torch/include/ATen/native/ConvUtils.h | h | 19.56 KiB |
torch/include/ATen/native/ConvolutionMM3d.h | h | 354 B |
torch/include/ATen/native/Copy.h | h | 392 B |
torch/include/ATen/native/Cross.h | h | 273 B |
torch/include/ATen/native/DilatedConvolutionUtils.h | h | 6.48 KiB |
torch/include/ATen/native/DispatchStub.h | h | 15.55 KiB |
torch/include/ATen/native/Distance.h | h | 740 B |
torch/include/ATen/native/DistributionTemplates.h | h | 18.45 KiB |
torch/include/ATen/native/Distributions.h | h | 21.57 KiB |
torch/include/ATen/native/EmbeddingBag.h | h | 5.24 KiB |
torch/include/ATen/native/Fill.h | h | 417 B |
torch/include/ATen/native/ForeachUtils.h | h | 15.11 KiB |
torch/include/ATen/native/FractionalMaxPooling.h | h | 2.19 KiB |
torch/include/ATen/native/FunctionOfAMatrixUtils.h | h | 408 B |
torch/include/ATen/native/FusedAdagrad.h | h | 515 B |
torch/include/ATen/native/FusedAdam.h | h | 710 B |
torch/include/ATen/native/FusedSGD.h | h | 537 B |
torch/include/ATen/native/Gelu.h | h | 887 B |
torch/include/ATen/native/GridSampler.h | h | 10.45 KiB |
torch/include/ATen/native/GridSamplerUtils.h | h | 3.52 KiB |
torch/include/ATen/native/Histogram.h | h | 761 B |
torch/include/ATen/native/IndexKernel.h | h | 1.70 KiB |
torch/include/ATen/native/IndexingUtils.h | h | 5.65 KiB |
torch/include/ATen/native/Lerp.h | h | 1.47 KiB |
torch/include/ATen/native/LinearAlgebra.h | h | 316 B |
torch/include/ATen/native/LinearAlgebraUtils.h | h | 26.35 KiB |
torch/include/ATen/native/LossMulti.h | h | 2.13 KiB |
torch/include/ATen/native/Math.h | h | 142.98 KiB |
torch/include/ATen/native/MathBitFallThroughLists.h | h | 4.11 KiB |
torch/include/ATen/native/MathBitsFallback.h | h | 7.29 KiB |
torch/include/ATen/native/MaxPooling.h | h | 3.29 KiB |
torch/include/ATen/native/NonEmptyUtils.h | h | 626 B |
torch/include/ATen/native/NonSymbolicBC.h | h | 2.83 KiB |
torch/include/ATen/native/Normalization.h | h | 572 B |
torch/include/ATen/native/Padding.h | h | 2.02 KiB |
torch/include/ATen/native/PixelShuffle.h | h | 1.76 KiB |
torch/include/ATen/native/PointwiseOps.h | h | 809 B |
torch/include/ATen/native/Pool.h | h | 13.12 KiB |
torch/include/ATen/native/Pow.h | h | 1.68 KiB |
torch/include/ATen/native/RNN.h | h | 2.50 KiB |
torch/include/ATen/native/RangeFactories.h | h | 366 B |
torch/include/ATen/native/ReduceAllOps.h | h | 413 B |
torch/include/ATen/native/ReduceOps.h | h | 1.78 KiB |
torch/include/ATen/native/ReduceOpsUtils.h | h | 16.49 KiB |
torch/include/ATen/native/ReductionType.h | h | 1.15 KiB |
torch/include/ATen/native/Repeat.h | h | 1.49 KiB |
torch/include/ATen/native/Resize.h | h | 6.59 KiB |
torch/include/ATen/native/ResizeCommon.h | h | 2.51 KiB |
torch/include/ATen/native/ScatterGatherChecks.h | h | 3.72 KiB |
torch/include/ATen/native/SegmentReduce.h | h | 1.28 KiB |
torch/include/ATen/native/SharedReduceOps.h | h | 16.32 KiB |
torch/include/ATen/native/SobolEngineOpsUtils.h | h | 1.85 KiB |
torch/include/ATen/native/Sorting.h | h | 644 B |
torch/include/ATen/native/SortingUtils.h | h | 2.70 KiB |
torch/include/ATen/native/SparseTensorUtils.h | h | 6.53 KiB |
torch/include/ATen/native/SpectralOpsUtils.h | h | 3.29 KiB |
torch/include/ATen/native/StridedRandomAccessor.h | h | 6.97 KiB |
torch/include/ATen/native/TensorAdvancedIndexing.h | h | 2.75 KiB |
torch/include/ATen/native/TensorAdvancedIndexingUtils.h | h | 3.10 KiB |
torch/include/ATen/native/TensorCompare.h | h | 1.49 KiB |
torch/include/ATen/native/TensorConversions.h | h | 845 B |
torch/include/ATen/native/TensorDimApply.h | h | 1.75 KiB |
torch/include/ATen/native/TensorFactories.h | h | 5.13 KiB |
torch/include/ATen/native/TensorIterator.h | h | 48 B |
torch/include/ATen/native/TensorIteratorDynamicCasting.h | h | 1.81 KiB |
torch/include/ATen/native/TensorProperties.h | h | 210 B |
torch/include/ATen/native/TensorShape.h | h | 4.37 KiB |
torch/include/ATen/native/TensorTransformations.h | h | 926 B |
torch/include/ATen/native/TopKImpl.h | h | 3.47 KiB |
torch/include/ATen/native/TransposeType.h | h | 601 B |
torch/include/ATen/native/TriangularOpsUtils.h | h | 2.01 KiB |
torch/include/ATen/native/TypeProperties.h | h | 678 B |
torch/include/ATen/native/UnaryOps.h | h | 5.50 KiB |
torch/include/ATen/native/Unfold2d.h | h | 1.00 KiB |
torch/include/ATen/native/Unfold3d.h | h | 922 B |
torch/include/ATen/native/UnfoldBackward.h | h | 3.13 KiB |
torch/include/ATen/native/UpSample.h | h | 19.26 KiB |
torch/include/ATen/native/batch_norm.h | h | 1.43 KiB |
torch/include/ATen/native/cpu | Folder | |
torch/include/ATen/native/cpu/AtomicAddFloat.h | h | 894 B |
torch/include/ATen/native/cpu/CatKernel.h | h | 318 B |
torch/include/ATen/native/cpu/ChannelShuffleKernel.h | h | 300 B |
torch/include/ATen/native/cpu/CopyKernel.h | h | 326 B |
torch/include/ATen/native/cpu/DepthwiseConvKernel.h | h | 491 B |
torch/include/ATen/native/cpu/DistributionTemplates.h | h | 16.49 KiB |
torch/include/ATen/native/cpu/Gelu.h | h | 2.81 KiB |
torch/include/ATen/native/cpu/GridSamplerKernel.h | h | 857 B |
torch/include/ATen/native/cpu/IndexKernelUtils.h | h | 2.94 KiB |
torch/include/ATen/native/cpu/Intrinsics.h | h | 1.22 KiB |
torch/include/ATen/native/cpu/IsContiguous.h | h | 2.38 KiB |
torch/include/ATen/native/cpu/LogAddExp.h | h | 2.45 KiB |
torch/include/ATen/native/cpu/Loops.h | h | 14.93 KiB |
torch/include/ATen/native/cpu/MaxUnpoolKernel.h | h | 320 B |
torch/include/ATen/native/cpu/PixelShuffleKernel.h | h | 334 B |
torch/include/ATen/native/cpu/Reduce.h | h | 12.09 KiB |
torch/include/ATen/native/cpu/ReduceUtils.h | h | 8.77 KiB |
torch/include/ATen/native/cpu/ReducedPrecisionFloatGemvFastPathKernel.h | h | 839 B |
torch/include/ATen/native/cpu/SampledAddmmKernel.h | h | 334 B |
torch/include/ATen/native/cpu/SerialStackImpl.h | h | 5.47 KiB |
torch/include/ATen/native/cpu/SoftmaxKernel.h | h | 963 B |
torch/include/ATen/native/cpu/SpmmReduceKernel.h | h | 1.34 KiB |
torch/include/ATen/native/cpu/StackKernel.h | h | 320 B |
torch/include/ATen/native/cpu/UpSampleKernelAVXAntialias.h | h | 58.16 KiB |
torch/include/ATen/native/cpu/WeightNormKernel.h | h | 570 B |
torch/include/ATen/native/cpu/avx_mathfun.h | h | 17.55 KiB |
torch/include/ATen/native/cpu/int_mm_kernel.h | h | 576 B |
torch/include/ATen/native/cpu/mixed_data_type.h | h | 1.42 KiB |
torch/include/ATen/native/cpu/moments_utils.h | h | 6.62 KiB |
torch/include/ATen/native/cpu/utils.h | h | 7.18 KiB |
torch/include/ATen/native/cpu/zmath.h | h | 6.71 KiB |
torch/include/ATen/native/cuda | Folder | |
torch/include/ATen/native/cuda/Activation.h | h | 568 B |
torch/include/ATen/native/cuda/BinaryInternal.h | h | 1.20 KiB |
torch/include/ATen/native/cuda/CUDAJitLoops.cuh | cuh | 10.83 KiB |
torch/include/ATen/native/cuda/CUDALoops.cuh | cuh | 12.88 KiB |
torch/include/ATen/native/cuda/CompositeRandomAccessor.h | h | 964 B |
torch/include/ATen/native/cuda/Copy.h | h | 165 B |
torch/include/ATen/native/cuda/CuFFTPlanCache.h | h | 18.02 KiB |
torch/include/ATen/native/cuda/CuFFTUtils.h | h | 1.90 KiB |
torch/include/ATen/native/cuda/DeviceSqrt.cuh | cuh | 610 B |
torch/include/ATen/native/cuda/DistributionTemplates.h | h | 28.07 KiB |
torch/include/ATen/native/cuda/Distributions.h | h | 666 B |
torch/include/ATen/native/cuda/EmbeddingBackwardKernel.cuh | cuh | 576 B |
torch/include/ATen/native/cuda/ForeachFunctors.cuh | cuh | 24.78 KiB |
torch/include/ATen/native/cuda/ForeachMinMaxFunctors.cuh | cuh | 448 B |
torch/include/ATen/native/cuda/GridSampler.cuh | cuh | 11.03 KiB |
torch/include/ATen/native/cuda/GridSampler.h | h | 1.16 KiB |
torch/include/ATen/native/cuda/IndexKernel.h | h | 366 B |
torch/include/ATen/native/cuda/JitLoops.cuh | cuh | 6.93 KiB |
torch/include/ATen/native/cuda/KernelUtils.cuh | cuh | 7.85 KiB |
torch/include/ATen/native/cuda/LaunchUtils.h | h | 298 B |
torch/include/ATen/native/cuda/Loops.cuh | cuh | 11.70 KiB |
torch/include/ATen/native/cuda/Math.cuh | cuh | 123.57 KiB |
torch/include/ATen/native/cuda/MemoryAccess.cuh | cuh | 15.89 KiB |
torch/include/ATen/native/cuda/MiscUtils.h | h | 990 B |
torch/include/ATen/native/cuda/MultiTensorApply.cuh | cuh | 13.98 KiB |
torch/include/ATen/native/cuda/Normalization.cuh | cuh | 74.19 KiB |
torch/include/ATen/native/cuda/PersistentSoftmax.cuh | cuh | 17.90 KiB |
torch/include/ATen/native/cuda/Pow.cuh | cuh | 2.13 KiB |
torch/include/ATen/native/cuda/Randperm.cuh | cuh | 2.13 KiB |
torch/include/ATen/native/cuda/Reduce.cuh | cuh | 49.52 KiB |
torch/include/ATen/native/cuda/ReduceOps.h | h | 520 B |
torch/include/ATen/native/cuda/Resize.h | h | 1.57 KiB |
torch/include/ATen/native/cuda/RowwiseScaledMM.h | h | 383 B |
torch/include/ATen/native/cuda/ScanKernels.h | h | 797 B |
torch/include/ATen/native/cuda/ScanUtils.cuh | cuh | 20.99 KiB |
torch/include/ATen/native/cuda/Sort.h | h | 418 B |
torch/include/ATen/native/cuda/SortStable.h | h | 456 B |
torch/include/ATen/native/cuda/SortUtils.cuh | cuh | 12.42 KiB |
torch/include/ATen/native/cuda/Sorting.h | h | 426 B |
torch/include/ATen/native/cuda/SortingCommon.cuh | cuh | 5.52 KiB |
torch/include/ATen/native/cuda/SortingRadixSelect.cuh | cuh | 12.44 KiB |
torch/include/ATen/native/cuda/TensorModeKernel.cuh | cuh | 14.57 KiB |
torch/include/ATen/native/cuda/TensorModeKernel.h | h | 450 B |
torch/include/ATen/native/cuda/TensorTopK.h | h | 280 B |
torch/include/ATen/native/cuda/UniqueCub.cuh | cuh | 314 B |
torch/include/ATen/native/cuda/UpSample.cuh | cuh | 11.71 KiB |
torch/include/ATen/native/cuda/block_reduce.cuh | cuh | 4.26 KiB |
torch/include/ATen/native/cuda/fused_adam_amsgrad_impl.cuh | cuh | 1.04 KiB |
torch/include/ATen/native/cuda/fused_adam_impl.cuh | cuh | 977 B |
torch/include/ATen/native/cuda/fused_adam_utils.cuh | cuh | 7.03 KiB |
torch/include/ATen/native/cuda/fused_adamw_amsgrad_impl.cuh | cuh | 1.04 KiB |
torch/include/ATen/native/cuda/fused_adamw_impl.cuh | cuh | 979 B |
torch/include/ATen/native/cuda/im2col.cuh | cuh | 9.96 KiB |
torch/include/ATen/native/cuda/jit_utils.h | h | 6.16 KiB |
torch/include/ATen/native/cuda/reduction_template.cuh | cuh | 21.82 KiB |
torch/include/ATen/native/cuda/thread_constants.h | h | 633 B |
torch/include/ATen/native/cuda/vol2col.cuh | cuh | 8.16 KiB |
torch/include/ATen/native/group_norm.h | h | 947 B |
torch/include/ATen/native/hip | Folder | |
torch/include/ATen/native/hip/ck_gemm.h | h | 646 B |
torch/include/ATen/native/hip/ck_gemm_template.h | h | 9.53 KiB |
torch/include/ATen/native/im2col.h | h | 5.25 KiB |
torch/include/ATen/native/im2col_shape_check.h | h | 7.00 KiB |
torch/include/ATen/native/layer_norm.h | h | 4.24 KiB |
torch/include/ATen/native/mkldnn | Folder | |
torch/include/ATen/native/mkldnn/xpu | Folder | |
torch/include/ATen/native/mkldnn/xpu/detail | Folder | |
torch/include/ATen/native/mkldnn/xpu/detail/Attr.h | h | 14.17 KiB |
torch/include/ATen/native/mkldnn/xpu/detail/Utils.h | h | 2.76 KiB |
torch/include/ATen/native/mkldnn/xpu/detail/oneDNN.h | h | 3.04 KiB |
torch/include/ATen/native/mkldnn/xpu/detail/oneDNNContext.h | h | 2.77 KiB |
torch/include/ATen/native/mps | Folder | |
torch/include/ATen/native/mps/Copy.h | h | 309 B |
torch/include/ATen/native/mps/MPSGraphSequoiaOps.h | h | 1.52 KiB |
torch/include/ATen/native/mps/MPSGraphSonomaOps.h | h | 2.32 KiB |
torch/include/ATen/native/mps/MPSGraphVenturaOps.h | h | 12.12 KiB |
torch/include/ATen/native/mps/OperationUtils.h | h | 21.33 KiB |
torch/include/ATen/native/mps/TensorFactory.h | h | 1018 B |
torch/include/ATen/native/mps/UnaryConstants.h | h | 2.81 KiB |
torch/include/ATen/native/nested | Folder | |
torch/include/ATen/native/nested/NestedTensorBinaryOps.h | h | 431 B |
torch/include/ATen/native/nested/NestedTensorMath.h | h | 2.74 KiB |
torch/include/ATen/native/nested/NestedTensorTransformerFunctions.h | h | 2.86 KiB |
torch/include/ATen/native/nested/NestedTensorTransformerUtils.h | h | 1.38 KiB |
torch/include/ATen/native/nested/NestedTensorUtils.h | h | 15.35 KiB |
torch/include/ATen/native/quantized | Folder | |
torch/include/ATen/native/quantized/AffineQuantizer.h | h | 3.79 KiB |
torch/include/ATen/native/quantized/AffineQuantizerBase.h | h | 1.47 KiB |
torch/include/ATen/native/quantized/ConvUtils.h | h | 2.25 KiB |
torch/include/ATen/native/quantized/Copy.h | h | 179 B |
torch/include/ATen/native/quantized/FakeQuantAffine.h | h | 1.81 KiB |
torch/include/ATen/native/quantized/IndexKernel.h | h | 618 B |
torch/include/ATen/native/quantized/PackedParams.h | h | 4.77 KiB |
torch/include/ATen/native/quantized/cpu | Folder | |
torch/include/ATen/native/quantized/cpu/BinaryOps.h | h | 174 B |
torch/include/ATen/native/quantized/cpu/EmbeddingPackedParams.h | h | 950 B |
torch/include/ATen/native/quantized/cpu/OnednnUtils.h | h | 13.89 KiB |
torch/include/ATen/native/quantized/cpu/QnnpackUtils.h | h | 17.79 KiB |
torch/include/ATen/native/quantized/cpu/QuantUtils.h | h | 8.37 KiB |
torch/include/ATen/native/quantized/cpu/QuantizedOps.h | h | 7.87 KiB |
torch/include/ATen/native/quantized/cpu/RuyUtils.h | h | 355 B |
torch/include/ATen/native/quantized/cpu/XnnpackUtils.h | h | 14.10 KiB |
torch/include/ATen/native/quantized/cpu/conv_serialization.h | h | 12.86 KiB |
torch/include/ATen/native/quantized/cpu/fbgemm_utils.h | h | 12.09 KiB |
torch/include/ATen/native/quantized/cpu/init_qnnpack.h | h | 132 B |
torch/include/ATen/native/quantized/cpu/qconv.h | h | 3.57 KiB |
torch/include/ATen/native/quantized/cpu/qembeddingbag.h | h | 1.02 KiB |
torch/include/ATen/native/quantized/cpu/qembeddingbag_prepack.h | h | 306 B |
torch/include/ATen/native/quantized/cpu/qlinear.h | h | 1.54 KiB |
torch/include/ATen/native/transformers | Folder | |
torch/include/ATen/native/transformers/attention.h | h | 2.32 KiB |
torch/include/ATen/native/transformers/sdp_utils_cpp.h | h | 17.82 KiB |
torch/include/ATen/native/utils | Folder | |
torch/include/ATen/native/utils/Factory.h | h | 577 B |
torch/include/ATen/native/utils/ParamUtils.h | h | 1.21 KiB |
torch/include/ATen/native/utils/ParamsHash.h | h | 3.15 KiB |
torch/include/ATen/native/verbose_wrapper.h | h | 201 B |
torch/include/ATen/native/vol2col.h | h | 3.58 KiB |
torch/include/ATen/ops | Folder | |
torch/include/ATen/ops/_adaptive_avg_pool2d.h | h | 4.11 KiB |
torch/include/ATen/ops/_adaptive_avg_pool2d_backward.h | h | 1.43 KiB |
torch/include/ATen/ops/_adaptive_avg_pool2d_backward_compositeexplicitautograd_dispatch.h | h | 1001 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_backward_cpu_dispatch.h | h | 798 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_backward_cuda_dispatch.h | h | 800 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_backward_native.h | h | 803 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_backward_ops.h | h | 1.86 KiB |
torch/include/ATen/ops/_adaptive_avg_pool2d_compositeexplicitautograd_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/_adaptive_avg_pool2d_cpu_dispatch.h | h | 895 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_cuda_dispatch.h | h | 897 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_native.h | h | 1001 B |
torch/include/ATen/ops/_adaptive_avg_pool2d_ops.h | h | 1.82 KiB |
torch/include/ATen/ops/_adaptive_avg_pool3d.h | h | 4.11 KiB |
torch/include/ATen/ops/_adaptive_avg_pool3d_backward.h | h | 1.43 KiB |
torch/include/ATen/ops/_adaptive_avg_pool3d_backward_compositeexplicitautograd_dispatch.h | h | 1001 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_backward_cpu_dispatch.h | h | 798 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_backward_cuda_dispatch.h | h | 800 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_backward_native.h | h | 803 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_backward_ops.h | h | 1.86 KiB |
torch/include/ATen/ops/_adaptive_avg_pool3d_compositeexplicitautograd_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/_adaptive_avg_pool3d_cpu_dispatch.h | h | 895 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_cuda_dispatch.h | h | 897 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_native.h | h | 889 B |
torch/include/ATen/ops/_adaptive_avg_pool3d_ops.h | h | 1.82 KiB |
torch/include/ATen/ops/_add_batch_dim.h | h | 754 B |
torch/include/ATen/ops/_add_batch_dim_compositeimplicitautograd_dispatch.h | h | 829 B |
torch/include/ATen/ops/_add_batch_dim_native.h | h | 539 B |
torch/include/ATen/ops/_add_batch_dim_ops.h | h | 1.07 KiB |
torch/include/ATen/ops/_add_relu.h | h | 2.75 KiB |
torch/include/ATen/ops/_add_relu_compositeexplicitautograd_dispatch.h | h | 1003 B |
torch/include/ATen/ops/_add_relu_cpu_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/_add_relu_meta_dispatch.h | h | 908 B |
torch/include/ATen/ops/_add_relu_native.h | h | 1.13 KiB |
torch/include/ATen/ops/_add_relu_ops.h | h | 4.79 KiB |
torch/include/ATen/ops/_addmm_activation.h | h | 1.82 KiB |
torch/include/ATen/ops/_addmm_activation_compositeexplicitautogradnonfunctional_dispatch.h | h | 950 B |
torch/include/ATen/ops/_addmm_activation_cpu_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_addmm_activation_cuda_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_addmm_activation_meta.h | h | 729 B |
torch/include/ATen/ops/_addmm_activation_meta_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_addmm_activation_native.h | h | 1.03 KiB |
torch/include/ATen/ops/_addmm_activation_ops.h | h | 2.37 KiB |
torch/include/ATen/ops/_aminmax.h | h | 2.23 KiB |
torch/include/ATen/ops/_aminmax_compositeexplicitautograd_dispatch.h | h | 1.28 KiB |
torch/include/ATen/ops/_aminmax_cpu_dispatch.h | h | 885 B |
torch/include/ATen/ops/_aminmax_cuda_dispatch.h | h | 887 B |
torch/include/ATen/ops/_aminmax_native.h | h | 930 B |
torch/include/ATen/ops/_aminmax_ops.h | h | 3.34 KiB |
torch/include/ATen/ops/_amp_foreach_non_finite_check_and_unscale.h | h | 2.09 KiB |
torch/include/ATen/ops/_amp_foreach_non_finite_check_and_unscale_compositeexplicitautograd_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/_amp_foreach_non_finite_check_and_unscale_cpu_dispatch.h | h | 823 B |
torch/include/ATen/ops/_amp_foreach_non_finite_check_and_unscale_cuda_dispatch.h | h | 825 B |
torch/include/ATen/ops/_amp_foreach_non_finite_check_and_unscale_native.h | h | 1.04 KiB |
torch/include/ATen/ops/_amp_foreach_non_finite_check_and_unscale_ops.h | h | 2.96 KiB |
torch/include/ATen/ops/_amp_update_scale.h | h | 2.72 KiB |
torch/include/ATen/ops/_amp_update_scale_compositeexplicitautograd_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/_amp_update_scale_cpu_dispatch.h | h | 892 B |
torch/include/ATen/ops/_amp_update_scale_cuda_dispatch.h | h | 894 B |
torch/include/ATen/ops/_amp_update_scale_meta_dispatch.h | h | 894 B |
torch/include/ATen/ops/_amp_update_scale_native.h | h | 1.30 KiB |
torch/include/ATen/ops/_amp_update_scale_ops.h | h | 3.59 KiB |
torch/include/ATen/ops/_assert_async.h | h | 878 B |
torch/include/ATen/ops/_assert_async_cpu_dispatch.h | h | 829 B |
torch/include/ATen/ops/_assert_async_cuda_dispatch.h | h | 831 B |
torch/include/ATen/ops/_assert_async_native.h | h | 750 B |
torch/include/ATen/ops/_assert_async_ops.h | h | 1.54 KiB |
torch/include/ATen/ops/_assert_scalar.h | h | 723 B |
torch/include/ATen/ops/_assert_scalar_compositeexplicitautograd_dispatch.h | h | 818 B |
torch/include/ATen/ops/_assert_scalar_native.h | h | 528 B |
torch/include/ATen/ops/_assert_scalar_ops.h | h | 1.03 KiB |
torch/include/ATen/ops/_assert_tensor_metadata.h | h | 2.42 KiB |
torch/include/ATen/ops/_assert_tensor_metadata_compositeimplicitautograd_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/_assert_tensor_metadata_native.h | h | 651 B |
torch/include/ATen/ops/_assert_tensor_metadata_ops.h | h | 1.32 KiB |
torch/include/ATen/ops/_autocast_to_full_precision.h | h | 527 B |
torch/include/ATen/ops/_autocast_to_full_precision_compositeimplicitautograd_dispatch.h | h | 845 B |
torch/include/ATen/ops/_autocast_to_full_precision_native.h | h | 555 B |
torch/include/ATen/ops/_autocast_to_full_precision_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/_autocast_to_reduced_precision.h | h | 530 B |
torch/include/ATen/ops/_autocast_to_reduced_precision_compositeimplicitautograd_dispatch.h | h | 901 B |
torch/include/ATen/ops/_autocast_to_reduced_precision_native.h | h | 611 B |
torch/include/ATen/ops/_autocast_to_reduced_precision_ops.h | h | 1.31 KiB |
torch/include/ATen/ops/_backward.h | h | 509 B |
torch/include/ATen/ops/_backward_compositeimplicitautograd_dispatch.h | h | 932 B |
torch/include/ATen/ops/_backward_native.h | h | 642 B |
torch/include/ATen/ops/_backward_ops.h | h | 1.34 KiB |
torch/include/ATen/ops/_batch_norm_impl_index.h | h | 1.23 KiB |
torch/include/ATen/ops/_batch_norm_impl_index_backward.h | h | 1.49 KiB |
torch/include/ATen/ops/_batch_norm_impl_index_backward_compositeimplicitautograd_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/_batch_norm_impl_index_backward_native.h | h | 949 B |
torch/include/ATen/ops/_batch_norm_impl_index_backward_ops.h | h | 2.37 KiB |
torch/include/ATen/ops/_batch_norm_impl_index_compositeimplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_batch_norm_impl_index_native.h | h | 818 B |
torch/include/ATen/ops/_batch_norm_impl_index_ops.h | h | 1.97 KiB |
torch/include/ATen/ops/_batch_norm_no_update.h | h | 2.76 KiB |
torch/include/ATen/ops/_batch_norm_no_update_compositeexplicitautograd_dispatch.h | h | 1.86 KiB |
torch/include/ATen/ops/_batch_norm_no_update_native.h | h | 1.17 KiB |
torch/include/ATen/ops/_batch_norm_no_update_ops.h | h | 3.53 KiB |
torch/include/ATen/ops/_batch_norm_with_update.h | h | 3.43 KiB |
torch/include/ATen/ops/_batch_norm_with_update_compositeexplicitautograd_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_batch_norm_with_update_cpu_dispatch.h | h | 1.72 KiB |
torch/include/ATen/ops/_batch_norm_with_update_cuda_dispatch.h | h | 1.72 KiB |
torch/include/ATen/ops/_batch_norm_with_update_native.h | h | 2.39 KiB |
torch/include/ATen/ops/_batch_norm_with_update_ops.h | h | 4.76 KiB |
torch/include/ATen/ops/_cast_Byte.h | h | 724 B |
torch/include/ATen/ops/_cast_Byte_compositeimplicitautograd_dispatch.h | h | 816 B |
torch/include/ATen/ops/_cast_Byte_native.h | h | 526 B |
torch/include/ATen/ops/_cast_Byte_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Char.h | h | 724 B |
torch/include/ATen/ops/_cast_Char_compositeimplicitautograd_dispatch.h | h | 816 B |
torch/include/ATen/ops/_cast_Char_native.h | h | 526 B |
torch/include/ATen/ops/_cast_Char_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Double.h | h | 732 B |
torch/include/ATen/ops/_cast_Double_compositeimplicitautograd_dispatch.h | h | 818 B |
torch/include/ATen/ops/_cast_Double_native.h | h | 528 B |
torch/include/ATen/ops/_cast_Double_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Float.h | h | 728 B |
torch/include/ATen/ops/_cast_Float_compositeimplicitautograd_dispatch.h | h | 817 B |
torch/include/ATen/ops/_cast_Float_native.h | h | 527 B |
torch/include/ATen/ops/_cast_Float_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Half.h | h | 724 B |
torch/include/ATen/ops/_cast_Half_compositeimplicitautograd_dispatch.h | h | 816 B |
torch/include/ATen/ops/_cast_Half_native.h | h | 526 B |
torch/include/ATen/ops/_cast_Half_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Int.h | h | 720 B |
torch/include/ATen/ops/_cast_Int_compositeimplicitautograd_dispatch.h | h | 815 B |
torch/include/ATen/ops/_cast_Int_native.h | h | 525 B |
torch/include/ATen/ops/_cast_Int_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Long.h | h | 724 B |
torch/include/ATen/ops/_cast_Long_compositeimplicitautograd_dispatch.h | h | 816 B |
torch/include/ATen/ops/_cast_Long_native.h | h | 526 B |
torch/include/ATen/ops/_cast_Long_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cast_Short.h | h | 728 B |
torch/include/ATen/ops/_cast_Short_compositeimplicitautograd_dispatch.h | h | 817 B |
torch/include/ATen/ops/_cast_Short_native.h | h | 527 B |
torch/include/ATen/ops/_cast_Short_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cdist_backward.h | h | 1.53 KiB |
torch/include/ATen/ops/_cdist_backward_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_cdist_backward_cpu_dispatch.h | h | 834 B |
torch/include/ATen/ops/_cdist_backward_cuda_dispatch.h | h | 836 B |
torch/include/ATen/ops/_cdist_backward_native.h | h | 758 B |
torch/include/ATen/ops/_cdist_backward_ops.h | h | 2.12 KiB |
torch/include/ATen/ops/_cdist_forward.h | h | 1.47 KiB |
torch/include/ATen/ops/_cdist_forward_compositeexplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/_cdist_forward_cpu_dispatch.h | h | 821 B |
torch/include/ATen/ops/_cdist_forward_cuda_dispatch.h | h | 823 B |
torch/include/ATen/ops/_cdist_forward_native.h | h | 732 B |
torch/include/ATen/ops/_cdist_forward_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/_cholesky_solve_helper.h | h | 1.37 KiB |
torch/include/ATen/ops/_cholesky_solve_helper_compositeexplicitautograd_dispatch.h | h | 991 B |
torch/include/ATen/ops/_cholesky_solve_helper_cpu_dispatch.h | h | 793 B |
torch/include/ATen/ops/_cholesky_solve_helper_cuda_dispatch.h | h | 795 B |
torch/include/ATen/ops/_cholesky_solve_helper_native.h | h | 790 B |
torch/include/ATen/ops/_cholesky_solve_helper_ops.h | h | 1.85 KiB |
torch/include/ATen/ops/_choose_qparams_per_tensor.h | h | 812 B |
torch/include/ATen/ops/_choose_qparams_per_tensor_compositeimplicitautograd_dispatch.h | h | 850 B |
torch/include/ATen/ops/_choose_qparams_per_tensor_native.h | h | 560 B |
torch/include/ATen/ops/_choose_qparams_per_tensor_ops.h | h | 1.12 KiB |
torch/include/ATen/ops/_chunk_cat.h | h | 1.30 KiB |
torch/include/ATen/ops/_chunk_cat_compositeexplicitautograd_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/_chunk_cat_cuda_dispatch.h | h | 1012 B |
torch/include/ATen/ops/_chunk_cat_native.h | h | 864 B |
torch/include/ATen/ops/_chunk_cat_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/_coalesce.h | h | 1.05 KiB |
torch/include/ATen/ops/_coalesce_compositeexplicitautograd_dispatch.h | h | 897 B |
torch/include/ATen/ops/_coalesce_native.h | h | 663 B |
torch/include/ATen/ops/_coalesce_ops.h | h | 1.54 KiB |
torch/include/ATen/ops/_coalesced.h | h | 1.18 KiB |
torch/include/ATen/ops/_coalesced_compositeexplicitautograd_dispatch.h | h | 1006 B |
torch/include/ATen/ops/_coalesced_meta_dispatch.h | h | 762 B |
torch/include/ATen/ops/_coalesced_native.h | h | 695 B |
torch/include/ATen/ops/_coalesced_ops.h | h | 2.22 KiB |
torch/include/ATen/ops/_compute_linear_combination.h | h | 1.43 KiB |
torch/include/ATen/ops/_compute_linear_combination_cpu_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_compute_linear_combination_cuda_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_compute_linear_combination_native.h | h | 686 B |
torch/include/ATen/ops/_compute_linear_combination_ops.h | h | 1.86 KiB |
torch/include/ATen/ops/_conj.h | h | 646 B |
torch/include/ATen/ops/_conj_compositeexplicitautograd_dispatch.h | h | 786 B |
torch/include/ATen/ops/_conj_copy.h | h | 1.06 KiB |
torch/include/ATen/ops/_conj_copy_compositeexplicitautograd_dispatch.h | h | 899 B |
torch/include/ATen/ops/_conj_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 817 B |
torch/include/ATen/ops/_conj_copy_native.h | h | 584 B |
torch/include/ATen/ops/_conj_copy_ops.h | h | 1.55 KiB |
torch/include/ATen/ops/_conj_native.h | h | 496 B |
torch/include/ATen/ops/_conj_ops.h | h | 965 B |
torch/include/ATen/ops/_conj_physical.h | h | 1.10 KiB |
torch/include/ATen/ops/_conj_physical_compositeexplicitautograd_dispatch.h | h | 970 B |
torch/include/ATen/ops/_conj_physical_native.h | h | 665 B |
torch/include/ATen/ops/_conj_physical_ops.h | h | 1.57 KiB |
torch/include/ATen/ops/_conv_depthwise2d.h | h | 7.31 KiB |
torch/include/ATen/ops/_conv_depthwise2d_cuda_dispatch.h | h | 2.28 KiB |
torch/include/ATen/ops/_conv_depthwise2d_native.h | h | 964 B |
torch/include/ATen/ops/_conv_depthwise2d_ops.h | h | 2.85 KiB |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr.h | h | 1.52 KiB |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_compositeexplicitautogradnonfunctional_dispatch.h | h | 875 B |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_cpu_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_cuda_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_meta.h | h | 658 B |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_meta_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_native.h | h | 976 B |
torch/include/ATen/ops/_convert_indices_from_coo_to_csr_ops.h | h | 1.89 KiB |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo.h | h | 1.85 KiB |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_compositeexplicitautogradnonfunctional_dispatch.h | h | 923 B |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_cpu_dispatch.h | h | 1.19 KiB |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_cuda_dispatch.h | h | 1.19 KiB |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_meta.h | h | 700 B |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_meta_dispatch.h | h | 1.19 KiB |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_native.h | h | 1.04 KiB |
torch/include/ATen/ops/_convert_indices_from_csr_to_coo_ops.h | h | 2.17 KiB |
torch/include/ATen/ops/_convert_weight_to_int4pack.h | h | 779 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_cpu_dispatch.h | h | 785 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_cuda_dispatch.h | h | 787 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_for_cpu.h | h | 811 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_for_cpu_cpu_dispatch.h | h | 793 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_for_cpu_native.h | h | 543 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_for_cpu_ops.h | h | 1.09 KiB |
torch/include/ATen/ops/_convert_weight_to_int4pack_native.h | h | 544 B |
torch/include/ATen/ops/_convert_weight_to_int4pack_ops.h | h | 1.07 KiB |
torch/include/ATen/ops/_convolution.h | h | 12.76 KiB |
torch/include/ATen/ops/_convolution_compositeexplicitautograd_dispatch.h | h | 2.91 KiB |
torch/include/ATen/ops/_convolution_compositeimplicitautograd_dispatch.h | h | 1.38 KiB |
torch/include/ATen/ops/_convolution_double_backward.h | h | 3.96 KiB |
torch/include/ATen/ops/_convolution_double_backward_compositeimplicitautograd_dispatch.h | h | 1.64 KiB |
torch/include/ATen/ops/_convolution_double_backward_native.h | h | 902 B |
torch/include/ATen/ops/_convolution_double_backward_ops.h | h | 2.31 KiB |
torch/include/ATen/ops/_convolution_mode.h | h | 2.42 KiB |
torch/include/ATen/ops/_convolution_mode_compositeimplicitautograd_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/_convolution_mode_native.h | h | 689 B |
torch/include/ATen/ops/_convolution_mode_ops.h | h | 1.53 KiB |
torch/include/ATen/ops/_convolution_native.h | h | 1.47 KiB |
torch/include/ATen/ops/_convolution_ops.h | h | 4.98 KiB |
torch/include/ATen/ops/_copy_from.h | h | 1.36 KiB |
torch/include/ATen/ops/_copy_from_and_resize.h | h | 1.29 KiB |
torch/include/ATen/ops/_copy_from_and_resize_compositeexplicitautograd_dispatch.h | h | 969 B |
torch/include/ATen/ops/_copy_from_and_resize_native.h | h | 560 B |
torch/include/ATen/ops/_copy_from_and_resize_ops.h | h | 1.77 KiB |
torch/include/ATen/ops/_copy_from_compositeexplicitautograd_dispatch.h | h | 991 B |
torch/include/ATen/ops/_copy_from_native.h | h | 568 B |
torch/include/ATen/ops/_copy_from_ops.h | h | 1.84 KiB |
torch/include/ATen/ops/_cslt_compress.h | h | 679 B |
torch/include/ATen/ops/_cslt_compress_cuda_dispatch.h | h | 754 B |
torch/include/ATen/ops/_cslt_compress_native.h | h | 506 B |
torch/include/ATen/ops/_cslt_compress_ops.h | h | 989 B |
torch/include/ATen/ops/_cslt_sparse_mm.h | h | 1.09 KiB |
torch/include/ATen/ops/_cslt_sparse_mm_cuda_dispatch.h | h | 986 B |
torch/include/ATen/ops/_cslt_sparse_mm_native.h | h | 738 B |
torch/include/ATen/ops/_cslt_sparse_mm_ops.h | h | 1.63 KiB |
torch/include/ATen/ops/_cslt_sparse_mm_search.h | h | 1.07 KiB |
torch/include/ATen/ops/_cslt_sparse_mm_search_cuda_dispatch.h | h | 972 B |
torch/include/ATen/ops/_cslt_sparse_mm_search_native.h | h | 724 B |
torch/include/ATen/ops/_cslt_sparse_mm_search_ops.h | h | 1.59 KiB |
torch/include/ATen/ops/_ctc_loss.h | h | 3.86 KiB |
torch/include/ATen/ops/_ctc_loss_backward.h | h | 3.25 KiB |
torch/include/ATen/ops/_ctc_loss_backward_compositeexplicitautograd_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_ctc_loss_backward_cpu_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/_ctc_loss_backward_cuda_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/_ctc_loss_backward_native.h | h | 1.63 KiB |
torch/include/ATen/ops/_ctc_loss_backward_ops.h | h | 4.33 KiB |
torch/include/ATen/ops/_ctc_loss_compositeexplicitautograd_dispatch.h | h | 1.76 KiB |
torch/include/ATen/ops/_ctc_loss_cpu_dispatch.h | h | 1.11 KiB |
torch/include/ATen/ops/_ctc_loss_cuda_dispatch.h | h | 1.11 KiB |
torch/include/ATen/ops/_ctc_loss_meta_dispatch.h | h | 912 B |
torch/include/ATen/ops/_ctc_loss_native.h | h | 1.84 KiB |
torch/include/ATen/ops/_ctc_loss_ops.h | h | 4.91 KiB |
torch/include/ATen/ops/_cudnn_ctc_loss.h | h | 2.89 KiB |
torch/include/ATen/ops/_cudnn_ctc_loss_compositeexplicitautograd_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_cudnn_ctc_loss_cuda_dispatch.h | h | 1.15 KiB |
torch/include/ATen/ops/_cudnn_ctc_loss_native.h | h | 1.19 KiB |
torch/include/ATen/ops/_cudnn_ctc_loss_ops.h | h | 3.91 KiB |
torch/include/ATen/ops/_cudnn_init_dropout_state.h | h | 2.22 KiB |
torch/include/ATen/ops/_cudnn_init_dropout_state_compositeexplicitautograd_dispatch.h | h | 979 B |
torch/include/ATen/ops/_cudnn_init_dropout_state_cuda_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_cudnn_init_dropout_state_native.h | h | 821 B |
torch/include/ATen/ops/_cudnn_init_dropout_state_ops.h | h | 2.29 KiB |
torch/include/ATen/ops/_cudnn_rnn.h | h | 13.10 KiB |
torch/include/ATen/ops/_cudnn_rnn_backward.h | h | 16.14 KiB |
torch/include/ATen/ops/_cudnn_rnn_backward_compositeexplicitautograd_dispatch.h | h | 3.62 KiB |
torch/include/ATen/ops/_cudnn_rnn_backward_cuda_dispatch.h | h | 2.10 KiB |
torch/include/ATen/ops/_cudnn_rnn_backward_native.h | h | 1.87 KiB |
torch/include/ATen/ops/_cudnn_rnn_backward_ops.h | h | 5.84 KiB |
torch/include/ATen/ops/_cudnn_rnn_compositeexplicitautograd_dispatch.h | h | 3.12 KiB |
torch/include/ATen/ops/_cudnn_rnn_cuda_dispatch.h | h | 1.66 KiB |
torch/include/ATen/ops/_cudnn_rnn_flatten_weight.h | h | 7.47 KiB |
torch/include/ATen/ops/_cudnn_rnn_flatten_weight_compositeexplicitautograd_dispatch.h | h | 1.75 KiB |
torch/include/ATen/ops/_cudnn_rnn_flatten_weight_cuda_dispatch.h | h | 1.15 KiB |
torch/include/ATen/ops/_cudnn_rnn_flatten_weight_native.h | h | 949 B |
torch/include/ATen/ops/_cudnn_rnn_flatten_weight_ops.h | h | 2.74 KiB |
torch/include/ATen/ops/_cudnn_rnn_native.h | h | 1.53 KiB |
torch/include/ATen/ops/_cudnn_rnn_ops.h | h | 4.78 KiB |
torch/include/ATen/ops/_cufft_clear_plan_cache.h | h | 728 B |
torch/include/ATen/ops/_cufft_clear_plan_cache_compositeimplicitautograd_dispatch.h | h | 803 B |
torch/include/ATen/ops/_cufft_clear_plan_cache_native.h | h | 513 B |
torch/include/ATen/ops/_cufft_clear_plan_cache_ops.h | h | 1011 B |
torch/include/ATen/ops/_cufft_get_plan_cache_max_size.h | h | 760 B |
torch/include/ATen/ops/_cufft_get_plan_cache_max_size_compositeimplicitautograd_dispatch.h | h | 813 B |
torch/include/ATen/ops/_cufft_get_plan_cache_max_size_native.h | h | 523 B |
torch/include/ATen/ops/_cufft_get_plan_cache_max_size_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/_cufft_get_plan_cache_size.h | h | 744 B |
torch/include/ATen/ops/_cufft_get_plan_cache_size_compositeimplicitautograd_dispatch.h | h | 809 B |
torch/include/ATen/ops/_cufft_get_plan_cache_size_native.h | h | 519 B |
torch/include/ATen/ops/_cufft_get_plan_cache_size_ops.h | h | 1.01 KiB |
torch/include/ATen/ops/_cufft_set_plan_cache_max_size.h | h | 798 B |
torch/include/ATen/ops/_cufft_set_plan_cache_max_size_compositeimplicitautograd_dispatch.h | h | 828 B |
torch/include/ATen/ops/_cufft_set_plan_cache_max_size_native.h | h | 538 B |
torch/include/ATen/ops/_cufft_set_plan_cache_max_size_ops.h | h | 1.07 KiB |
torch/include/ATen/ops/_cummax_helper.h | h | 792 B |
torch/include/ATen/ops/_cummax_helper_cpu_dispatch.h | h | 801 B |
torch/include/ATen/ops/_cummax_helper_cuda_dispatch.h | h | 803 B |
torch/include/ATen/ops/_cummax_helper_native.h | h | 675 B |
torch/include/ATen/ops/_cummax_helper_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/_cummin_helper.h | h | 792 B |
torch/include/ATen/ops/_cummin_helper_cpu_dispatch.h | h | 801 B |
torch/include/ATen/ops/_cummin_helper_cuda_dispatch.h | h | 803 B |
torch/include/ATen/ops/_cummin_helper_native.h | h | 675 B |
torch/include/ATen/ops/_cummin_helper_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/_debug_has_internal_overlap.h | h | 722 B |
torch/include/ATen/ops/_debug_has_internal_overlap_compositeimplicitautograd_dispatch.h | h | 805 B |
torch/include/ATen/ops/_debug_has_internal_overlap_native.h | h | 515 B |
torch/include/ATen/ops/_debug_has_internal_overlap_ops.h | h | 1013 B |
torch/include/ATen/ops/_dimI.h | h | 505 B |
torch/include/ATen/ops/_dimI_native.h | h | 505 B |
torch/include/ATen/ops/_dimI_ops.h | h | 947 B |
torch/include/ATen/ops/_dimV.h | h | 505 B |
torch/include/ATen/ops/_dimV_native.h | h | 504 B |
torch/include/ATen/ops/_dimV_ops.h | h | 947 B |
torch/include/ATen/ops/_dim_arange.h | h | 691 B |
torch/include/ATen/ops/_dim_arange_compositeimplicitautograd_dispatch.h | h | 805 B |
torch/include/ATen/ops/_dim_arange_native.h | h | 515 B |
torch/include/ATen/ops/_dim_arange_ops.h | h | 1021 B |
torch/include/ATen/ops/_dirichlet_grad.h | h | 1.36 KiB |
torch/include/ATen/ops/_dirichlet_grad_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_dirichlet_grad_cpu_dispatch.h | h | 801 B |
torch/include/ATen/ops/_dirichlet_grad_cuda_dispatch.h | h | 803 B |
torch/include/ATen/ops/_dirichlet_grad_native.h | h | 814 B |
torch/include/ATen/ops/_dirichlet_grad_ops.h | h | 1.90 KiB |
torch/include/ATen/ops/_efficient_attention_backward.h | h | 5.87 KiB |
torch/include/ATen/ops/_efficient_attention_backward_cuda_dispatch.h | h | 2.19 KiB |
torch/include/ATen/ops/_efficient_attention_backward_native.h | h | 1.18 KiB |
torch/include/ATen/ops/_efficient_attention_backward_ops.h | h | 3.10 KiB |
torch/include/ATen/ops/_efficient_attention_forward.h | h | 5.30 KiB |
torch/include/ATen/ops/_efficient_attention_forward_cuda_dispatch.h | h | 1.94 KiB |
torch/include/ATen/ops/_efficient_attention_forward_native.h | h | 1.06 KiB |
torch/include/ATen/ops/_efficient_attention_forward_ops.h | h | 2.81 KiB |
torch/include/ATen/ops/_efficientzerotensor.h | h | 5.99 KiB |
torch/include/ATen/ops/_efficientzerotensor_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/_efficientzerotensor_cpu_dispatch.h | h | 1.29 KiB |
torch/include/ATen/ops/_efficientzerotensor_cuda_dispatch.h | h | 1.29 KiB |
torch/include/ATen/ops/_efficientzerotensor_meta_dispatch.h | h | 1.29 KiB |
torch/include/ATen/ops/_efficientzerotensor_native.h | h | 1.20 KiB |
torch/include/ATen/ops/_efficientzerotensor_ops.h | h | 2.10 KiB |
torch/include/ATen/ops/_embedding_bag.h | h | 3.12 KiB |
torch/include/ATen/ops/_embedding_bag_backward.h | h | 3.49 KiB |
torch/include/ATen/ops/_embedding_bag_backward_cpu_dispatch.h | h | 1.42 KiB |
torch/include/ATen/ops/_embedding_bag_backward_cuda_dispatch.h | h | 1.42 KiB |
torch/include/ATen/ops/_embedding_bag_backward_native.h | h | 830 B |
torch/include/ATen/ops/_embedding_bag_backward_ops.h | h | 1.97 KiB |
torch/include/ATen/ops/_embedding_bag_compositeexplicitautograd_dispatch.h | h | 1.54 KiB |
torch/include/ATen/ops/_embedding_bag_cpu_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_embedding_bag_cuda_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_embedding_bag_dense_backward.h | h | 9.14 KiB |
torch/include/ATen/ops/_embedding_bag_dense_backward_compositeexplicitautograd_dispatch.h | h | 2.16 KiB |
torch/include/ATen/ops/_embedding_bag_dense_backward_cpu_dispatch.h | h | 1.35 KiB |
torch/include/ATen/ops/_embedding_bag_dense_backward_cuda_dispatch.h | h | 1.35 KiB |
torch/include/ATen/ops/_embedding_bag_dense_backward_native.h | h | 1.47 KiB |
torch/include/ATen/ops/_embedding_bag_dense_backward_ops.h | h | 3.35 KiB |
torch/include/ATen/ops/_embedding_bag_forward_only.h | h | 3.25 KiB |
torch/include/ATen/ops/_embedding_bag_forward_only_compositeexplicitautograd_dispatch.h | h | 1.57 KiB |
torch/include/ATen/ops/_embedding_bag_forward_only_cpu_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/_embedding_bag_forward_only_cuda_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/_embedding_bag_forward_only_native.h | h | 1.56 KiB |
torch/include/ATen/ops/_embedding_bag_forward_only_ops.h | h | 3.62 KiB |
torch/include/ATen/ops/_embedding_bag_native.h | h | 1.52 KiB |
torch/include/ATen/ops/_embedding_bag_ops.h | h | 3.54 KiB |
torch/include/ATen/ops/_embedding_bag_per_sample_weights_backward.h | h | 2.26 KiB |
torch/include/ATen/ops/_embedding_bag_per_sample_weights_backward_compositeexplicitautograd_dispatch.h | h | 1.23 KiB |
torch/include/ATen/ops/_embedding_bag_per_sample_weights_backward_cpu_dispatch.h | h | 931 B |
torch/include/ATen/ops/_embedding_bag_per_sample_weights_backward_cuda_dispatch.h | h | 933 B |
torch/include/ATen/ops/_embedding_bag_per_sample_weights_backward_native.h | h | 1.17 KiB |
torch/include/ATen/ops/_embedding_bag_per_sample_weights_backward_ops.h | h | 2.70 KiB |
torch/include/ATen/ops/_embedding_bag_sparse_backward.h | h | 3.20 KiB |
torch/include/ATen/ops/_embedding_bag_sparse_backward_compositeimplicitautograd_dispatch.h | h | 1.38 KiB |
torch/include/ATen/ops/_embedding_bag_sparse_backward_native.h | h | 788 B |
torch/include/ATen/ops/_embedding_bag_sparse_backward_ops.h | h | 1.83 KiB |
torch/include/ATen/ops/_empty_affine_quantized.h | h | 9.09 KiB |
torch/include/ATen/ops/_empty_affine_quantized_compositeexplicitautograd_dispatch.h | h | 1.49 KiB |
torch/include/ATen/ops/_empty_affine_quantized_cpu_dispatch.h | h | 1.70 KiB |
torch/include/ATen/ops/_empty_affine_quantized_native.h | h | 1.30 KiB |
torch/include/ATen/ops/_empty_affine_quantized_ops.h | h | 2.70 KiB |
torch/include/ATen/ops/_empty_per_channel_affine_quantized.h | h | 10.37 KiB |
torch/include/ATen/ops/_empty_per_channel_affine_quantized_compositeexplicitautograd_dispatch.h | h | 1.68 KiB |
torch/include/ATen/ops/_empty_per_channel_affine_quantized_cpu_dispatch.h | h | 1.89 KiB |
torch/include/ATen/ops/_empty_per_channel_affine_quantized_native.h | h | 1.44 KiB |
torch/include/ATen/ops/_empty_per_channel_affine_quantized_ops.h | h | 3.01 KiB |
torch/include/ATen/ops/_euclidean_dist.h | h | 1.20 KiB |
torch/include/ATen/ops/_euclidean_dist_compositeexplicitautograd_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_euclidean_dist_native.h | h | 636 B |
torch/include/ATen/ops/_euclidean_dist_ops.h | h | 1.72 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine.h | h | 2.22 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_backward.h | h | 1.19 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_backward_cpu_dispatch.h | h | 983 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_backward_cuda_dispatch.h | h | 985 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_backward_native.h | h | 737 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_backward_ops.h | h | 1.70 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_compositeexplicitautograd_dispatch.h | h | 1.20 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_cpu_dispatch.h | h | 913 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_cuda_dispatch.h | h | 915 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_native.h | h | 912 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_channel_affine_ops.h | h | 2.58 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine.h | h | 2.13 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_backward.h | h | 1.15 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_backward_cpu_dispatch.h | h | 968 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_backward_cuda_dispatch.h | h | 970 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_backward_native.h | h | 722 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_backward_ops.h | h | 1.65 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_compositeexplicitautograd_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_cpu_dispatch.h | h | 898 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_cuda_dispatch.h | h | 900 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_native.h | h | 882 B |
torch/include/ATen/ops/_fake_quantize_learnable_per_tensor_affine_ops.h | h | 2.48 KiB |
torch/include/ATen/ops/_fake_quantize_per_tensor_affine_cachemask_tensor_qparams.h | h | 2.57 KiB |
torch/include/ATen/ops/_fake_quantize_per_tensor_affine_cachemask_tensor_qparams_compositeexplicitautograd_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/_fake_quantize_per_tensor_affine_cachemask_tensor_qparams_cpu_dispatch.h | h | 953 B |
torch/include/ATen/ops/_fake_quantize_per_tensor_affine_cachemask_tensor_qparams_cuda_dispatch.h | h | 955 B |
torch/include/ATen/ops/_fake_quantize_per_tensor_affine_cachemask_tensor_qparams_native.h | h | 1018 B |
torch/include/ATen/ops/_fake_quantize_per_tensor_affine_cachemask_tensor_qparams_ops.h | h | 2.93 KiB |
torch/include/ATen/ops/_fft_c2c.h | h | 4.42 KiB |
torch/include/ATen/ops/_fft_c2c_cpu_dispatch.h | h | 1.47 KiB |
torch/include/ATen/ops/_fft_c2c_cuda_dispatch.h | h | 1.48 KiB |
torch/include/ATen/ops/_fft_c2c_native.h | h | 970 B |
torch/include/ATen/ops/_fft_c2c_ops.h | h | 1.94 KiB |
torch/include/ATen/ops/_fft_c2r.h | h | 4.47 KiB |
torch/include/ATen/ops/_fft_c2r_cpu_dispatch.h | h | 1.53 KiB |
torch/include/ATen/ops/_fft_c2r_cuda_dispatch.h | h | 1.53 KiB |
torch/include/ATen/ops/_fft_c2r_native.h | h | 1006 B |
torch/include/ATen/ops/_fft_c2r_ops.h | h | 1.99 KiB |
torch/include/ATen/ops/_fft_r2c.h | h | 1.43 KiB |
torch/include/ATen/ops/_fft_r2c_cpu_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_fft_r2c_cuda_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_fft_r2c_native.h | h | 974 B |
torch/include/ATen/ops/_fft_r2c_ops.h | h | 1.92 KiB |
torch/include/ATen/ops/_fill_mem_eff_dropout_mask.h | h | 843 B |
torch/include/ATen/ops/_fill_mem_eff_dropout_mask_cuda_dispatch.h | h | 810 B |
torch/include/ATen/ops/_fill_mem_eff_dropout_mask_meta_dispatch.h | h | 810 B |
torch/include/ATen/ops/_fill_mem_eff_dropout_mask_native.h | h | 562 B |
torch/include/ATen/ops/_fill_mem_eff_dropout_mask_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/_flash_attention_backward.h | h | 5.05 KiB |
torch/include/ATen/ops/_flash_attention_backward_cuda_dispatch.h | h | 1.86 KiB |
torch/include/ATen/ops/_flash_attention_backward_native.h | h | 1.01 KiB |
torch/include/ATen/ops/_flash_attention_backward_ops.h | h | 2.60 KiB |
torch/include/ATen/ops/_flash_attention_forward.h | h | 5.27 KiB |
torch/include/ATen/ops/_flash_attention_forward_cuda_dispatch.h | h | 1.93 KiB |
torch/include/ATen/ops/_flash_attention_forward_native.h | h | 1.05 KiB |
torch/include/ATen/ops/_flash_attention_forward_ops.h | h | 2.76 KiB |
torch/include/ATen/ops/_foobar.h | h | 1.35 KiB |
torch/include/ATen/ops/_foobar_compositeexplicitautograd_dispatch.h | h | 974 B |
torch/include/ATen/ops/_foobar_cpu_dispatch.h | h | 792 B |
torch/include/ATen/ops/_foobar_native.h | h | 658 B |
torch/include/ATen/ops/_foobar_ops.h | h | 1.79 KiB |
torch/include/ATen/ops/_foreach_abs.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_abs_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_abs_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_abs_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_abs_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_acos.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_acos_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_acos_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_acos_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_acos_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_add.h | h | 4.79 KiB |
torch/include/ATen/ops/_foreach_add_compositeexplicitautograd_dispatch.h | h | 2.43 KiB |
torch/include/ATen/ops/_foreach_add_cuda_dispatch.h | h | 1.48 KiB |
torch/include/ATen/ops/_foreach_add_native.h | h | 2.92 KiB |
torch/include/ATen/ops/_foreach_add_ops.h | h | 8.75 KiB |
torch/include/ATen/ops/_foreach_addcdiv.h | h | 4.82 KiB |
torch/include/ATen/ops/_foreach_addcdiv_compositeexplicitautograd_dispatch.h | h | 2.48 KiB |
torch/include/ATen/ops/_foreach_addcdiv_cuda_dispatch.h | h | 1.52 KiB |
torch/include/ATen/ops/_foreach_addcdiv_native.h | h | 2.83 KiB |
torch/include/ATen/ops/_foreach_addcdiv_ops.h | h | 7.91 KiB |
torch/include/ATen/ops/_foreach_addcmul.h | h | 4.82 KiB |
torch/include/ATen/ops/_foreach_addcmul_compositeexplicitautograd_dispatch.h | h | 2.48 KiB |
torch/include/ATen/ops/_foreach_addcmul_cuda_dispatch.h | h | 1.52 KiB |
torch/include/ATen/ops/_foreach_addcmul_native.h | h | 2.83 KiB |
torch/include/ATen/ops/_foreach_addcmul_ops.h | h | 7.91 KiB |
torch/include/ATen/ops/_foreach_asin.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_asin_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_asin_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_asin_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_asin_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_atan.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_atan_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_atan_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_atan_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_atan_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_ceil.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_ceil_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_ceil_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_ceil_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_ceil_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_clamp_max.h | h | 3.65 KiB |
torch/include/ATen/ops/_foreach_clamp_max_compositeexplicitautograd_dispatch.h | h | 1.92 KiB |
torch/include/ATen/ops/_foreach_clamp_max_cuda_dispatch.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_clamp_max_native.h | h | 2.19 KiB |
torch/include/ATen/ops/_foreach_clamp_max_ops.h | h | 6.46 KiB |
torch/include/ATen/ops/_foreach_clamp_min.h | h | 3.65 KiB |
torch/include/ATen/ops/_foreach_clamp_min_compositeexplicitautograd_dispatch.h | h | 1.92 KiB |
torch/include/ATen/ops/_foreach_clamp_min_cuda_dispatch.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_clamp_min_native.h | h | 2.19 KiB |
torch/include/ATen/ops/_foreach_clamp_min_ops.h | h | 6.46 KiB |
torch/include/ATen/ops/_foreach_copy.h | h | 1.63 KiB |
torch/include/ATen/ops/_foreach_copy_compositeexplicitautograd_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_foreach_copy_cuda_dispatch.h | h | 788 B |
torch/include/ATen/ops/_foreach_copy_native.h | h | 917 B |
torch/include/ATen/ops/_foreach_copy_ops.h | h | 2.48 KiB |
torch/include/ATen/ops/_foreach_cos.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_cos_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_cos_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_cos_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_cos_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_cosh.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_cosh_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_cosh_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_cosh_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_cosh_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_div.h | h | 4.39 KiB |
torch/include/ATen/ops/_foreach_div_compositeexplicitautograd_dispatch.h | h | 2.21 KiB |
torch/include/ATen/ops/_foreach_div_cuda_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/_foreach_div_native.h | h | 2.65 KiB |
torch/include/ATen/ops/_foreach_div_ops.h | h | 8.22 KiB |
torch/include/ATen/ops/_foreach_erf.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_erf_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_erf_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_erf_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_erf_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_erfc.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_erfc_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_erfc_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_erfc_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_erfc_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_exp.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_exp_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_exp_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_exp_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_exp_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_expm1.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_expm1_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_expm1_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_expm1_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_expm1_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_floor.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_floor_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_floor_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_floor_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_floor_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_frac.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_frac_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_frac_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_frac_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_frac_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_lerp.h | h | 4.12 KiB |
torch/include/ATen/ops/_foreach_lerp_compositeexplicitautograd_dispatch.h | h | 2.15 KiB |
torch/include/ATen/ops/_foreach_lerp_cuda_dispatch.h | h | 1.35 KiB |
torch/include/ATen/ops/_foreach_lerp_native.h | h | 2.44 KiB |
torch/include/ATen/ops/_foreach_lerp_ops.h | h | 7.08 KiB |
torch/include/ATen/ops/_foreach_lgamma.h | h | 1.25 KiB |
torch/include/ATen/ops/_foreach_lgamma_compositeexplicitautograd_dispatch.h | h | 1019 B |
torch/include/ATen/ops/_foreach_lgamma_cuda_dispatch.h | h | 820 B |
torch/include/ATen/ops/_foreach_lgamma_native.h | h | 824 B |
torch/include/ATen/ops/_foreach_lgamma_ops.h | h | 2.08 KiB |
torch/include/ATen/ops/_foreach_log.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_log10.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_log10_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_log10_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_log10_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_log10_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_log1p.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_log1p_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_log1p_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_log1p_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_log1p_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_log2.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_log2_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_log2_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_log2_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_log2_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_log_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_log_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_log_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_log_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_max.h | h | 1.06 KiB |
torch/include/ATen/ops/_foreach_max_compositeexplicitautograd_dispatch.h | h | 955 B |
torch/include/ATen/ops/_foreach_max_cuda_dispatch.h | h | 762 B |
torch/include/ATen/ops/_foreach_max_native.h | h | 683 B |
torch/include/ATen/ops/_foreach_max_ops.h | h | 1.56 KiB |
torch/include/ATen/ops/_foreach_maximum.h | h | 3.58 KiB |
torch/include/ATen/ops/_foreach_maximum_compositeexplicitautograd_dispatch.h | h | 1.89 KiB |
torch/include/ATen/ops/_foreach_maximum_cuda_dispatch.h | h | 1.22 KiB |
torch/include/ATen/ops/_foreach_maximum_native.h | h | 2.19 KiB |
torch/include/ATen/ops/_foreach_maximum_ops.h | h | 6.40 KiB |
torch/include/ATen/ops/_foreach_minimum.h | h | 3.58 KiB |
torch/include/ATen/ops/_foreach_minimum_compositeexplicitautograd_dispatch.h | h | 1.89 KiB |
torch/include/ATen/ops/_foreach_minimum_cuda_dispatch.h | h | 1.22 KiB |
torch/include/ATen/ops/_foreach_minimum_native.h | h | 2.19 KiB |
torch/include/ATen/ops/_foreach_minimum_ops.h | h | 6.40 KiB |
torch/include/ATen/ops/_foreach_mul.h | h | 4.39 KiB |
torch/include/ATen/ops/_foreach_mul_compositeexplicitautograd_dispatch.h | h | 2.21 KiB |
torch/include/ATen/ops/_foreach_mul_cuda_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/_foreach_mul_native.h | h | 2.65 KiB |
torch/include/ATen/ops/_foreach_mul_ops.h | h | 8.22 KiB |
torch/include/ATen/ops/_foreach_neg.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_neg_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_neg_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_neg_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_neg_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_norm.h | h | 1.48 KiB |
torch/include/ATen/ops/_foreach_norm_compositeexplicitautograd_dispatch.h | h | 1.15 KiB |
torch/include/ATen/ops/_foreach_norm_cuda_dispatch.h | h | 843 B |
torch/include/ATen/ops/_foreach_norm_native.h | h | 916 B |
torch/include/ATen/ops/_foreach_norm_ops.h | h | 2.03 KiB |
torch/include/ATen/ops/_foreach_pow.h | h | 3.76 KiB |
torch/include/ATen/ops/_foreach_pow_compositeexplicitautograd_dispatch.h | h | 1.97 KiB |
torch/include/ATen/ops/_foreach_pow_cuda_dispatch.h | h | 1.31 KiB |
torch/include/ATen/ops/_foreach_pow_native.h | h | 2.38 KiB |
torch/include/ATen/ops/_foreach_pow_ops.h | h | 7.04 KiB |
torch/include/ATen/ops/_foreach_reciprocal.h | h | 1.30 KiB |
torch/include/ATen/ops/_foreach_reciprocal_compositeexplicitautograd_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_foreach_reciprocal_cuda_dispatch.h | h | 828 B |
torch/include/ATen/ops/_foreach_reciprocal_native.h | h | 844 B |
torch/include/ATen/ops/_foreach_reciprocal_ops.h | h | 2.12 KiB |
torch/include/ATen/ops/_foreach_round.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_round_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_round_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_round_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_round_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_rsqrt.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_rsqrt_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_rsqrt_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_rsqrt_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_rsqrt_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_sigmoid.h | h | 1.26 KiB |
torch/include/ATen/ops/_foreach_sigmoid_compositeexplicitautograd_dispatch.h | h | 1023 B |
torch/include/ATen/ops/_foreach_sigmoid_cuda_dispatch.h | h | 822 B |
torch/include/ATen/ops/_foreach_sigmoid_native.h | h | 829 B |
torch/include/ATen/ops/_foreach_sigmoid_ops.h | h | 2.09 KiB |
torch/include/ATen/ops/_foreach_sign.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_sign_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_sign_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_sign_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_sign_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_sin.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_sin_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_sin_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_sin_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_sin_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_sinh.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_sinh_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_sinh_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_sinh_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_sinh_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_sqrt.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_sqrt_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_sqrt_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_sqrt_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_sqrt_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_sub.h | h | 3.64 KiB |
torch/include/ATen/ops/_foreach_sub_compositeexplicitautograd_dispatch.h | h | 1.95 KiB |
torch/include/ATen/ops/_foreach_sub_cuda_dispatch.h | h | 1.25 KiB |
torch/include/ATen/ops/_foreach_sub_native.h | h | 2.24 KiB |
torch/include/ATen/ops/_foreach_sub_ops.h | h | 6.56 KiB |
torch/include/ATen/ops/_foreach_tan.h | h | 1.21 KiB |
torch/include/ATen/ops/_foreach_tan_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_foreach_tan_cuda_dispatch.h | h | 814 B |
torch/include/ATen/ops/_foreach_tan_native.h | h | 809 B |
torch/include/ATen/ops/_foreach_tan_ops.h | h | 2.06 KiB |
torch/include/ATen/ops/_foreach_tanh.h | h | 1.23 KiB |
torch/include/ATen/ops/_foreach_tanh_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_tanh_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_foreach_tanh_native.h | h | 814 B |
torch/include/ATen/ops/_foreach_tanh_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_trunc.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_trunc_compositeexplicitautograd_dispatch.h | h | 1015 B |
torch/include/ATen/ops/_foreach_trunc_cuda_dispatch.h | h | 818 B |
torch/include/ATen/ops/_foreach_trunc_native.h | h | 819 B |
torch/include/ATen/ops/_foreach_trunc_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_foreach_zero.h | h | 1.24 KiB |
torch/include/ATen/ops/_foreach_zero_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_foreach_zero_cuda_dispatch.h | h | 743 B |
torch/include/ATen/ops/_foreach_zero_native.h | h | 719 B |
torch/include/ATen/ops/_foreach_zero_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_functional_assert_async.h | h | 840 B |
torch/include/ATen/ops/_functional_assert_async_cpu_dispatch.h | h | 820 B |
torch/include/ATen/ops/_functional_assert_async_native.h | h | 582 B |
torch/include/ATen/ops/_functional_assert_async_ops.h | h | 1.19 KiB |
torch/include/ATen/ops/_functional_assert_scalar.h | h | 836 B |
torch/include/ATen/ops/_functional_assert_scalar_compositeexplicitautograd_dispatch.h | h | 865 B |
torch/include/ATen/ops/_functional_assert_scalar_native.h | h | 575 B |
torch/include/ATen/ops/_functional_assert_scalar_ops.h | h | 1.18 KiB |
torch/include/ATen/ops/_functional_sym_constrain_range.h | h | 893 B |
torch/include/ATen/ops/_functional_sym_constrain_range_compositeexplicitautograd_dispatch.h | h | 902 B |
torch/include/ATen/ops/_functional_sym_constrain_range_for_size.h | h | 929 B |
torch/include/ATen/ops/_functional_sym_constrain_range_for_size_compositeexplicitautograd_dispatch.h | h | 911 B |
torch/include/ATen/ops/_functional_sym_constrain_range_for_size_native.h | h | 621 B |
torch/include/ATen/ops/_functional_sym_constrain_range_for_size_ops.h | h | 1.32 KiB |
torch/include/ATen/ops/_functional_sym_constrain_range_native.h | h | 612 B |
torch/include/ATen/ops/_functional_sym_constrain_range_ops.h | h | 1.30 KiB |
torch/include/ATen/ops/_fused_adagrad.h | h | 3.52 KiB |
torch/include/ATen/ops/_fused_adagrad_compositeexplicitautograd_dispatch.h | h | 1.77 KiB |
torch/include/ATen/ops/_fused_adagrad_cpu_dispatch.h | h | 996 B |
torch/include/ATen/ops/_fused_adagrad_native.h | h | 1.47 KiB |
torch/include/ATen/ops/_fused_adagrad_ops.h | h | 4.90 KiB |
torch/include/ATen/ops/_fused_adam.h | h | 8.25 KiB |
torch/include/ATen/ops/_fused_adam_compositeexplicitautograd_dispatch.h | h | 3.37 KiB |
torch/include/ATen/ops/_fused_adam_cpu_dispatch.h | h | 1.44 KiB |
torch/include/ATen/ops/_fused_adam_cuda_dispatch.h | h | 1.44 KiB |
torch/include/ATen/ops/_fused_adam_native.h | h | 3.87 KiB |
torch/include/ATen/ops/_fused_adam_ops.h | h | 11.38 KiB |
torch/include/ATen/ops/_fused_adamw.h | h | 8.27 KiB |
torch/include/ATen/ops/_fused_adamw_compositeexplicitautograd_dispatch.h | h | 3.38 KiB |
torch/include/ATen/ops/_fused_adamw_cpu_dispatch.h | h | 1.44 KiB |
torch/include/ATen/ops/_fused_adamw_cuda_dispatch.h | h | 1.45 KiB |
torch/include/ATen/ops/_fused_adamw_native.h | h | 3.88 KiB |
torch/include/ATen/ops/_fused_adamw_ops.h | h | 11.40 KiB |
torch/include/ATen/ops/_fused_dropout.h | h | 1.63 KiB |
torch/include/ATen/ops/_fused_dropout_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/_fused_dropout_cuda_dispatch.h | h | 845 B |
torch/include/ATen/ops/_fused_dropout_native.h | h | 787 B |
torch/include/ATen/ops/_fused_dropout_ops.h | h | 2.17 KiB |
torch/include/ATen/ops/_fused_moving_avg_obs_fq_helper.h | h | 4.68 KiB |
torch/include/ATen/ops/_fused_moving_avg_obs_fq_helper_compositeexplicitautograd_dispatch.h | h | 2.06 KiB |
torch/include/ATen/ops/_fused_moving_avg_obs_fq_helper_cpu_dispatch.h | h | 1.07 KiB |
torch/include/ATen/ops/_fused_moving_avg_obs_fq_helper_cuda_dispatch.h | h | 1.07 KiB |
torch/include/ATen/ops/_fused_moving_avg_obs_fq_helper_native.h | h | 2.14 KiB |
torch/include/ATen/ops/_fused_moving_avg_obs_fq_helper_ops.h | h | 5.76 KiB |
torch/include/ATen/ops/_fused_sdp_choice.h | h | 1.07 KiB |
torch/include/ATen/ops/_fused_sdp_choice_cpu_dispatch.h | h | 965 B |
torch/include/ATen/ops/_fused_sdp_choice_cuda_dispatch.h | h | 967 B |
torch/include/ATen/ops/_fused_sdp_choice_meta_dispatch.h | h | 967 B |
torch/include/ATen/ops/_fused_sdp_choice_native.h | h | 1.26 KiB |
torch/include/ATen/ops/_fused_sdp_choice_ops.h | h | 1.57 KiB |
torch/include/ATen/ops/_fused_sgd.h | h | 7.10 KiB |
torch/include/ATen/ops/_fused_sgd_compositeexplicitautograd_dispatch.h | h | 2.91 KiB |
torch/include/ATen/ops/_fused_sgd_cpu_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/_fused_sgd_cuda_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/_fused_sgd_native.h | h | 3.29 KiB |
torch/include/ATen/ops/_fused_sgd_ops.h | h | 9.71 KiB |
torch/include/ATen/ops/_fw_primal.h | h | 510 B |
torch/include/ATen/ops/_fw_primal_compositeexplicitautograd_dispatch.h | h | 806 B |
torch/include/ATen/ops/_fw_primal_copy.h | h | 1.20 KiB |
torch/include/ATen/ops/_fw_primal_copy_compositeexplicitautograd_dispatch.h | h | 939 B |
torch/include/ATen/ops/_fw_primal_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 837 B |
torch/include/ATen/ops/_fw_primal_copy_native.h | h | 624 B |
torch/include/ATen/ops/_fw_primal_copy_ops.h | h | 1.68 KiB |
torch/include/ATen/ops/_fw_primal_native.h | h | 516 B |
torch/include/ATen/ops/_fw_primal_ops.h | h | 1.01 KiB |
torch/include/ATen/ops/_gather_sparse_backward.h | h | 830 B |
torch/include/ATen/ops/_gather_sparse_backward_compositeimplicitautograd_dispatch.h | h | 868 B |
torch/include/ATen/ops/_gather_sparse_backward_native.h | h | 578 B |
torch/include/ATen/ops/_gather_sparse_backward_ops.h | h | 1.20 KiB |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback.h | h | 1.91 KiB |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_backward.h | h | 1.07 KiB |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_backward_compositeimplicitautograd_dispatch.h | h | 972 B |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_backward_native.h | h | 682 B |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_backward_ops.h | h | 1.52 KiB |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_compositeexplicitautograd_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_native.h | h | 814 B |
torch/include/ATen/ops/_grid_sampler_2d_cpu_fallback_ops.h | h | 2.27 KiB |
torch/include/ATen/ops/_has_compatible_shallow_copy_type.h | h | 788 B |
torch/include/ATen/ops/_has_compatible_shallow_copy_type_compositeimplicitautograd_dispatch.h | h | 833 B |
torch/include/ATen/ops/_has_compatible_shallow_copy_type_native.h | h | 543 B |
torch/include/ATen/ops/_has_compatible_shallow_copy_type_ops.h | h | 1.08 KiB |
torch/include/ATen/ops/_has_same_storage_numel.h | h | 751 B |
torch/include/ATen/ops/_has_same_storage_numel_compositeexplicitautograd_dispatch.h | h | 824 B |
torch/include/ATen/ops/_has_same_storage_numel_native.h | h | 534 B |
torch/include/ATen/ops/_has_same_storage_numel_ops.h | h | 1.05 KiB |
torch/include/ATen/ops/_histogramdd_bin_edges.h | h | 1.89 KiB |
torch/include/ATen/ops/_histogramdd_bin_edges_compositeexplicitautograd_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_histogramdd_bin_edges_cpu_dispatch.h | h | 923 B |
torch/include/ATen/ops/_histogramdd_bin_edges_native.h | h | 890 B |
torch/include/ATen/ops/_histogramdd_bin_edges_ops.h | h | 2.47 KiB |
torch/include/ATen/ops/_histogramdd_from_bin_cts.h | h | 1.92 KiB |
torch/include/ATen/ops/_histogramdd_from_bin_cts_compositeexplicitautograd_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_histogramdd_from_bin_cts_cpu_dispatch.h | h | 911 B |
torch/include/ATen/ops/_histogramdd_from_bin_cts_native.h | h | 875 B |
torch/include/ATen/ops/_histogramdd_from_bin_cts_ops.h | h | 2.47 KiB |
torch/include/ATen/ops/_histogramdd_from_bin_tensors.h | h | 1.73 KiB |
torch/include/ATen/ops/_histogramdd_from_bin_tensors_compositeexplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_histogramdd_from_bin_tensors_cpu_dispatch.h | h | 854 B |
torch/include/ATen/ops/_histogramdd_from_bin_tensors_native.h | h | 772 B |
torch/include/ATen/ops/_histogramdd_from_bin_tensors_ops.h | h | 2.20 KiB |
torch/include/ATen/ops/_index_put_impl.h | h | 2.19 KiB |
torch/include/ATen/ops/_index_put_impl_compositeexplicitautograd_dispatch.h | h | 1.30 KiB |
torch/include/ATen/ops/_index_put_impl_cpu_dispatch.h | h | 874 B |
torch/include/ATen/ops/_index_put_impl_cuda_dispatch.h | h | 876 B |
torch/include/ATen/ops/_index_put_impl_meta_dispatch.h | h | 876 B |
torch/include/ATen/ops/_index_put_impl_native.h | h | 1.39 KiB |
torch/include/ATen/ops/_index_put_impl_ops.h | h | 3.22 KiB |
torch/include/ATen/ops/_indices.h | h | 508 B |
torch/include/ATen/ops/_indices_copy.h | h | 1.09 KiB |
torch/include/ATen/ops/_indices_copy_compositeexplicitautograd_dispatch.h | h | 905 B |
torch/include/ATen/ops/_indices_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 820 B |
torch/include/ATen/ops/_indices_copy_native.h | h | 590 B |
torch/include/ATen/ops/_indices_copy_ops.h | h | 1.57 KiB |
torch/include/ATen/ops/_indices_native.h | h | 506 B |
torch/include/ATen/ops/_indices_ops.h | h | 974 B |
torch/include/ATen/ops/_int_mm.h | h | 1.16 KiB |
torch/include/ATen/ops/_int_mm_cpu_dispatch.h | h | 980 B |
torch/include/ATen/ops/_int_mm_cuda_dispatch.h | h | 982 B |
torch/include/ATen/ops/_int_mm_native.h | h | 832 B |
torch/include/ATen/ops/_int_mm_ops.h | h | 1.69 KiB |
torch/include/ATen/ops/_is_all_true.h | h | 668 B |
torch/include/ATen/ops/_is_all_true_compositeexplicitautograd_dispatch.h | h | 793 B |
torch/include/ATen/ops/_is_all_true_native.h | h | 503 B |
torch/include/ATen/ops/_is_all_true_ops.h | h | 980 B |
torch/include/ATen/ops/_is_any_true.h | h | 668 B |
torch/include/ATen/ops/_is_any_true_compositeexplicitautograd_dispatch.h | h | 793 B |
torch/include/ATen/ops/_is_any_true_native.h | h | 503 B |
torch/include/ATen/ops/_is_any_true_ops.h | h | 980 B |
torch/include/ATen/ops/_is_zerotensor.h | h | 679 B |
torch/include/ATen/ops/_is_zerotensor_compositeimplicitautograd_dispatch.h | h | 789 B |
torch/include/ATen/ops/_is_zerotensor_native.h | h | 499 B |
torch/include/ATen/ops/_is_zerotensor_ops.h | h | 966 B |
torch/include/ATen/ops/_jagged_to_padded_dense_forward.h | h | 2.09 KiB |
torch/include/ATen/ops/_jagged_to_padded_dense_forward_cpu_dispatch.h | h | 1021 B |
torch/include/ATen/ops/_jagged_to_padded_dense_forward_cuda_dispatch.h | h | 1023 B |
torch/include/ATen/ops/_jagged_to_padded_dense_forward_native.h | h | 775 B |
torch/include/ATen/ops/_jagged_to_padded_dense_forward_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/_lazy_clone.h | h | 664 B |
torch/include/ATen/ops/_lazy_clone_compositeexplicitautograd_dispatch.h | h | 792 B |
torch/include/ATen/ops/_lazy_clone_native.h | h | 502 B |
torch/include/ATen/ops/_lazy_clone_ops.h | h | 977 B |
torch/include/ATen/ops/_linalg_check_errors.h | h | 787 B |
torch/include/ATen/ops/_linalg_check_errors_compositeexplicitautograd_dispatch.h | h | 838 B |
torch/include/ATen/ops/_linalg_check_errors_native.h | h | 548 B |
torch/include/ATen/ops/_linalg_check_errors_ops.h | h | 1.10 KiB |
torch/include/ATen/ops/_linalg_det.h | h | 1.47 KiB |
torch/include/ATen/ops/_linalg_det_compositeexplicitautogradnonfunctional_dispatch.h | h | 851 B |
torch/include/ATen/ops/_linalg_det_cpu_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_linalg_det_cuda_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_linalg_det_meta.h | h | 604 B |
torch/include/ATen/ops/_linalg_det_meta_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_linalg_det_native.h | h | 684 B |
torch/include/ATen/ops/_linalg_det_ops.h | h | 1.99 KiB |
torch/include/ATen/ops/_linalg_eigh.h | h | 1.73 KiB |
torch/include/ATen/ops/_linalg_eigh_compositeexplicitautogradnonfunctional_dispatch.h | h | 889 B |
torch/include/ATen/ops/_linalg_eigh_cpu_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_linalg_eigh_cuda_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_linalg_eigh_meta.h | h | 644 B |
torch/include/ATen/ops/_linalg_eigh_meta_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_linalg_eigh_native.h | h | 714 B |
torch/include/ATen/ops/_linalg_eigh_ops.h | h | 2.18 KiB |
torch/include/ATen/ops/_linalg_eigvals.h | h | 680 B |
torch/include/ATen/ops/_linalg_eigvals_cpu_dispatch.h | h | 752 B |
torch/include/ATen/ops/_linalg_eigvals_cuda_dispatch.h | h | 754 B |
torch/include/ATen/ops/_linalg_eigvals_native.h | h | 506 B |
torch/include/ATen/ops/_linalg_eigvals_ops.h | h | 989 B |
torch/include/ATen/ops/_linalg_slogdet.h | h | 1.69 KiB |
torch/include/ATen/ops/_linalg_slogdet_compositeexplicitautogradnonfunctional_dispatch.h | h | 866 B |
torch/include/ATen/ops/_linalg_slogdet_cpu_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_linalg_slogdet_cuda_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_linalg_slogdet_meta.h | h | 608 B |
torch/include/ATen/ops/_linalg_slogdet_meta_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_linalg_slogdet_native.h | h | 724 B |
torch/include/ATen/ops/_linalg_slogdet_ops.h | h | 2.19 KiB |
torch/include/ATen/ops/_linalg_solve_ex.h | h | 2.07 KiB |
torch/include/ATen/ops/_linalg_solve_ex_compositeexplicitautogradnonfunctional_dispatch.h | h | 930 B |
torch/include/ATen/ops/_linalg_solve_ex_cpu_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_linalg_solve_ex_cuda_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_linalg_solve_ex_meta.h | h | 661 B |
torch/include/ATen/ops/_linalg_solve_ex_meta_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_linalg_solve_ex_native.h | h | 776 B |
torch/include/ATen/ops/_linalg_solve_ex_ops.h | h | 2.55 KiB |
torch/include/ATen/ops/_linalg_svd.h | h | 1.94 KiB |
torch/include/ATen/ops/_linalg_svd_compositeexplicitautogradnonfunctional_dispatch.h | h | 956 B |
torch/include/ATen/ops/_linalg_svd_cpu_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_linalg_svd_cuda_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_linalg_svd_meta.h | h | 683 B |
torch/include/ATen/ops/_linalg_svd_meta_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/_linalg_svd_native.h | h | 753 B |
torch/include/ATen/ops/_linalg_svd_ops.h | h | 2.46 KiB |
torch/include/ATen/ops/_local_scalar_dense.h | h | 696 B |
torch/include/ATen/ops/_local_scalar_dense_cpu_dispatch.h | h | 756 B |
torch/include/ATen/ops/_local_scalar_dense_cuda_dispatch.h | h | 758 B |
torch/include/ATen/ops/_local_scalar_dense_native.h | h | 587 B |
torch/include/ATen/ops/_local_scalar_dense_ops.h | h | 1001 B |
torch/include/ATen/ops/_log_softmax.h | h | 1.32 KiB |
torch/include/ATen/ops/_log_softmax_backward_data.h | h | 1.69 KiB |
torch/include/ATen/ops/_log_softmax_backward_data_compositeexplicitautogradnonfunctional_dispatch.h | h | 908 B |
torch/include/ATen/ops/_log_softmax_backward_data_cpu_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_log_softmax_backward_data_cuda_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_log_softmax_backward_data_meta.h | h | 697 B |
torch/include/ATen/ops/_log_softmax_backward_data_meta_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_log_softmax_backward_data_native.h | h | 1010 B |
torch/include/ATen/ops/_log_softmax_backward_data_ops.h | h | 2.13 KiB |
torch/include/ATen/ops/_log_softmax_compositeexplicitautogradnonfunctional_dispatch.h | h | 852 B |
torch/include/ATen/ops/_log_softmax_cpu_dispatch.h | h | 1019 B |
torch/include/ATen/ops/_log_softmax_cuda_dispatch.h | h | 1021 B |
torch/include/ATen/ops/_log_softmax_meta.h | h | 641 B |
torch/include/ATen/ops/_log_softmax_meta_dispatch.h | h | 1021 B |
torch/include/ATen/ops/_log_softmax_native.h | h | 866 B |
torch/include/ATen/ops/_log_softmax_ops.h | h | 1.78 KiB |
torch/include/ATen/ops/_logcumsumexp.h | h | 1.17 KiB |
torch/include/ATen/ops/_logcumsumexp_cpu_dispatch.h | h | 962 B |
torch/include/ATen/ops/_logcumsumexp_cuda_dispatch.h | h | 964 B |
torch/include/ATen/ops/_logcumsumexp_native.h | h | 808 B |
torch/include/ATen/ops/_logcumsumexp_ops.h | h | 1.65 KiB |
torch/include/ATen/ops/_lstm_mps.h | h | 2.89 KiB |
torch/include/ATen/ops/_lstm_mps_compositeexplicitautograd_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/_lstm_mps_native.h | h | 845 B |
torch/include/ATen/ops/_lstm_mps_ops.h | h | 3.38 KiB |
torch/include/ATen/ops/_lu_with_info.h | h | 844 B |
torch/include/ATen/ops/_lu_with_info_compositeimplicitautograd_dispatch.h | h | 871 B |
torch/include/ATen/ops/_lu_with_info_native.h | h | 581 B |
torch/include/ATen/ops/_lu_with_info_ops.h | h | 1.21 KiB |
torch/include/ATen/ops/_make_dep_token.h | h | 1.52 KiB |
torch/include/ATen/ops/_make_dep_token_cpu_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/_make_dep_token_native.h | h | 706 B |
torch/include/ATen/ops/_make_dep_token_ops.h | h | 1.53 KiB |
torch/include/ATen/ops/_make_dual.h | h | 758 B |
torch/include/ATen/ops/_make_dual_compositeexplicitautograd_dispatch.h | h | 836 B |
torch/include/ATen/ops/_make_dual_copy.h | h | 1.38 KiB |
torch/include/ATen/ops/_make_dual_copy_compositeexplicitautograd_dispatch.h | h | 999 B |
torch/include/ATen/ops/_make_dual_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 867 B |
torch/include/ATen/ops/_make_dual_copy_native.h | h | 684 B |
torch/include/ATen/ops/_make_dual_copy_ops.h | h | 1.87 KiB |
torch/include/ATen/ops/_make_dual_native.h | h | 546 B |
torch/include/ATen/ops/_make_dual_ops.h | h | 1.10 KiB |
torch/include/ATen/ops/_make_per_channel_quantized_tensor.h | h | 1.70 KiB |
torch/include/ATen/ops/_make_per_channel_quantized_tensor_compositeexplicitautograd_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_make_per_channel_quantized_tensor_cpu_dispatch.h | h | 842 B |
torch/include/ATen/ops/_make_per_channel_quantized_tensor_cuda_dispatch.h | h | 844 B |
torch/include/ATen/ops/_make_per_channel_quantized_tensor_native.h | h | 935 B |
torch/include/ATen/ops/_make_per_channel_quantized_tensor_ops.h | h | 2.15 KiB |
torch/include/ATen/ops/_make_per_tensor_quantized_tensor.h | h | 1.52 KiB |
torch/include/ATen/ops/_make_per_tensor_quantized_tensor_compositeexplicitautograd_dispatch.h | h | 1013 B |
torch/include/ATen/ops/_make_per_tensor_quantized_tensor_cpu_dispatch.h | h | 804 B |
torch/include/ATen/ops/_make_per_tensor_quantized_tensor_cuda_dispatch.h | h | 806 B |
torch/include/ATen/ops/_make_per_tensor_quantized_tensor_native.h | h | 821 B |
torch/include/ATen/ops/_make_per_tensor_quantized_tensor_ops.h | h | 1.91 KiB |
torch/include/ATen/ops/_masked_scale.h | h | 1.32 KiB |
torch/include/ATen/ops/_masked_scale_compositeexplicitautograd_dispatch.h | h | 983 B |
torch/include/ATen/ops/_masked_scale_cuda_dispatch.h | h | 791 B |
torch/include/ATen/ops/_masked_scale_native.h | h | 672 B |
torch/include/ATen/ops/_masked_scale_ops.h | h | 1.83 KiB |
torch/include/ATen/ops/_masked_softmax.h | h | 1.64 KiB |
torch/include/ATen/ops/_masked_softmax_backward.h | h | 1.71 KiB |
torch/include/ATen/ops/_masked_softmax_backward_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/_masked_softmax_backward_cpu_dispatch.h | h | 865 B |
torch/include/ATen/ops/_masked_softmax_backward_cuda_dispatch.h | h | 867 B |
torch/include/ATen/ops/_masked_softmax_backward_native.h | h | 989 B |
torch/include/ATen/ops/_masked_softmax_backward_ops.h | h | 2.21 KiB |
torch/include/ATen/ops/_masked_softmax_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/_masked_softmax_cpu_dispatch.h | h | 873 B |
torch/include/ATen/ops/_masked_softmax_cuda_dispatch.h | h | 875 B |
torch/include/ATen/ops/_masked_softmax_native.h | h | 998 B |
torch/include/ATen/ops/_masked_softmax_ops.h | h | 2.17 KiB |
torch/include/ATen/ops/_mixed_dtypes_linear.h | h | 968 B |
torch/include/ATen/ops/_mixed_dtypes_linear_cuda_dispatch.h | h | 919 B |
torch/include/ATen/ops/_mixed_dtypes_linear_native.h | h | 671 B |
torch/include/ATen/ops/_mixed_dtypes_linear_ops.h | h | 1.44 KiB |
torch/include/ATen/ops/_mkldnn_reshape.h | h | 1.23 KiB |
torch/include/ATen/ops/_mkldnn_reshape_compositeexplicitautograd_dispatch.h | h | 955 B |
torch/include/ATen/ops/_mkldnn_reshape_native.h | h | 639 B |
torch/include/ATen/ops/_mkldnn_reshape_ops.h | h | 1.73 KiB |
torch/include/ATen/ops/_mkldnn_transpose.h | h | 1.54 KiB |
torch/include/ATen/ops/_mkldnn_transpose_compositeexplicitautograd_dispatch.h | h | 969 B |
torch/include/ATen/ops/_mkldnn_transpose_meta_dispatch.h | h | 781 B |
torch/include/ATen/ops/_mkldnn_transpose_native.h | h | 743 B |
torch/include/ATen/ops/_mkldnn_transpose_ops.h | h | 2.40 KiB |
torch/include/ATen/ops/_mps_convolution.h | h | 6.76 KiB |
torch/include/ATen/ops/_mps_convolution_compositeexplicitautograd_dispatch.h | h | 1.73 KiB |
torch/include/ATen/ops/_mps_convolution_native.h | h | 714 B |
torch/include/ATen/ops/_mps_convolution_ops.h | h | 2.72 KiB |
torch/include/ATen/ops/_mps_convolution_transpose.h | h | 7.30 KiB |
torch/include/ATen/ops/_mps_convolution_transpose_compositeexplicitautograd_dispatch.h | h | 1.74 KiB |
torch/include/ATen/ops/_mps_convolution_transpose_native.h | h | 718 B |
torch/include/ATen/ops/_mps_convolution_transpose_ops.h | h | 2.75 KiB |
torch/include/ATen/ops/_native_batch_norm_legit.h | h | 5.28 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_compositeexplicitautograd_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_cpu_dispatch.h | h | 2.54 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_cuda_dispatch.h | h | 2.54 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_native.h | h | 3.69 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_no_training.h | h | 2.65 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_no_training_compositeexplicitautograd_dispatch.h | h | 1.73 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_no_training_native.h | h | 1.08 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_no_training_ops.h | h | 3.26 KiB |
torch/include/ATen/ops/_native_batch_norm_legit_ops.h | h | 7.23 KiB |
torch/include/ATen/ops/_native_multi_head_attention.h | h | 3.52 KiB |
torch/include/ATen/ops/_native_multi_head_attention_compositeexplicitautograd_dispatch.h | h | 1.68 KiB |
torch/include/ATen/ops/_native_multi_head_attention_cpu_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/_native_multi_head_attention_cuda_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/_native_multi_head_attention_native.h | h | 1.82 KiB |
torch/include/ATen/ops/_native_multi_head_attention_ops.h | h | 4.06 KiB |
torch/include/ATen/ops/_neg_view.h | h | 662 B |
torch/include/ATen/ops/_neg_view_compositeexplicitautograd_dispatch.h | h | 790 B |
torch/include/ATen/ops/_neg_view_copy.h | h | 1.10 KiB |
torch/include/ATen/ops/_neg_view_copy_compositeexplicitautograd_dispatch.h | h | 907 B |
torch/include/ATen/ops/_neg_view_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 821 B |
torch/include/ATen/ops/_neg_view_copy_native.h | h | 592 B |
torch/include/ATen/ops/_neg_view_copy_ops.h | h | 1.57 KiB |
torch/include/ATen/ops/_neg_view_native.h | h | 500 B |
torch/include/ATen/ops/_neg_view_ops.h | h | 977 B |
torch/include/ATen/ops/_nested_compute_contiguous_strides_offsets.h | h | 844 B |
torch/include/ATen/ops/_nested_compute_contiguous_strides_offsets_cpu_dispatch.h | h | 811 B |
torch/include/ATen/ops/_nested_compute_contiguous_strides_offsets_cuda_dispatch.h | h | 813 B |
torch/include/ATen/ops/_nested_compute_contiguous_strides_offsets_native.h | h | 565 B |
torch/include/ATen/ops/_nested_compute_contiguous_strides_offsets_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/_nested_from_padded.h | h | 1.71 KiB |
torch/include/ATen/ops/_nested_from_padded_and_nested_example.h | h | 1.53 KiB |
torch/include/ATen/ops/_nested_from_padded_and_nested_example_compositeexplicitautograd_dispatch.h | h | 1021 B |
torch/include/ATen/ops/_nested_from_padded_and_nested_example_native.h | h | 711 B |
torch/include/ATen/ops/_nested_from_padded_and_nested_example_ops.h | h | 1.92 KiB |
torch/include/ATen/ops/_nested_from_padded_compositeexplicitautograd_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_nested_from_padded_cpu_dispatch.h | h | 835 B |
torch/include/ATen/ops/_nested_from_padded_cuda_dispatch.h | h | 837 B |
torch/include/ATen/ops/_nested_from_padded_native.h | h | 912 B |
torch/include/ATen/ops/_nested_from_padded_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/_nested_from_padded_tensor.h | h | 2.89 KiB |
torch/include/ATen/ops/_nested_from_padded_tensor_native.h | h | 442 B |
torch/include/ATen/ops/_nested_from_padded_tensor_ops.h | h | 1.67 KiB |
torch/include/ATen/ops/_nested_get_jagged_dummy.h | h | 713 B |
torch/include/ATen/ops/_nested_get_jagged_dummy_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_jagged_dummy_ops.h | h | 1013 B |
torch/include/ATen/ops/_nested_get_lengths.h | h | 696 B |
torch/include/ATen/ops/_nested_get_lengths_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_lengths_ops.h | h | 1001 B |
torch/include/ATen/ops/_nested_get_max_seqlen.h | h | 708 B |
torch/include/ATen/ops/_nested_get_max_seqlen_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_max_seqlen_ops.h | h | 1010 B |
torch/include/ATen/ops/_nested_get_min_seqlen.h | h | 708 B |
torch/include/ATen/ops/_nested_get_min_seqlen_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_min_seqlen_ops.h | h | 1010 B |
torch/include/ATen/ops/_nested_get_offsets.h | h | 696 B |
torch/include/ATen/ops/_nested_get_offsets_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_offsets_ops.h | h | 1001 B |
torch/include/ATen/ops/_nested_get_ragged_idx.h | h | 702 B |
torch/include/ATen/ops/_nested_get_ragged_idx_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_ragged_idx_ops.h | h | 998 B |
torch/include/ATen/ops/_nested_get_values.h | h | 698 B |
torch/include/ATen/ops/_nested_get_values_copy.h | h | 1.18 KiB |
torch/include/ATen/ops/_nested_get_values_copy_compositeexplicitautograd_dispatch.h | h | 925 B |
torch/include/ATen/ops/_nested_get_values_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 830 B |
torch/include/ATen/ops/_nested_get_values_copy_native.h | h | 610 B |
torch/include/ATen/ops/_nested_get_values_copy_ops.h | h | 1.63 KiB |
torch/include/ATen/ops/_nested_get_values_native.h | h | 442 B |
torch/include/ATen/ops/_nested_get_values_ops.h | h | 1004 B |
torch/include/ATen/ops/_nested_select_backward.h | h | 1.77 KiB |
torch/include/ATen/ops/_nested_select_backward_native.h | h | 585 B |
torch/include/ATen/ops/_nested_select_backward_ops.h | h | 1.20 KiB |
torch/include/ATen/ops/_nested_sum_backward.h | h | 840 B |
torch/include/ATen/ops/_nested_sum_backward_native.h | h | 589 B |
torch/include/ATen/ops/_nested_sum_backward_ops.h | h | 1.21 KiB |
torch/include/ATen/ops/_nested_tensor_from_mask.h | h | 1.46 KiB |
torch/include/ATen/ops/_nested_tensor_from_mask_compositeexplicitautograd_dispatch.h | h | 1010 B |
torch/include/ATen/ops/_nested_tensor_from_mask_cpu_dispatch.h | h | 805 B |
torch/include/ATen/ops/_nested_tensor_from_mask_cuda_dispatch.h | h | 807 B |
torch/include/ATen/ops/_nested_tensor_from_mask_left_aligned.h | h | 795 B |
torch/include/ATen/ops/_nested_tensor_from_mask_left_aligned_cpu_dispatch.h | h | 790 B |
torch/include/ATen/ops/_nested_tensor_from_mask_left_aligned_cuda_dispatch.h | h | 792 B |
torch/include/ATen/ops/_nested_tensor_from_mask_left_aligned_native.h | h | 556 B |
torch/include/ATen/ops/_nested_tensor_from_mask_left_aligned_ops.h | h | 1.08 KiB |
torch/include/ATen/ops/_nested_tensor_from_mask_native.h | h | 707 B |
torch/include/ATen/ops/_nested_tensor_from_mask_ops.h | h | 1.90 KiB |
torch/include/ATen/ops/_nested_tensor_from_tensor_list.h | h | 2.16 KiB |
torch/include/ATen/ops/_nested_tensor_from_tensor_list_compositeexplicitautograd_dispatch.h | h | 1.53 KiB |
torch/include/ATen/ops/_nested_tensor_from_tensor_list_native.h | h | 968 B |
torch/include/ATen/ops/_nested_tensor_from_tensor_list_ops.h | h | 2.62 KiB |
torch/include/ATen/ops/_nested_tensor_size.h | h | 994 B |
torch/include/ATen/ops/_nested_tensor_size_compositeexplicitautograd_dispatch.h | h | 917 B |
torch/include/ATen/ops/_nested_tensor_size_native.h | h | 602 B |
torch/include/ATen/ops/_nested_tensor_size_ops.h | h | 1.60 KiB |
torch/include/ATen/ops/_nested_tensor_softmax_with_shape.h | h | 799 B |
torch/include/ATen/ops/_nested_tensor_softmax_with_shape_native.h | h | 653 B |
torch/include/ATen/ops/_nested_tensor_softmax_with_shape_ops.h | h | 1.10 KiB |
torch/include/ATen/ops/_nested_tensor_storage_offsets.h | h | 1.05 KiB |
torch/include/ATen/ops/_nested_tensor_storage_offsets_compositeexplicitautograd_dispatch.h | h | 939 B |
torch/include/ATen/ops/_nested_tensor_storage_offsets_native.h | h | 624 B |
torch/include/ATen/ops/_nested_tensor_storage_offsets_ops.h | h | 1.67 KiB |
torch/include/ATen/ops/_nested_tensor_strides.h | h | 1015 B |
torch/include/ATen/ops/_nested_tensor_strides_compositeexplicitautograd_dispatch.h | h | 923 B |
torch/include/ATen/ops/_nested_tensor_strides_native.h | h | 608 B |
torch/include/ATen/ops/_nested_tensor_strides_ops.h | h | 1.62 KiB |
torch/include/ATen/ops/_nested_view_from_buffer.h | h | 914 B |
torch/include/ATen/ops/_nested_view_from_buffer_copy.h | h | 1.81 KiB |
torch/include/ATen/ops/_nested_view_from_buffer_copy_compositeexplicitautograd_dispatch.h | h | 1.10 KiB |
torch/include/ATen/ops/_nested_view_from_buffer_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 931 B |
torch/include/ATen/ops/_nested_view_from_buffer_copy_native.h | h | 812 B |
torch/include/ATen/ops/_nested_view_from_buffer_copy_ops.h | h | 2.26 KiB |
torch/include/ATen/ops/_nested_view_from_buffer_cpu_dispatch.h | h | 856 B |
torch/include/ATen/ops/_nested_view_from_buffer_cuda_dispatch.h | h | 858 B |
torch/include/ATen/ops/_nested_view_from_buffer_native.h | h | 610 B |
torch/include/ATen/ops/_nested_view_from_buffer_ops.h | h | 1.30 KiB |
torch/include/ATen/ops/_nested_view_from_jagged.h | h | 1.10 KiB |
torch/include/ATen/ops/_nested_view_from_jagged_copy.h | h | 2.42 KiB |
torch/include/ATen/ops/_nested_view_from_jagged_copy_compositeexplicitautograd_dispatch.h | h | 1.35 KiB |
torch/include/ATen/ops/_nested_view_from_jagged_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_nested_view_from_jagged_copy_native.h | h | 1.04 KiB |
torch/include/ATen/ops/_nested_view_from_jagged_copy_ops.h | h | 3.05 KiB |
torch/include/ATen/ops/_nested_view_from_jagged_native.h | h | 442 B |
torch/include/ATen/ops/_nested_view_from_jagged_ops.h | h | 1.69 KiB |
torch/include/ATen/ops/_new_zeros_with_same_feature_meta.h | h | 1.65 KiB |
torch/include/ATen/ops/_new_zeros_with_same_feature_meta_compositeexplicitautograd_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/_new_zeros_with_same_feature_meta_native.h | h | 742 B |
torch/include/ATen/ops/_new_zeros_with_same_feature_meta_ops.h | h | 2.04 KiB |
torch/include/ATen/ops/_nnpack_available.h | h | 642 B |
torch/include/ATen/ops/_nnpack_available_compositeimplicitautograd_dispatch.h | h | 769 B |
torch/include/ATen/ops/_nnpack_available_native.h | h | 479 B |
torch/include/ATen/ops/_nnpack_available_ops.h | h | 898 B |
torch/include/ATen/ops/_nnpack_spatial_convolution.h | h | 6.11 KiB |
torch/include/ATen/ops/_nnpack_spatial_convolution_compositeexplicitautograd_dispatch.h | h | 2.03 KiB |
torch/include/ATen/ops/_nnpack_spatial_convolution_native.h | h | 873 B |
torch/include/ATen/ops/_nnpack_spatial_convolution_ops.h | h | 2.47 KiB |
torch/include/ATen/ops/_nnz.h | h | 504 B |
torch/include/ATen/ops/_nnz_native.h | h | 560 B |
torch/include/ATen/ops/_nnz_ops.h | h | 944 B |
torch/include/ATen/ops/_pack_padded_sequence.h | h | 1.67 KiB |
torch/include/ATen/ops/_pack_padded_sequence_backward.h | h | 2.05 KiB |
torch/include/ATen/ops/_pack_padded_sequence_backward_compositeimplicitautograd_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/_pack_padded_sequence_backward_native.h | h | 610 B |
torch/include/ATen/ops/_pack_padded_sequence_backward_ops.h | h | 1.27 KiB |
torch/include/ATen/ops/_pack_padded_sequence_compositeexplicitautograd_dispatch.h | h | 1.22 KiB |
torch/include/ATen/ops/_pack_padded_sequence_native.h | h | 772 B |
torch/include/ATen/ops/_pack_padded_sequence_ops.h | h | 2.16 KiB |
torch/include/ATen/ops/_pad_circular.h | h | 1.43 KiB |
torch/include/ATen/ops/_pad_circular_compositeimplicitautograd_dispatch.h | h | 909 B |
torch/include/ATen/ops/_pad_circular_native.h | h | 536 B |
torch/include/ATen/ops/_pad_circular_ops.h | h | 1.04 KiB |
torch/include/ATen/ops/_pad_enum.h | h | 1.73 KiB |
torch/include/ATen/ops/_pad_enum_compositeimplicitautograd_dispatch.h | h | 1021 B |
torch/include/ATen/ops/_pad_enum_native.h | h | 592 B |
torch/include/ATen/ops/_pad_enum_ops.h | h | 1.18 KiB |
torch/include/ATen/ops/_pad_packed_sequence.h | h | 974 B |
torch/include/ATen/ops/_pad_packed_sequence_compositeimplicitautograd_dispatch.h | h | 932 B |
torch/include/ATen/ops/_pad_packed_sequence_native.h | h | 642 B |
torch/include/ATen/ops/_pad_packed_sequence_ops.h | h | 1.40 KiB |
torch/include/ATen/ops/_padded_dense_to_jagged_forward.h | h | 2.03 KiB |
torch/include/ATen/ops/_padded_dense_to_jagged_forward_cpu_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_padded_dense_to_jagged_forward_cuda_dispatch.h | h | 1009 B |
torch/include/ATen/ops/_padded_dense_to_jagged_forward_native.h | h | 765 B |
torch/include/ATen/ops/_padded_dense_to_jagged_forward_ops.h | h | 1.22 KiB |
torch/include/ATen/ops/_pdist_backward.h | h | 1.44 KiB |
torch/include/ATen/ops/_pdist_backward_compositeexplicitautograd_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_pdist_backward_cpu_dispatch.h | h | 813 B |
torch/include/ATen/ops/_pdist_backward_cuda_dispatch.h | h | 815 B |
torch/include/ATen/ops/_pdist_backward_native.h | h | 716 B |
torch/include/ATen/ops/_pdist_backward_ops.h | h | 1.98 KiB |
torch/include/ATen/ops/_pdist_forward.h | h | 1.17 KiB |
torch/include/ATen/ops/_pdist_forward_compositeexplicitautograd_dispatch.h | h | 929 B |
torch/include/ATen/ops/_pdist_forward_cpu_dispatch.h | h | 763 B |
torch/include/ATen/ops/_pdist_forward_cuda_dispatch.h | h | 765 B |
torch/include/ATen/ops/_pdist_forward_native.h | h | 614 B |
torch/include/ATen/ops/_pdist_forward_ops.h | h | 1.65 KiB |
torch/include/ATen/ops/_pin_memory.h | h | 1.29 KiB |
torch/include/ATen/ops/_pin_memory_compositeexplicitautograd_dispatch.h | h | 1.07 KiB |
torch/include/ATen/ops/_pin_memory_native.h | h | 1.02 KiB |
torch/include/ATen/ops/_pin_memory_ops.h | h | 1.79 KiB |
torch/include/ATen/ops/_prelu_kernel.h | h | 722 B |
torch/include/ATen/ops/_prelu_kernel_backward.h | h | 858 B |
torch/include/ATen/ops/_prelu_kernel_backward_cpu_dispatch.h | h | 843 B |
torch/include/ATen/ops/_prelu_kernel_backward_cuda_dispatch.h | h | 845 B |
torch/include/ATen/ops/_prelu_kernel_backward_native.h | h | 751 B |
torch/include/ATen/ops/_prelu_kernel_backward_ops.h | h | 1.26 KiB |
torch/include/ATen/ops/_prelu_kernel_cpu_dispatch.h | h | 777 B |
torch/include/ATen/ops/_prelu_kernel_cuda_dispatch.h | h | 779 B |
torch/include/ATen/ops/_prelu_kernel_native.h | h | 722 B |
torch/include/ATen/ops/_prelu_kernel_ops.h | h | 1.05 KiB |
torch/include/ATen/ops/_print.h | h | 620 B |
torch/include/ATen/ops/_print_compositeexplicitautograd_dispatch.h | h | 776 B |
torch/include/ATen/ops/_print_native.h | h | 486 B |
torch/include/ATen/ops/_print_ops.h | h | 922 B |
torch/include/ATen/ops/_propagate_xla_data.h | h | 739 B |
torch/include/ATen/ops/_propagate_xla_data_compositeimplicitautograd_dispatch.h | h | 822 B |
torch/include/ATen/ops/_propagate_xla_data_native.h | h | 532 B |
torch/include/ATen/ops/_propagate_xla_data_ops.h | h | 1.05 KiB |
torch/include/ATen/ops/_remove_batch_dim.h | h | 1.69 KiB |
torch/include/ATen/ops/_remove_batch_dim_compositeimplicitautograd_dispatch.h | h | 979 B |
torch/include/ATen/ops/_remove_batch_dim_native.h | h | 560 B |
torch/include/ATen/ops/_remove_batch_dim_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/_reshape_alias.h | h | 1.68 KiB |
torch/include/ATen/ops/_reshape_alias_copy.h | h | 4.52 KiB |
torch/include/ATen/ops/_reshape_alias_copy_compositeexplicitautograd_dispatch.h | h | 1.29 KiB |
torch/include/ATen/ops/_reshape_alias_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 1001 B |
torch/include/ATen/ops/_reshape_alias_copy_native.h | h | 724 B |
torch/include/ATen/ops/_reshape_alias_copy_ops.h | h | 1.96 KiB |
torch/include/ATen/ops/_reshape_alias_cpu_dispatch.h | h | 921 B |
torch/include/ATen/ops/_reshape_alias_cuda_dispatch.h | h | 923 B |
torch/include/ATen/ops/_reshape_alias_meta_dispatch.h | h | 923 B |
torch/include/ATen/ops/_reshape_alias_native.h | h | 551 B |
torch/include/ATen/ops/_reshape_alias_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/_reshape_copy.h | h | 1.44 KiB |
torch/include/ATen/ops/_reshape_copy_compositeexplicitautograd_dispatch.h | h | 911 B |
torch/include/ATen/ops/_reshape_copy_native.h | h | 537 B |
torch/include/ATen/ops/_reshape_copy_ops.h | h | 1.05 KiB |
torch/include/ATen/ops/_reshape_from_tensor.h | h | 747 B |
torch/include/ATen/ops/_reshape_from_tensor_compositeimplicitautograd_dispatch.h | h | 827 B |
torch/include/ATen/ops/_reshape_from_tensor_native.h | h | 537 B |
torch/include/ATen/ops/_reshape_from_tensor_ops.h | h | 1.06 KiB |
torch/include/ATen/ops/_resize_output.h | h | 5.36 KiB |
torch/include/ATen/ops/_resize_output_compositeexplicitautograd_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/_resize_output_meta_dispatch.h | h | 927 B |
torch/include/ATen/ops/_resize_output_native.h | h | 821 B |
torch/include/ATen/ops/_resize_output_ops.h | h | 2.61 KiB |
torch/include/ATen/ops/_rowwise_prune.h | h | 865 B |
torch/include/ATen/ops/_rowwise_prune_compositeimplicitautograd_dispatch.h | h | 888 B |
torch/include/ATen/ops/_rowwise_prune_native.h | h | 598 B |
torch/include/ATen/ops/_rowwise_prune_ops.h | h | 1.26 KiB |
torch/include/ATen/ops/_safe_softmax.h | h | 784 B |
torch/include/ATen/ops/_safe_softmax_compositeexplicitautograd_dispatch.h | h | 861 B |
torch/include/ATen/ops/_safe_softmax_native.h | h | 571 B |
torch/include/ATen/ops/_safe_softmax_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/_sample_dirichlet.h | h | 1.39 KiB |
torch/include/ATen/ops/_sample_dirichlet_compositeexplicitautograd_dispatch.h | h | 1012 B |
torch/include/ATen/ops/_sample_dirichlet_cpu_dispatch.h | h | 811 B |
torch/include/ATen/ops/_sample_dirichlet_cuda_dispatch.h | h | 813 B |
torch/include/ATen/ops/_sample_dirichlet_native.h | h | 819 B |
torch/include/ATen/ops/_sample_dirichlet_ops.h | h | 1.87 KiB |
torch/include/ATen/ops/_saturate_weight_to_fp16.h | h | 722 B |
torch/include/ATen/ops/_saturate_weight_to_fp16_compositeimplicitautograd_dispatch.h | h | 807 B |
torch/include/ATen/ops/_saturate_weight_to_fp16_native.h | h | 517 B |
torch/include/ATen/ops/_saturate_weight_to_fp16_ops.h | h | 1022 B |
torch/include/ATen/ops/_scaled_dot_product_attention_math.h | h | 1.27 KiB |
torch/include/ATen/ops/_scaled_dot_product_attention_math_compositeimplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_scaled_dot_product_attention_math_for_mps.h | h | 1.25 KiB |
torch/include/ATen/ops/_scaled_dot_product_attention_math_for_mps_native.h | h | 442 B |
torch/include/ATen/ops/_scaled_dot_product_attention_math_for_mps_ops.h | h | 1.83 KiB |
torch/include/ATen/ops/_scaled_dot_product_attention_math_native.h | h | 817 B |
torch/include/ATen/ops/_scaled_dot_product_attention_math_ops.h | h | 1.87 KiB |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention.h | h | 1.49 KiB |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_backward.h | h | 4.38 KiB |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_backward_cuda_dispatch.h | h | 1.72 KiB |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_backward_native.h | h | 976 B |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_backward_ops.h | h | 2.45 KiB |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_cuda_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_native.h | h | 878 B |
torch/include/ATen/ops/_scaled_dot_product_cudnn_attention_ops.h | h | 2.18 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention.h | h | 1.28 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_backward.h | h | 1.55 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_backward_cuda_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_backward_native.h | h | 946 B |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_backward_ops.h | h | 2.31 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_cuda_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_native.h | h | 1.13 KiB |
torch/include/ATen/ops/_scaled_dot_product_efficient_attention_ops.h | h | 1.86 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention.h | h | 1.34 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_backward.h | h | 4.25 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_backward_cuda_dispatch.h | h | 1.66 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_backward_native.h | h | 1.42 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_backward_ops.h | h | 2.38 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_cuda_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu.h | h | 1.18 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_backward.h | h | 1.38 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_backward_cpu_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_backward_native.h | h | 839 B |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_backward_ops.h | h | 2.03 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_cpu_dispatch.h | h | 996 B |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_native.h | h | 746 B |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_for_cpu_ops.h | h | 1.69 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_native.h | h | 1.16 KiB |
torch/include/ATen/ops/_scaled_dot_product_flash_attention_ops.h | h | 1.96 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable.h | h | 1.48 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_backward.h | h | 4.93 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_backward_compositeexplicitautograd_dispatch.h | h | 1.88 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_backward_native.h | h | 1.01 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_backward_ops.h | h | 2.69 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_compositeexplicitautograd_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_native.h | h | 864 B |
torch/include/ATen/ops/_scaled_dot_product_fused_attention_overrideable_ops.h | h | 2.15 KiB |
torch/include/ATen/ops/_scaled_mm.h | h | 2.44 KiB |
torch/include/ATen/ops/_scaled_mm_cuda_dispatch.h | h | 1.64 KiB |
torch/include/ATen/ops/_scaled_mm_native.h | h | 1.07 KiB |
torch/include/ATen/ops/_scaled_mm_ops.h | h | 3.10 KiB |
torch/include/ATen/ops/_segment_reduce_backward.h | h | 2.41 KiB |
torch/include/ATen/ops/_segment_reduce_backward_compositeexplicitautograd_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/_segment_reduce_backward_cpu_dispatch.h | h | 1010 B |
torch/include/ATen/ops/_segment_reduce_backward_cuda_dispatch.h | h | 1012 B |
torch/include/ATen/ops/_segment_reduce_backward_native.h | h | 1.07 KiB |
torch/include/ATen/ops/_segment_reduce_backward_ops.h | h | 3.10 KiB |
torch/include/ATen/ops/_shape_as_tensor.h | h | 684 B |
torch/include/ATen/ops/_shape_as_tensor_compositeimplicitautograd_dispatch.h | h | 797 B |
torch/include/ATen/ops/_shape_as_tensor_native.h | h | 507 B |
torch/include/ATen/ops/_shape_as_tensor_ops.h | h | 992 B |
torch/include/ATen/ops/_slow_conv2d_backward.h | h | 14.30 KiB |
torch/include/ATen/ops/_slow_conv2d_backward_compositeexplicitautograd_dispatch.h | h | 2.10 KiB |
torch/include/ATen/ops/_slow_conv2d_backward_cpu_dispatch.h | h | 2.55 KiB |
torch/include/ATen/ops/_slow_conv2d_backward_cuda_dispatch.h | h | 2.55 KiB |
torch/include/ATen/ops/_slow_conv2d_backward_native.h | h | 1.99 KiB |
torch/include/ATen/ops/_slow_conv2d_backward_ops.h | h | 4.77 KiB |
torch/include/ATen/ops/_slow_conv2d_forward.h | h | 6.69 KiB |
torch/include/ATen/ops/_slow_conv2d_forward_cpu_dispatch.h | h | 2.10 KiB |
torch/include/ATen/ops/_slow_conv2d_forward_cuda_dispatch.h | h | 2.10 KiB |
torch/include/ATen/ops/_slow_conv2d_forward_native.h | h | 1.34 KiB |
torch/include/ATen/ops/_slow_conv2d_forward_ops.h | h | 2.66 KiB |
torch/include/ATen/ops/_sobol_engine_draw.h | h | 980 B |
torch/include/ATen/ops/_sobol_engine_draw_compositeimplicitautograd_dispatch.h | h | 948 B |
torch/include/ATen/ops/_sobol_engine_draw_native.h | h | 658 B |
torch/include/ATen/ops/_sobol_engine_draw_ops.h | h | 1.46 KiB |
torch/include/ATen/ops/_sobol_engine_ff.h | h | 876 B |
torch/include/ATen/ops/_sobol_engine_ff_compositeimplicitautograd_dispatch.h | h | 878 B |
torch/include/ATen/ops/_sobol_engine_ff_native.h | h | 588 B |
torch/include/ATen/ops/_sobol_engine_ff_ops.h | h | 1.24 KiB |
torch/include/ATen/ops/_sobol_engine_initialize_state.h | h | 792 B |
torch/include/ATen/ops/_sobol_engine_initialize_state_compositeimplicitautograd_dispatch.h | h | 827 B |
torch/include/ATen/ops/_sobol_engine_initialize_state_native.h | h | 537 B |
torch/include/ATen/ops/_sobol_engine_initialize_state_ops.h | h | 1.07 KiB |
torch/include/ATen/ops/_sobol_engine_scramble.h | h | 801 B |
torch/include/ATen/ops/_sobol_engine_scramble_compositeimplicitautograd_dispatch.h | h | 843 B |
torch/include/ATen/ops/_sobol_engine_scramble_native.h | h | 553 B |
torch/include/ATen/ops/_sobol_engine_scramble_ops.h | h | 1.12 KiB |
torch/include/ATen/ops/_softmax.h | h | 1.28 KiB |
torch/include/ATen/ops/_softmax_backward_data.h | h | 1.69 KiB |
torch/include/ATen/ops/_softmax_backward_data_compositeexplicitautogradnonfunctional_dispatch.h | h | 904 B |
torch/include/ATen/ops/_softmax_backward_data_cpu_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_softmax_backward_data_cuda_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_softmax_backward_data_meta.h | h | 693 B |
torch/include/ATen/ops/_softmax_backward_data_meta_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_softmax_backward_data_native.h | h | 1.12 KiB |
torch/include/ATen/ops/_softmax_backward_data_ops.h | h | 2.13 KiB |
torch/include/ATen/ops/_softmax_compositeexplicitautogradnonfunctional_dispatch.h | h | 848 B |
torch/include/ATen/ops/_softmax_cpu_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_softmax_cuda_dispatch.h | h | 1009 B |
torch/include/ATen/ops/_softmax_meta.h | h | 637 B |
torch/include/ATen/ops/_softmax_meta_dispatch.h | h | 1009 B |
torch/include/ATen/ops/_softmax_native.h | h | 1.01 KiB |
torch/include/ATen/ops/_softmax_ops.h | h | 1.75 KiB |
torch/include/ATen/ops/_sparse_addmm.h | h | 1.63 KiB |
torch/include/ATen/ops/_sparse_addmm_compositeexplicitautograd_dispatch.h | h | 1.25 KiB |
torch/include/ATen/ops/_sparse_addmm_native.h | h | 796 B |
torch/include/ATen/ops/_sparse_addmm_ops.h | h | 2.23 KiB |
torch/include/ATen/ops/_sparse_broadcast_to.h | h | 746 B |
torch/include/ATen/ops/_sparse_broadcast_to_copy.h | h | 1.32 KiB |
torch/include/ATen/ops/_sparse_broadcast_to_copy_compositeexplicitautograd_dispatch.h | h | 973 B |
torch/include/ATen/ops/_sparse_broadcast_to_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 854 B |
torch/include/ATen/ops/_sparse_broadcast_to_copy_native.h | h | 658 B |
torch/include/ATen/ops/_sparse_broadcast_to_copy_ops.h | h | 1.78 KiB |
torch/include/ATen/ops/_sparse_broadcast_to_native.h | h | 532 B |
torch/include/ATen/ops/_sparse_broadcast_to_ops.h | h | 1.06 KiB |
torch/include/ATen/ops/_sparse_bsc_tensor_unsafe.h | h | 1.75 KiB |
torch/include/ATen/ops/_sparse_bsc_tensor_unsafe_compositeimplicitautograd_dispatch.h | h | 1.20 KiB |
torch/include/ATen/ops/_sparse_bsc_tensor_unsafe_native.h | h | 762 B |
torch/include/ATen/ops/_sparse_bsc_tensor_unsafe_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/_sparse_bsr_tensor_unsafe.h | h | 1.75 KiB |
torch/include/ATen/ops/_sparse_bsr_tensor_unsafe_compositeimplicitautograd_dispatch.h | h | 1.20 KiB |
torch/include/ATen/ops/_sparse_bsr_tensor_unsafe_native.h | h | 762 B |
torch/include/ATen/ops/_sparse_bsr_tensor_unsafe_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/_sparse_compressed_tensor_unsafe.h | h | 5.51 KiB |
torch/include/ATen/ops/_sparse_compressed_tensor_unsafe_compositeimplicitautograd_dispatch.h | h | 1.77 KiB |
torch/include/ATen/ops/_sparse_compressed_tensor_unsafe_native.h | h | 788 B |
torch/include/ATen/ops/_sparse_compressed_tensor_unsafe_ops.h | h | 1.82 KiB |
torch/include/ATen/ops/_sparse_compressed_tensor_with_dims.h | h | 1.83 KiB |
torch/include/ATen/ops/_sparse_compressed_tensor_with_dims_compositeexplicitautograd_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/_sparse_compressed_tensor_with_dims_native.h | h | 766 B |
torch/include/ATen/ops/_sparse_compressed_tensor_with_dims_ops.h | h | 1.78 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_unsafe.h | h | 5.23 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_unsafe_compositeimplicitautograd_dispatch.h | h | 1.74 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_unsafe_native.h | h | 787 B |
torch/include/ATen/ops/_sparse_coo_tensor_unsafe_ops.h | h | 1.78 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims.h | h | 2.29 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_and_tensors.h | h | 10.45 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_and_tensors_compositeexplicitautograd_dispatch.h | h | 1.71 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_and_tensors_meta_dispatch.h | h | 1.90 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_and_tensors_native.h | h | 1.06 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_and_tensors_ops.h | h | 3.06 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_meta_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_native.h | h | 841 B |
torch/include/ATen/ops/_sparse_coo_tensor_with_dims_ops.h | h | 2.37 KiB |
torch/include/ATen/ops/_sparse_csc_tensor_unsafe.h | h | 1.75 KiB |
torch/include/ATen/ops/_sparse_csc_tensor_unsafe_compositeimplicitautograd_dispatch.h | h | 1.20 KiB |
torch/include/ATen/ops/_sparse_csc_tensor_unsafe_native.h | h | 762 B |
torch/include/ATen/ops/_sparse_csc_tensor_unsafe_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/_sparse_csr_prod.h | h | 1.66 KiB |
torch/include/ATen/ops/_sparse_csr_prod_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_sparse_csr_prod_native.h | h | 944 B |
torch/include/ATen/ops/_sparse_csr_prod_ops.h | h | 2.15 KiB |
torch/include/ATen/ops/_sparse_csr_sum.h | h | 1.65 KiB |
torch/include/ATen/ops/_sparse_csr_sum_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_sparse_csr_sum_native.h | h | 941 B |
torch/include/ATen/ops/_sparse_csr_sum_ops.h | h | 2.15 KiB |
torch/include/ATen/ops/_sparse_csr_tensor_unsafe.h | h | 1.75 KiB |
torch/include/ATen/ops/_sparse_csr_tensor_unsafe_compositeimplicitautograd_dispatch.h | h | 1.20 KiB |
torch/include/ATen/ops/_sparse_csr_tensor_unsafe_native.h | h | 762 B |
torch/include/ATen/ops/_sparse_csr_tensor_unsafe_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/_sparse_log_softmax.h | h | 1.99 KiB |
torch/include/ATen/ops/_sparse_log_softmax_backward_data.h | h | 1.70 KiB |
torch/include/ATen/ops/_sparse_log_softmax_backward_data_compositeexplicitautograd_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_sparse_log_softmax_backward_data_native.h | h | 925 B |
torch/include/ATen/ops/_sparse_log_softmax_backward_data_ops.h | h | 2.15 KiB |
torch/include/ATen/ops/_sparse_log_softmax_compositeexplicitautograd_dispatch.h | h | 983 B |
torch/include/ATen/ops/_sparse_log_softmax_compositeimplicitautograd_dispatch.h | h | 1006 B |
torch/include/ATen/ops/_sparse_log_softmax_native.h | h | 1.03 KiB |
torch/include/ATen/ops/_sparse_log_softmax_ops.h | h | 3.29 KiB |
torch/include/ATen/ops/_sparse_mask_projection.h | h | 1.24 KiB |
torch/include/ATen/ops/_sparse_mask_projection_compositeexplicitautograd_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/_sparse_mask_projection_native.h | h | 715 B |
torch/include/ATen/ops/_sparse_mask_projection_ops.h | h | 1.96 KiB |
torch/include/ATen/ops/_sparse_mm.h | h | 977 B |
torch/include/ATen/ops/_sparse_mm_compositeimplicitautograd_dispatch.h | h | 931 B |
torch/include/ATen/ops/_sparse_mm_native.h | h | 641 B |
torch/include/ATen/ops/_sparse_mm_ops.h | h | 1.74 KiB |
torch/include/ATen/ops/_sparse_mm_reduce_impl.h | h | 835 B |
torch/include/ATen/ops/_sparse_mm_reduce_impl_backward.h | h | 1.03 KiB |
torch/include/ATen/ops/_sparse_mm_reduce_impl_backward_native.h | h | 705 B |
torch/include/ATen/ops/_sparse_mm_reduce_impl_backward_ops.h | h | 1.55 KiB |
torch/include/ATen/ops/_sparse_mm_reduce_impl_native.h | h | 604 B |
torch/include/ATen/ops/_sparse_mm_reduce_impl_ops.h | h | 1.23 KiB |
torch/include/ATen/ops/_sparse_semi_structured_addmm.h | h | 1.06 KiB |
torch/include/ATen/ops/_sparse_semi_structured_addmm_cuda_dispatch.h | h | 962 B |
torch/include/ATen/ops/_sparse_semi_structured_addmm_native.h | h | 714 B |
torch/include/ATen/ops/_sparse_semi_structured_addmm_ops.h | h | 1.58 KiB |
torch/include/ATen/ops/_sparse_semi_structured_apply.h | h | 842 B |
torch/include/ATen/ops/_sparse_semi_structured_apply_cuda_dispatch.h | h | 827 B |
torch/include/ATen/ops/_sparse_semi_structured_apply_dense.h | h | 831 B |
torch/include/ATen/ops/_sparse_semi_structured_apply_dense_cuda_dispatch.h | h | 808 B |
torch/include/ATen/ops/_sparse_semi_structured_apply_dense_native.h | h | 560 B |
torch/include/ATen/ops/_sparse_semi_structured_apply_dense_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/_sparse_semi_structured_apply_native.h | h | 579 B |
torch/include/ATen/ops/_sparse_semi_structured_apply_ops.h | h | 1.20 KiB |
torch/include/ATen/ops/_sparse_semi_structured_linear.h | h | 1.08 KiB |
torch/include/ATen/ops/_sparse_semi_structured_linear_cuda_dispatch.h | h | 986 B |
torch/include/ATen/ops/_sparse_semi_structured_linear_native.h | h | 738 B |
torch/include/ATen/ops/_sparse_semi_structured_linear_ops.h | h | 1.61 KiB |
torch/include/ATen/ops/_sparse_semi_structured_mm.h | h | 927 B |
torch/include/ATen/ops/_sparse_semi_structured_mm_cuda_dispatch.h | h | 878 B |
torch/include/ATen/ops/_sparse_semi_structured_mm_native.h | h | 630 B |
torch/include/ATen/ops/_sparse_semi_structured_mm_ops.h | h | 1.32 KiB |
torch/include/ATen/ops/_sparse_semi_structured_tile.h | h | 946 B |
torch/include/ATen/ops/_sparse_semi_structured_tile_cuda_dispatch.h | h | 880 B |
torch/include/ATen/ops/_sparse_semi_structured_tile_native.h | h | 632 B |
torch/include/ATen/ops/_sparse_semi_structured_tile_ops.h | h | 1.37 KiB |
torch/include/ATen/ops/_sparse_softmax.h | h | 1.93 KiB |
torch/include/ATen/ops/_sparse_softmax_backward_data.h | h | 1.66 KiB |
torch/include/ATen/ops/_sparse_softmax_backward_data_compositeexplicitautograd_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_sparse_softmax_backward_data_native.h | h | 913 B |
torch/include/ATen/ops/_sparse_softmax_backward_data_ops.h | h | 2.12 KiB |
torch/include/ATen/ops/_sparse_softmax_compositeexplicitautograd_dispatch.h | h | 975 B |
torch/include/ATen/ops/_sparse_softmax_compositeimplicitautograd_dispatch.h | h | 998 B |
torch/include/ATen/ops/_sparse_softmax_native.h | h | 1.01 KiB |
torch/include/ATen/ops/_sparse_softmax_ops.h | h | 3.25 KiB |
torch/include/ATen/ops/_sparse_sparse_matmul.h | h | 1.30 KiB |
torch/include/ATen/ops/_sparse_sparse_matmul_compositeexplicitautograd_dispatch.h | h | 973 B |
torch/include/ATen/ops/_sparse_sparse_matmul_native.h | h | 761 B |
torch/include/ATen/ops/_sparse_sparse_matmul_ops.h | h | 1.78 KiB |
torch/include/ATen/ops/_sparse_sum.h | h | 1.82 KiB |
torch/include/ATen/ops/_sparse_sum_backward.h | h | 1.39 KiB |
torch/include/ATen/ops/_sparse_sum_backward_compositeexplicitautograd_dispatch.h | h | 1011 B |
torch/include/ATen/ops/_sparse_sum_backward_native.h | h | 820 B |
torch/include/ATen/ops/_sparse_sum_backward_ops.h | h | 1.91 KiB |
torch/include/ATen/ops/_sparse_sum_compositeexplicitautograd_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/_sparse_sum_compositeimplicitautograd_dispatch.h | h | 977 B |
torch/include/ATen/ops/_sparse_sum_native.h | h | 877 B |
torch/include/ATen/ops/_sparse_sum_ops.h | h | 3.53 KiB |
torch/include/ATen/ops/_spdiags.h | h | 1.58 KiB |
torch/include/ATen/ops/_spdiags_compositeexplicitautograd_dispatch.h | h | 1.07 KiB |
torch/include/ATen/ops/_spdiags_cpu_dispatch.h | h | 852 B |
torch/include/ATen/ops/_spdiags_native.h | h | 778 B |
torch/include/ATen/ops/_spdiags_ops.h | h | 2.13 KiB |
torch/include/ATen/ops/_spsolve.h | h | 719 B |
torch/include/ATen/ops/_spsolve_native.h | h | 550 B |
torch/include/ATen/ops/_spsolve_ops.h | h | 1.05 KiB |
torch/include/ATen/ops/_stack.h | h | 1.13 KiB |
torch/include/ATen/ops/_stack_compositeexplicitautograd_dispatch.h | h | 986 B |
torch/include/ATen/ops/_stack_cpu_dispatch.h | h | 942 B |
torch/include/ATen/ops/_stack_native.h | h | 770 B |
torch/include/ATen/ops/_stack_ops.h | h | 1.61 KiB |
torch/include/ATen/ops/_standard_gamma.h | h | 1.37 KiB |
torch/include/ATen/ops/_standard_gamma_compositeexplicitautograd_dispatch.h | h | 1008 B |
torch/include/ATen/ops/_standard_gamma_cpu_dispatch.h | h | 809 B |
torch/include/ATen/ops/_standard_gamma_cuda_dispatch.h | h | 811 B |
torch/include/ATen/ops/_standard_gamma_grad.h | h | 1.30 KiB |
torch/include/ATen/ops/_standard_gamma_grad_compositeexplicitautograd_dispatch.h | h | 973 B |
torch/include/ATen/ops/_standard_gamma_grad_cpu_dispatch.h | h | 784 B |
torch/include/ATen/ops/_standard_gamma_grad_cuda_dispatch.h | h | 786 B |
torch/include/ATen/ops/_standard_gamma_grad_native.h | h | 763 B |
torch/include/ATen/ops/_standard_gamma_grad_ops.h | h | 1.78 KiB |
torch/include/ATen/ops/_standard_gamma_native.h | h | 809 B |
torch/include/ATen/ops/_standard_gamma_ops.h | h | 1.86 KiB |
torch/include/ATen/ops/_test_ambiguous_defaults.h | h | 1.01 KiB |
torch/include/ATen/ops/_test_ambiguous_defaults_compositeimplicitautograd_dispatch.h | h | 937 B |
torch/include/ATen/ops/_test_ambiguous_defaults_native.h | h | 653 B |
torch/include/ATen/ops/_test_ambiguous_defaults_ops.h | h | 1.75 KiB |
torch/include/ATen/ops/_test_autograd_multiple_dispatch.h | h | 1.59 KiB |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_compositeexplicitautograd_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_compositeimplicitautograd_dispatch.h | h | 821 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_native.h | h | 750 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_ops.h | h | 2.38 KiB |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view.h | h | 774 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_compositeexplicitautograd_dispatch.h | h | 818 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_copy.h | h | 1.37 KiB |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_copy_compositeexplicitautograd_dispatch.h | h | 963 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 849 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_copy_native.h | h | 648 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_copy_ops.h | h | 1.74 KiB |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_native.h | h | 528 B |
torch/include/ATen/ops/_test_autograd_multiple_dispatch_view_ops.h | h | 1.04 KiB |
torch/include/ATen/ops/_test_check_tensor.h | h | 692 B |
torch/include/ATen/ops/_test_check_tensor_compositeimplicitautograd_dispatch.h | h | 799 B |
torch/include/ATen/ops/_test_check_tensor_native.h | h | 509 B |
torch/include/ATen/ops/_test_check_tensor_ops.h | h | 998 B |
torch/include/ATen/ops/_test_functorch_fallback.h | h | 1.33 KiB |
torch/include/ATen/ops/_test_functorch_fallback_compositeexplicitautograd_dispatch.h | h | 979 B |
torch/include/ATen/ops/_test_functorch_fallback_cpu_dispatch.h | h | 787 B |
torch/include/ATen/ops/_test_functorch_fallback_native.h | h | 664 B |
torch/include/ATen/ops/_test_functorch_fallback_ops.h | h | 1.80 KiB |
torch/include/ATen/ops/_test_optional_filled_intlist.h | h | 1.43 KiB |
torch/include/ATen/ops/_test_optional_filled_intlist_compositeexplicitautograd_dispatch.h | h | 1007 B |
torch/include/ATen/ops/_test_optional_filled_intlist_cpu_dispatch.h | h | 801 B |
torch/include/ATen/ops/_test_optional_filled_intlist_native.h | h | 685 B |
torch/include/ATen/ops/_test_optional_filled_intlist_ops.h | h | 1.88 KiB |
torch/include/ATen/ops/_test_optional_floatlist.h | h | 1.43 KiB |
torch/include/ATen/ops/_test_optional_floatlist_compositeexplicitautograd_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/_test_optional_floatlist_cpu_dispatch.h | h | 810 B |
torch/include/ATen/ops/_test_optional_floatlist_native.h | h | 710 B |
torch/include/ATen/ops/_test_optional_floatlist_ops.h | h | 1.94 KiB |
torch/include/ATen/ops/_test_optional_intlist.h | h | 1.36 KiB |
torch/include/ATen/ops/_test_optional_intlist_compositeexplicitautograd_dispatch.h | h | 993 B |
torch/include/ATen/ops/_test_optional_intlist_cpu_dispatch.h | h | 794 B |
torch/include/ATen/ops/_test_optional_intlist_native.h | h | 678 B |
torch/include/ATen/ops/_test_optional_intlist_ops.h | h | 1.84 KiB |
torch/include/ATen/ops/_test_parallel_materialize.h | h | 836 B |
torch/include/ATen/ops/_test_parallel_materialize_compositeexplicitautograd_dispatch.h | h | 852 B |
torch/include/ATen/ops/_test_parallel_materialize_native.h | h | 562 B |
torch/include/ATen/ops/_test_parallel_materialize_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/_test_serialization_subcmul.h | h | 826 B |
torch/include/ATen/ops/_test_serialization_subcmul_compositeimplicitautograd_dispatch.h | h | 862 B |
torch/include/ATen/ops/_test_serialization_subcmul_native.h | h | 572 B |
torch/include/ATen/ops/_test_serialization_subcmul_ops.h | h | 1.17 KiB |
torch/include/ATen/ops/_test_string_default.h | h | 795 B |
torch/include/ATen/ops/_test_string_default_compositeimplicitautograd_dispatch.h | h | 858 B |
torch/include/ATen/ops/_test_string_default_native.h | h | 568 B |
torch/include/ATen/ops/_test_string_default_ops.h | h | 1.14 KiB |
torch/include/ATen/ops/_test_warn_in_autograd.h | h | 1.17 KiB |
torch/include/ATen/ops/_test_warn_in_autograd_compositeexplicitautograd_dispatch.h | h | 994 B |
torch/include/ATen/ops/_test_warn_in_autograd_native.h | h | 608 B |
torch/include/ATen/ops/_test_warn_in_autograd_ops.h | h | 1.62 KiB |
torch/include/ATen/ops/_thnn_differentiable_gru_cell_backward.h | h | 1.18 KiB |
torch/include/ATen/ops/_thnn_differentiable_gru_cell_backward_compositeimplicitautograd_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_thnn_differentiable_gru_cell_backward_native.h | h | 775 B |
torch/include/ATen/ops/_thnn_differentiable_gru_cell_backward_ops.h | h | 1.83 KiB |
torch/include/ATen/ops/_thnn_differentiable_lstm_cell_backward.h | h | 1.31 KiB |
torch/include/ATen/ops/_thnn_differentiable_lstm_cell_backward_compositeimplicitautograd_dispatch.h | h | 1.12 KiB |
torch/include/ATen/ops/_thnn_differentiable_lstm_cell_backward_native.h | h | 861 B |
torch/include/ATen/ops/_thnn_differentiable_lstm_cell_backward_ops.h | h | 2.10 KiB |
torch/include/ATen/ops/_thnn_fused_gru_cell.h | h | 2.25 KiB |
torch/include/ATen/ops/_thnn_fused_gru_cell_backward.h | h | 2.21 KiB |
torch/include/ATen/ops/_thnn_fused_gru_cell_backward_compositeexplicitautograd_dispatch.h | h | 1.29 KiB |
torch/include/ATen/ops/_thnn_fused_gru_cell_backward_cuda_dispatch.h | h | 874 B |
torch/include/ATen/ops/_thnn_fused_gru_cell_backward_native.h | h | 924 B |
torch/include/ATen/ops/_thnn_fused_gru_cell_backward_ops.h | h | 2.69 KiB |
torch/include/ATen/ops/_thnn_fused_gru_cell_compositeexplicitautograd_dispatch.h | h | 1.31 KiB |
torch/include/ATen/ops/_thnn_fused_gru_cell_cuda_dispatch.h | h | 950 B |
torch/include/ATen/ops/_thnn_fused_gru_cell_native.h | h | 1007 B |
torch/include/ATen/ops/_thnn_fused_gru_cell_ops.h | h | 2.88 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell.h | h | 2.41 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward.h | h | 1.08 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_compositeimplicitautograd_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_impl.h | h | 2.45 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_impl_compositeexplicitautograd_dispatch.h | h | 1.39 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_impl_cuda_dispatch.h | h | 966 B |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_impl_native.h | h | 1.04 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_impl_ops.h | h | 3.09 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_native.h | h | 735 B |
torch/include/ATen/ops/_thnn_fused_lstm_cell_backward_ops.h | h | 1.71 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_compositeexplicitautograd_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_cuda_dispatch.h | h | 962 B |
torch/include/ATen/ops/_thnn_fused_lstm_cell_native.h | h | 1.03 KiB |
torch/include/ATen/ops/_thnn_fused_lstm_cell_ops.h | h | 3.04 KiB |
torch/include/ATen/ops/_to_copy.h | h | 2.40 KiB |
torch/include/ATen/ops/_to_copy_compositeexplicitautograd_dispatch.h | h | 1.46 KiB |
torch/include/ATen/ops/_to_copy_native.h | h | 1.18 KiB |
torch/include/ATen/ops/_to_copy_ops.h | h | 2.48 KiB |
torch/include/ATen/ops/_to_cpu.h | h | 672 B |
torch/include/ATen/ops/_to_cpu_compositeimplicitautograd_dispatch.h | h | 802 B |
torch/include/ATen/ops/_to_cpu_native.h | h | 512 B |
torch/include/ATen/ops/_to_cpu_ops.h | h | 1011 B |
torch/include/ATen/ops/_to_dense.h | h | 1.21 KiB |
torch/include/ATen/ops/_to_dense_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_to_dense_native.h | h | 1.09 KiB |
torch/include/ATen/ops/_to_dense_ops.h | h | 2.04 KiB |
torch/include/ATen/ops/_to_sparse.h | h | 1.90 KiB |
torch/include/ATen/ops/_to_sparse_bsc.h | h | 1.19 KiB |
torch/include/ATen/ops/_to_sparse_bsc_compositeexplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/_to_sparse_bsc_cpu_dispatch.h | h | 829 B |
torch/include/ATen/ops/_to_sparse_bsc_cuda_dispatch.h | h | 831 B |
torch/include/ATen/ops/_to_sparse_bsc_native.h | h | 1.02 KiB |
torch/include/ATen/ops/_to_sparse_bsc_ops.h | h | 1.98 KiB |
torch/include/ATen/ops/_to_sparse_bsr.h | h | 1.19 KiB |
torch/include/ATen/ops/_to_sparse_bsr_compositeexplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/_to_sparse_bsr_cpu_dispatch.h | h | 829 B |
torch/include/ATen/ops/_to_sparse_bsr_cuda_dispatch.h | h | 831 B |
torch/include/ATen/ops/_to_sparse_bsr_native.h | h | 1.02 KiB |
torch/include/ATen/ops/_to_sparse_bsr_ops.h | h | 1.98 KiB |
torch/include/ATen/ops/_to_sparse_compositeexplicitautograd_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/_to_sparse_cpu_dispatch.h | h | 978 B |
torch/include/ATen/ops/_to_sparse_csc.h | h | 1.08 KiB |
torch/include/ATen/ops/_to_sparse_csc_compositeexplicitautograd_dispatch.h | h | 994 B |
torch/include/ATen/ops/_to_sparse_csc_cpu_dispatch.h | h | 802 B |
torch/include/ATen/ops/_to_sparse_csc_cuda_dispatch.h | h | 804 B |
torch/include/ATen/ops/_to_sparse_csc_native.h | h | 932 B |
torch/include/ATen/ops/_to_sparse_csc_ops.h | h | 1.81 KiB |
torch/include/ATen/ops/_to_sparse_csr.h | h | 1.08 KiB |
torch/include/ATen/ops/_to_sparse_csr_compositeexplicitautograd_dispatch.h | h | 994 B |
torch/include/ATen/ops/_to_sparse_csr_cpu_dispatch.h | h | 802 B |
torch/include/ATen/ops/_to_sparse_csr_cuda_dispatch.h | h | 804 B |
torch/include/ATen/ops/_to_sparse_csr_native.h | h | 932 B |
torch/include/ATen/ops/_to_sparse_csr_ops.h | h | 1.81 KiB |
torch/include/ATen/ops/_to_sparse_cuda_dispatch.h | h | 980 B |
torch/include/ATen/ops/_to_sparse_native.h | h | 1.64 KiB |
torch/include/ATen/ops/_to_sparse_ops.h | h | 3.56 KiB |
torch/include/ATen/ops/_to_sparse_semi_structured.h | h | 762 B |
torch/include/ATen/ops/_to_sparse_semi_structured_cuda_dispatch.h | h | 791 B |
torch/include/ATen/ops/_to_sparse_semi_structured_native.h | h | 543 B |
torch/include/ATen/ops/_to_sparse_semi_structured_ops.h | h | 1.08 KiB |
torch/include/ATen/ops/_transform_bias_rescale_qkv.h | h | 1.86 KiB |
torch/include/ATen/ops/_transform_bias_rescale_qkv_compositeexplicitautograd_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/_transform_bias_rescale_qkv_cpu_dispatch.h | h | 847 B |
torch/include/ATen/ops/_transform_bias_rescale_qkv_cuda_dispatch.h | h | 849 B |
torch/include/ATen/ops/_transform_bias_rescale_qkv_native.h | h | 993 B |
torch/include/ATen/ops/_transform_bias_rescale_qkv_ops.h | h | 2.35 KiB |
torch/include/ATen/ops/_transformer_encoder_layer_fwd.h | h | 4.54 KiB |
torch/include/ATen/ops/_transformer_encoder_layer_fwd_compositeexplicitautograd_dispatch.h | h | 1.99 KiB |
torch/include/ATen/ops/_transformer_encoder_layer_fwd_cpu_dispatch.h | h | 1.29 KiB |
torch/include/ATen/ops/_transformer_encoder_layer_fwd_cuda_dispatch.h | h | 1.30 KiB |
torch/include/ATen/ops/_transformer_encoder_layer_fwd_native.h | h | 1.69 KiB |
torch/include/ATen/ops/_transformer_encoder_layer_fwd_ops.h | h | 5.13 KiB |
torch/include/ATen/ops/_trilinear.h | h | 1.98 KiB |
torch/include/ATen/ops/_trilinear_compositeexplicitautograd_dispatch.h | h | 1.20 KiB |
torch/include/ATen/ops/_trilinear_compositeexplicitautogradnonfunctional_dispatch.h | h | 982 B |
torch/include/ATen/ops/_trilinear_native.h | h | 912 B |
torch/include/ATen/ops/_trilinear_ops.h | h | 2.60 KiB |
torch/include/ATen/ops/_triton_multi_head_attention.h | h | 2.67 KiB |
torch/include/ATen/ops/_triton_multi_head_attention_compositeexplicitautograd_dispatch.h | h | 1.41 KiB |
torch/include/ATen/ops/_triton_multi_head_attention_cuda_dispatch.h | h | 1022 B |
torch/include/ATen/ops/_triton_multi_head_attention_native.h | h | 1.10 KiB |
torch/include/ATen/ops/_triton_multi_head_attention_ops.h | h | 3.27 KiB |
torch/include/ATen/ops/_triton_scaled_dot_attention.h | h | 1.57 KiB |
torch/include/ATen/ops/_triton_scaled_dot_attention_compositeexplicitautograd_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/_triton_scaled_dot_attention_cuda_dispatch.h | h | 830 B |
torch/include/ATen/ops/_triton_scaled_dot_attention_native.h | h | 741 B |
torch/include/ATen/ops/_triton_scaled_dot_attention_ops.h | h | 2.05 KiB |
torch/include/ATen/ops/_unique.h | h | 1.55 KiB |
torch/include/ATen/ops/_unique2.h | h | 1.90 KiB |
torch/include/ATen/ops/_unique2_compositeexplicitautograd_dispatch.h | h | 1.15 KiB |
torch/include/ATen/ops/_unique2_cpu_dispatch.h | h | 852 B |
torch/include/ATen/ops/_unique2_cuda_dispatch.h | h | 854 B |
torch/include/ATen/ops/_unique2_native.h | h | 993 B |
torch/include/ATen/ops/_unique2_ops.h | h | 2.33 KiB |
torch/include/ATen/ops/_unique_compositeexplicitautograd_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_unique_cpu_dispatch.h | h | 814 B |
torch/include/ATen/ops/_unique_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/_unique_native.h | h | 864 B |
torch/include/ATen/ops/_unique_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/_unpack_dual.h | h | 757 B |
torch/include/ATen/ops/_unpack_dual_compositeimplicitautograd_dispatch.h | h | 833 B |
torch/include/ATen/ops/_unpack_dual_native.h | h | 543 B |
torch/include/ATen/ops/_unpack_dual_ops.h | h | 1.11 KiB |
torch/include/ATen/ops/_unsafe_index.h | h | 770 B |
torch/include/ATen/ops/_unsafe_index_compositeexplicitautograd_dispatch.h | h | 850 B |
torch/include/ATen/ops/_unsafe_index_native.h | h | 560 B |
torch/include/ATen/ops/_unsafe_index_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/_unsafe_index_put.h | h | 880 B |
torch/include/ATen/ops/_unsafe_index_put_compositeexplicitautograd_dispatch.h | h | 904 B |
torch/include/ATen/ops/_unsafe_index_put_native.h | h | 614 B |
torch/include/ATen/ops/_unsafe_index_put_ops.h | h | 1.29 KiB |
torch/include/ATen/ops/_unsafe_masked_index.h | h | 872 B |
torch/include/ATen/ops/_unsafe_masked_index_compositeexplicitautograd_dispatch.h | h | 907 B |
torch/include/ATen/ops/_unsafe_masked_index_native.h | h | 617 B |
torch/include/ATen/ops/_unsafe_masked_index_ops.h | h | 1.32 KiB |
torch/include/ATen/ops/_unsafe_masked_index_put_accumulate.h | h | 938 B |
torch/include/ATen/ops/_unsafe_masked_index_put_accumulate_compositeexplicitautograd_dispatch.h | h | 924 B |
torch/include/ATen/ops/_unsafe_masked_index_put_accumulate_native.h | h | 634 B |
torch/include/ATen/ops/_unsafe_masked_index_put_accumulate_ops.h | h | 1.37 KiB |
torch/include/ATen/ops/_unsafe_view.h | h | 3.66 KiB |
torch/include/ATen/ops/_unsafe_view_compositeexplicitautograd_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/_unsafe_view_native.h | h | 643 B |
torch/include/ATen/ops/_unsafe_view_ops.h | h | 1.73 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa.h | h | 8.00 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward.h | h | 7.69 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_cpu_dispatch.h | h | 2.32 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_cuda_dispatch.h | h | 2.32 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_meta.h | h | 791 B |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_meta_dispatch.h | h | 2.32 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_native.h | h | 1.21 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_backward_ops.h | h | 2.76 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_compositeimplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_cpu_dispatch.h | h | 2.02 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_cuda_dispatch.h | h | 2.02 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_meta.h | h | 741 B |
torch/include/ATen/ops/_upsample_bicubic2d_aa_meta_dispatch.h | h | 2.02 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_native.h | h | 1.25 KiB |
torch/include/ATen/ops/_upsample_bicubic2d_aa_ops.h | h | 3.33 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa.h | h | 8.04 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward.h | h | 7.72 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_cpu_dispatch.h | h | 2.32 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_cuda_dispatch.h | h | 2.32 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_meta.h | h | 792 B |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_meta_dispatch.h | h | 2.32 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_native.h | h | 1.21 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_backward_ops.h | h | 2.76 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_compositeimplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_cpu_dispatch.h | h | 2.02 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_cuda_dispatch.h | h | 2.03 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_meta.h | h | 742 B |
torch/include/ATen/ops/_upsample_bilinear2d_aa_meta_dispatch.h | h | 2.03 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_native.h | h | 1.26 KiB |
torch/include/ATen/ops/_upsample_bilinear2d_aa_ops.h | h | 3.34 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d.h | h | 6.60 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward.h | h | 6.43 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_cpu_dispatch.h | h | 1.95 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_cuda_dispatch.h | h | 1.95 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_meta.h | h | 738 B |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_meta_dispatch.h | h | 1.95 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_native.h | h | 1.12 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_backward_ops.h | h | 2.41 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_compositeimplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_cpu_dispatch.h | h | 1.65 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_cuda_dispatch.h | h | 1.65 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_meta.h | h | 688 B |
torch/include/ATen/ops/_upsample_nearest_exact1d_meta_dispatch.h | h | 1.65 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_native.h | h | 1.14 KiB |
torch/include/ATen/ops/_upsample_nearest_exact1d_ops.h | h | 2.93 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d.h | h | 7.42 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward.h | h | 7.25 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.23 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_cpu_dispatch.h | h | 2.22 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_cuda_dispatch.h | h | 2.22 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_meta.h | h | 774 B |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_meta_dispatch.h | h | 2.22 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_native.h | h | 1.19 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_backward_ops.h | h | 2.65 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_compositeimplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_cpu_dispatch.h | h | 1.92 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_cuda_dispatch.h | h | 1.92 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_meta.h | h | 724 B |
torch/include/ATen/ops/_upsample_nearest_exact2d_meta_dispatch.h | h | 1.92 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_native.h | h | 1.42 KiB |
torch/include/ATen/ops/_upsample_nearest_exact2d_ops.h | h | 3.16 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d.h | h | 8.18 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward.h | h | 8.02 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_cpu_dispatch.h | h | 2.47 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_cuda_dispatch.h | h | 2.48 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_meta.h | h | 808 B |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_meta_dispatch.h | h | 2.48 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_native.h | h | 1.25 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_backward_ops.h | h | 2.87 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.24 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_compositeimplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_cpu_dispatch.h | h | 2.18 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_cuda_dispatch.h | h | 2.18 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_meta.h | h | 758 B |
torch/include/ATen/ops/_upsample_nearest_exact3d_meta_dispatch.h | h | 2.18 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_native.h | h | 1.53 KiB |
torch/include/ATen/ops/_upsample_nearest_exact3d_ops.h | h | 3.39 KiB |
torch/include/ATen/ops/_use_cudnn_ctc_loss.h | h | 1.33 KiB |
torch/include/ATen/ops/_use_cudnn_ctc_loss_cuda_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/_use_cudnn_ctc_loss_native.h | h | 801 B |
torch/include/ATen/ops/_use_cudnn_ctc_loss_ops.h | h | 2.22 KiB |
torch/include/ATen/ops/_use_cudnn_rnn_flatten_weight.h | h | 690 B |
torch/include/ATen/ops/_use_cudnn_rnn_flatten_weight_compositeimplicitautograd_dispatch.h | h | 781 B |
torch/include/ATen/ops/_use_cudnn_rnn_flatten_weight_native.h | h | 491 B |
torch/include/ATen/ops/_use_cudnn_rnn_flatten_weight_ops.h | h | 934 B |
torch/include/ATen/ops/_validate_compressed_sparse_indices.h | h | 960 B |
torch/include/ATen/ops/_validate_compressed_sparse_indices_cpu_dispatch.h | h | 860 B |
torch/include/ATen/ops/_validate_compressed_sparse_indices_cuda_dispatch.h | h | 862 B |
torch/include/ATen/ops/_validate_compressed_sparse_indices_native.h | h | 795 B |
torch/include/ATen/ops/_validate_compressed_sparse_indices_ops.h | h | 1.31 KiB |
torch/include/ATen/ops/_validate_sparse_bsc_tensor_args.h | h | 917 B |
torch/include/ATen/ops/_validate_sparse_bsc_tensor_args_compositeimplicitautograd_dispatch.h | h | 896 B |
torch/include/ATen/ops/_validate_sparse_bsc_tensor_args_native.h | h | 606 B |
torch/include/ATen/ops/_validate_sparse_bsc_tensor_args_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/_validate_sparse_bsr_tensor_args.h | h | 917 B |
torch/include/ATen/ops/_validate_sparse_bsr_tensor_args_compositeimplicitautograd_dispatch.h | h | 896 B |
torch/include/ATen/ops/_validate_sparse_bsr_tensor_args_native.h | h | 606 B |
torch/include/ATen/ops/_validate_sparse_bsr_tensor_args_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/_validate_sparse_compressed_tensor_args.h | h | 1011 B |
torch/include/ATen/ops/_validate_sparse_compressed_tensor_args_compositeimplicitautograd_dispatch.h | h | 930 B |
torch/include/ATen/ops/_validate_sparse_compressed_tensor_args_native.h | h | 640 B |
torch/include/ATen/ops/_validate_sparse_compressed_tensor_args_ops.h | h | 1.38 KiB |
torch/include/ATen/ops/_validate_sparse_coo_tensor_args.h | h | 927 B |
torch/include/ATen/ops/_validate_sparse_coo_tensor_args_compositeimplicitautograd_dispatch.h | h | 910 B |
torch/include/ATen/ops/_validate_sparse_coo_tensor_args_native.h | h | 620 B |
torch/include/ATen/ops/_validate_sparse_coo_tensor_args_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/_validate_sparse_csc_tensor_args.h | h | 917 B |
torch/include/ATen/ops/_validate_sparse_csc_tensor_args_compositeimplicitautograd_dispatch.h | h | 896 B |
torch/include/ATen/ops/_validate_sparse_csc_tensor_args_native.h | h | 606 B |
torch/include/ATen/ops/_validate_sparse_csc_tensor_args_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/_validate_sparse_csr_tensor_args.h | h | 917 B |
torch/include/ATen/ops/_validate_sparse_csr_tensor_args_compositeimplicitautograd_dispatch.h | h | 896 B |
torch/include/ATen/ops/_validate_sparse_csr_tensor_args_native.h | h | 606 B |
torch/include/ATen/ops/_validate_sparse_csr_tensor_args_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/_values.h | h | 507 B |
torch/include/ATen/ops/_values_copy.h | h | 1.08 KiB |
torch/include/ATen/ops/_values_copy_compositeexplicitautograd_dispatch.h | h | 903 B |
torch/include/ATen/ops/_values_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 819 B |
torch/include/ATen/ops/_values_copy_native.h | h | 588 B |
torch/include/ATen/ops/_values_copy_ops.h | h | 1.56 KiB |
torch/include/ATen/ops/_values_native.h | h | 505 B |
torch/include/ATen/ops/_values_ops.h | h | 971 B |
torch/include/ATen/ops/_version.h | h | 508 B |
torch/include/ATen/ops/_version_compositeimplicitautograd_dispatch.h | h | 786 B |
torch/include/ATen/ops/_version_native.h | h | 496 B |
torch/include/ATen/ops/_version_ops.h | h | 956 B |
torch/include/ATen/ops/_weight_int4pack_mm.h | h | 862 B |
torch/include/ATen/ops/_weight_int4pack_mm_cpu_dispatch.h | h | 836 B |
torch/include/ATen/ops/_weight_int4pack_mm_cuda_dispatch.h | h | 838 B |
torch/include/ATen/ops/_weight_int4pack_mm_for_cpu.h | h | 894 B |
torch/include/ATen/ops/_weight_int4pack_mm_for_cpu_cpu_dispatch.h | h | 844 B |
torch/include/ATen/ops/_weight_int4pack_mm_for_cpu_native.h | h | 594 B |
torch/include/ATen/ops/_weight_int4pack_mm_for_cpu_ops.h | h | 1.26 KiB |
torch/include/ATen/ops/_weight_int4pack_mm_native.h | h | 595 B |
torch/include/ATen/ops/_weight_int4pack_mm_ops.h | h | 1.23 KiB |
torch/include/ATen/ops/_weight_int8pack_mm.h | h | 790 B |
torch/include/ATen/ops/_weight_int8pack_mm_cpu_dispatch.h | h | 808 B |
torch/include/ATen/ops/_weight_int8pack_mm_native.h | h | 566 B |
torch/include/ATen/ops/_weight_int8pack_mm_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/_weight_norm.h | h | 725 B |
torch/include/ATen/ops/_weight_norm_compositeimplicitautograd_dispatch.h | h | 827 B |
torch/include/ATen/ops/_weight_norm_differentiable_backward.h | h | 1003 B |
torch/include/ATen/ops/_weight_norm_differentiable_backward_compositeimplicitautograd_dispatch.h | h | 945 B |
torch/include/ATen/ops/_weight_norm_differentiable_backward_native.h | h | 655 B |
torch/include/ATen/ops/_weight_norm_differentiable_backward_ops.h | h | 1.44 KiB |
torch/include/ATen/ops/_weight_norm_interface.h | h | 1.54 KiB |
torch/include/ATen/ops/_weight_norm_interface_backward.h | h | 2.06 KiB |
torch/include/ATen/ops/_weight_norm_interface_backward_compositeexplicitautograd_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/_weight_norm_interface_backward_cpu_dispatch.h | h | 896 B |
torch/include/ATen/ops/_weight_norm_interface_backward_cuda_dispatch.h | h | 898 B |
torch/include/ATen/ops/_weight_norm_interface_backward_native.h | h | 1.07 KiB |
torch/include/ATen/ops/_weight_norm_interface_backward_ops.h | h | 2.58 KiB |
torch/include/ATen/ops/_weight_norm_interface_compositeexplicitautograd_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/_weight_norm_interface_cpu_dispatch.h | h | 818 B |
torch/include/ATen/ops/_weight_norm_interface_cuda_dispatch.h | h | 820 B |
torch/include/ATen/ops/_weight_norm_interface_native.h | h | 863 B |
torch/include/ATen/ops/_weight_norm_interface_ops.h | h | 2.08 KiB |
torch/include/ATen/ops/_weight_norm_native.h | h | 537 B |
torch/include/ATen/ops/_weight_norm_ops.h | h | 1.07 KiB |
torch/include/ATen/ops/_wrapped_linear_prepack.h | h | 913 B |
torch/include/ATen/ops/_wrapped_linear_prepack_compositeimplicitautograd_dispatch.h | h | 902 B |
torch/include/ATen/ops/_wrapped_linear_prepack_native.h | h | 612 B |
torch/include/ATen/ops/_wrapped_linear_prepack_ops.h | h | 1.30 KiB |
torch/include/ATen/ops/_wrapped_quantized_linear_prepacked.h | h | 1.15 KiB |
torch/include/ATen/ops/_wrapped_quantized_linear_prepacked_compositeimplicitautograd_dispatch.h | h | 1012 B |
torch/include/ATen/ops/_wrapped_quantized_linear_prepacked_native.h | h | 722 B |
torch/include/ATen/ops/_wrapped_quantized_linear_prepacked_ops.h | h | 1.64 KiB |
torch/include/ATen/ops/abs.h | h | 1.12 KiB |
torch/include/ATen/ops/abs_compositeexplicitautograd_dispatch.h | h | 833 B |
torch/include/ATen/ops/abs_cpu_dispatch.h | h | 841 B |
torch/include/ATen/ops/abs_cuda_dispatch.h | h | 843 B |
torch/include/ATen/ops/abs_native.h | h | 1.13 KiB |
torch/include/ATen/ops/abs_ops.h | h | 2.00 KiB |
torch/include/ATen/ops/absolute.h | h | 1.04 KiB |
torch/include/ATen/ops/absolute_compositeimplicitautograd_dispatch.h | h | 1006 B |
torch/include/ATen/ops/absolute_native.h | h | 634 B |
torch/include/ATen/ops/absolute_ops.h | h | 2.04 KiB |
torch/include/ATen/ops/acos.h | h | 1.14 KiB |
torch/include/ATen/ops/acos_compositeexplicitautogradnonfunctional_dispatch.h | h | 861 B |
torch/include/ATen/ops/acos_cpu_dispatch.h | h | 946 B |
torch/include/ATen/ops/acos_cuda_dispatch.h | h | 948 B |
torch/include/ATen/ops/acos_meta.h | h | 600 B |
torch/include/ATen/ops/acos_meta_dispatch.h | h | 948 B |
torch/include/ATen/ops/acos_native.h | h | 613 B |
torch/include/ATen/ops/acos_ops.h | h | 2.01 KiB |
torch/include/ATen/ops/acosh.h | h | 1.15 KiB |
torch/include/ATen/ops/acosh_compositeexplicitautogradnonfunctional_dispatch.h | h | 863 B |
torch/include/ATen/ops/acosh_cpu_dispatch.h | h | 950 B |
torch/include/ATen/ops/acosh_cuda_dispatch.h | h | 952 B |
torch/include/ATen/ops/acosh_meta.h | h | 601 B |
torch/include/ATen/ops/acosh_meta_dispatch.h | h | 952 B |
torch/include/ATen/ops/acosh_native.h | h | 616 B |
torch/include/ATen/ops/acosh_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/adaptive_avg_pool1d.h | h | 758 B |
torch/include/ATen/ops/adaptive_avg_pool1d_compositeimplicitautograd_dispatch.h | h | 829 B |
torch/include/ATen/ops/adaptive_avg_pool1d_native.h | h | 539 B |
torch/include/ATen/ops/adaptive_avg_pool1d_ops.h | h | 1.07 KiB |
torch/include/ATen/ops/adaptive_avg_pool2d.h | h | 4.08 KiB |
torch/include/ATen/ops/adaptive_avg_pool2d_compositeimplicitautograd_dispatch.h | h | 937 B |
torch/include/ATen/ops/adaptive_avg_pool2d_cpu_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/adaptive_avg_pool2d_cuda_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/adaptive_avg_pool2d_native.h | h | 934 B |
torch/include/ATen/ops/adaptive_avg_pool2d_ops.h | h | 1.82 KiB |
torch/include/ATen/ops/adaptive_avg_pool3d.h | h | 4.08 KiB |
torch/include/ATen/ops/adaptive_avg_pool3d_backward.h | h | 1.23 KiB |
torch/include/ATen/ops/adaptive_avg_pool3d_backward_cpu_dispatch.h | h | 969 B |
torch/include/ATen/ops/adaptive_avg_pool3d_backward_cuda_dispatch.h | h | 971 B |
torch/include/ATen/ops/adaptive_avg_pool3d_backward_native.h | h | 731 B |
torch/include/ATen/ops/adaptive_avg_pool3d_backward_ops.h | h | 1.23 KiB |
torch/include/ATen/ops/adaptive_avg_pool3d_compositeimplicitautograd_dispatch.h | h | 937 B |
torch/include/ATen/ops/adaptive_avg_pool3d_cpu_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/adaptive_avg_pool3d_cuda_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/adaptive_avg_pool3d_native.h | h | 936 B |
torch/include/ATen/ops/adaptive_avg_pool3d_ops.h | h | 1.82 KiB |
torch/include/ATen/ops/adaptive_max_pool1d.h | h | 793 B |
torch/include/ATen/ops/adaptive_max_pool1d_compositeimplicitautograd_dispatch.h | h | 854 B |
torch/include/ATen/ops/adaptive_max_pool1d_native.h | h | 564 B |
torch/include/ATen/ops/adaptive_max_pool1d_ops.h | h | 1.15 KiB |
torch/include/ATen/ops/adaptive_max_pool2d.h | h | 1.54 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_backward.h | h | 1.65 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 895 B |
torch/include/ATen/ops/adaptive_max_pool2d_backward_cpu_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_backward_cuda_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_backward_meta.h | h | 684 B |
torch/include/ATen/ops/adaptive_max_pool2d_backward_meta_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_backward_native.h | h | 1016 B |
torch/include/ATen/ops/adaptive_max_pool2d_backward_ops.h | h | 2.08 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_compositeexplicitautogradnonfunctional_dispatch.h | h | 880 B |
torch/include/ATen/ops/adaptive_max_pool2d_cpu_dispatch.h | h | 1.12 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_cuda_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_meta.h | h | 644 B |
torch/include/ATen/ops/adaptive_max_pool2d_meta_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/adaptive_max_pool2d_native.h | h | 951 B |
torch/include/ATen/ops/adaptive_max_pool2d_ops.h | h | 2.04 KiB |
torch/include/ATen/ops/adaptive_max_pool3d.h | h | 1.54 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_backward.h | h | 1.65 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 895 B |
torch/include/ATen/ops/adaptive_max_pool3d_backward_cpu_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_backward_cuda_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_backward_meta.h | h | 684 B |
torch/include/ATen/ops/adaptive_max_pool3d_backward_meta_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_backward_native.h | h | 1016 B |
torch/include/ATen/ops/adaptive_max_pool3d_backward_ops.h | h | 2.08 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_compositeexplicitautogradnonfunctional_dispatch.h | h | 880 B |
torch/include/ATen/ops/adaptive_max_pool3d_cpu_dispatch.h | h | 1.12 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_cuda_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_meta.h | h | 644 B |
torch/include/ATen/ops/adaptive_max_pool3d_meta_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/adaptive_max_pool3d_native.h | h | 951 B |
torch/include/ATen/ops/adaptive_max_pool3d_ops.h | h | 2.04 KiB |
torch/include/ATen/ops/add.h | h | 2.12 KiB |
torch/include/ATen/ops/add_compositeexplicitautograd_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/add_compositeexplicitautogradnonfunctional_dispatch.h | h | 967 B |
torch/include/ATen/ops/add_cpu_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/add_cuda_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/add_meta.h | h | 658 B |
torch/include/ATen/ops/add_meta_dispatch.h | h | 1.13 KiB |
torch/include/ATen/ops/add_native.h | h | 2.91 KiB |
torch/include/ATen/ops/add_ops.h | h | 4.68 KiB |
torch/include/ATen/ops/addbmm.h | h | 1.60 KiB |
torch/include/ATen/ops/addbmm_cpu_dispatch.h | h | 1.35 KiB |
torch/include/ATen/ops/addbmm_cuda_dispatch.h | h | 1.36 KiB |
torch/include/ATen/ops/addbmm_meta_dispatch.h | h | 851 B |
torch/include/ATen/ops/addbmm_native.h | h | 951 B |
torch/include/ATen/ops/addbmm_ops.h | h | 3.06 KiB |
torch/include/ATen/ops/addcdiv.h | h | 1.49 KiB |
torch/include/ATen/ops/addcdiv_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/addcdiv_cpu_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/addcdiv_cuda_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/addcdiv_meta.h | h | 685 B |
torch/include/ATen/ops/addcdiv_meta_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/addcdiv_native.h | h | 704 B |
torch/include/ATen/ops/addcdiv_ops.h | h | 2.84 KiB |
torch/include/ATen/ops/addcmul.h | h | 1.49 KiB |
torch/include/ATen/ops/addcmul_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/addcmul_cpu_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/addcmul_cuda_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/addcmul_meta.h | h | 685 B |
torch/include/ATen/ops/addcmul_meta_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/addcmul_native.h | h | 704 B |
torch/include/ATen/ops/addcmul_ops.h | h | 2.84 KiB |
torch/include/ATen/ops/addmm.h | h | 1.56 KiB |
torch/include/ATen/ops/addmm_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/addmm_cpu_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/addmm_cuda_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/addmm_meta.h | h | 702 B |
torch/include/ATen/ops/addmm_meta_dispatch.h | h | 1.34 KiB |
torch/include/ATen/ops/addmm_native.h | h | 2.59 KiB |
torch/include/ATen/ops/addmm_ops.h | h | 3.01 KiB |
torch/include/ATen/ops/addmv.h | h | 1.86 KiB |
torch/include/ATen/ops/addmv_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/addmv_cpu_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/addmv_cuda_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/addmv_meta.h | h | 700 B |
torch/include/ATen/ops/addmv_meta_dispatch.h | h | 1.33 KiB |
torch/include/ATen/ops/addmv_native.h | h | 1.33 KiB |
torch/include/ATen/ops/addmv_ops.h | h | 3.00 KiB |
torch/include/ATen/ops/addr.h | h | 1.55 KiB |
torch/include/ATen/ops/addr_compositeexplicitautograd_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/addr_cpu_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/addr_cuda_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/addr_native.h | h | 1.25 KiB |
torch/include/ATen/ops/addr_ops.h | h | 3.01 KiB |
torch/include/ATen/ops/adjoint.h | h | 654 B |
torch/include/ATen/ops/adjoint_compositeimplicitautograd_dispatch.h | h | 788 B |
torch/include/ATen/ops/adjoint_native.h | h | 498 B |
torch/include/ATen/ops/adjoint_ops.h | h | 971 B |
torch/include/ATen/ops/affine_grid_generator.h | h | 4.49 KiB |
torch/include/ATen/ops/affine_grid_generator_backward.h | h | 1.80 KiB |
torch/include/ATen/ops/affine_grid_generator_backward_compositeimplicitautograd_dispatch.h | h | 985 B |
torch/include/ATen/ops/affine_grid_generator_backward_native.h | h | 563 B |
torch/include/ATen/ops/affine_grid_generator_backward_ops.h | h | 1.16 KiB |
torch/include/ATen/ops/affine_grid_generator_compositeexplicitautograd_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/affine_grid_generator_native.h | h | 703 B |
torch/include/ATen/ops/affine_grid_generator_ops.h | h | 1.92 KiB |
torch/include/ATen/ops/alias.h | h | 646 B |
torch/include/ATen/ops/alias_compositeexplicitautograd_dispatch.h | h | 786 B |
torch/include/ATen/ops/alias_copy.h | h | 1.06 KiB |
torch/include/ATen/ops/alias_copy_compositeexplicitautograd_dispatch.h | h | 899 B |
torch/include/ATen/ops/alias_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 817 B |
torch/include/ATen/ops/alias_copy_native.h | h | 584 B |
torch/include/ATen/ops/alias_copy_ops.h | h | 1.55 KiB |
torch/include/ATen/ops/alias_native.h | h | 557 B |
torch/include/ATen/ops/alias_ops.h | h | 965 B |
torch/include/ATen/ops/align_as.h | h | 508 B |
torch/include/ATen/ops/align_as_compositeimplicitautograd_dispatch.h | h | 815 B |
torch/include/ATen/ops/align_as_native.h | h | 525 B |
torch/include/ATen/ops/align_as_ops.h | h | 1.03 KiB |
torch/include/ATen/ops/align_tensors.h | h | 696 B |
torch/include/ATen/ops/align_tensors_compositeimplicitautograd_dispatch.h | h | 808 B |
torch/include/ATen/ops/align_tensors_native.h | h | 518 B |
torch/include/ATen/ops/align_tensors_ops.h | h | 1.00 KiB |
torch/include/ATen/ops/align_to.h | h | 508 B |
torch/include/ATen/ops/align_to_compositeimplicitautograd_dispatch.h | h | 914 B |
torch/include/ATen/ops/align_to_native.h | h | 624 B |
torch/include/ATen/ops/align_to_ops.h | h | 1.72 KiB |
torch/include/ATen/ops/all.h | h | 3.33 KiB |
torch/include/ATen/ops/all_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/all_compositeexplicitautogradnonfunctional_dispatch.h | h | 996 B |
torch/include/ATen/ops/all_compositeimplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/all_cpu_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/all_cuda_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/all_meta.h | h | 918 B |
torch/include/ATen/ops/all_meta_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/all_native.h | h | 1.49 KiB |
torch/include/ATen/ops/all_ops.h | h | 5.53 KiB |
torch/include/ATen/ops/allclose.h | h | 832 B |
torch/include/ATen/ops/allclose_compositeexplicitautograd_dispatch.h | h | 869 B |
torch/include/ATen/ops/allclose_native.h | h | 579 B |
torch/include/ATen/ops/allclose_ops.h | h | 1.17 KiB |
torch/include/ATen/ops/alpha_dropout.h | h | 949 B |
torch/include/ATen/ops/alpha_dropout_compositeimplicitautograd_dispatch.h | h | 898 B |
torch/include/ATen/ops/alpha_dropout_native.h | h | 608 B |
torch/include/ATen/ops/alpha_dropout_ops.h | h | 1.64 KiB |
torch/include/ATen/ops/amax.h | h | 1.26 KiB |
torch/include/ATen/ops/amax_compositeexplicitautogradnonfunctional_dispatch.h | h | 855 B |
torch/include/ATen/ops/amax_cpu_dispatch.h | h | 1019 B |
torch/include/ATen/ops/amax_cuda_dispatch.h | h | 1021 B |
torch/include/ATen/ops/amax_meta.h | h | 635 B |
torch/include/ATen/ops/amax_meta_dispatch.h | h | 1021 B |
torch/include/ATen/ops/amax_native.h | h | 648 B |
torch/include/ATen/ops/amax_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/amin.h | h | 1.26 KiB |
torch/include/ATen/ops/amin_compositeexplicitautogradnonfunctional_dispatch.h | h | 855 B |
torch/include/ATen/ops/amin_cpu_dispatch.h | h | 1019 B |
torch/include/ATen/ops/amin_cuda_dispatch.h | h | 1021 B |
torch/include/ATen/ops/amin_meta.h | h | 635 B |
torch/include/ATen/ops/amin_meta_dispatch.h | h | 1021 B |
torch/include/ATen/ops/amin_native.h | h | 648 B |
torch/include/ATen/ops/amin_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/aminmax.h | h | 1.56 KiB |
torch/include/ATen/ops/aminmax_compositeexplicitautogradnonfunctional_dispatch.h | h | 904 B |
torch/include/ATen/ops/aminmax_cpu_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/aminmax_cuda_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/aminmax_meta.h | h | 647 B |
torch/include/ATen/ops/aminmax_meta_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/aminmax_native.h | h | 690 B |
torch/include/ATen/ops/aminmax_ops.h | h | 2.09 KiB |
torch/include/ATen/ops/and.h | h | 909 B |
torch/include/ATen/ops/and_compositeimplicitautograd_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/and_native.h | h | 764 B |
torch/include/ATen/ops/and_ops.h | h | 2.86 KiB |
torch/include/ATen/ops/angle.h | h | 1.01 KiB |
torch/include/ATen/ops/angle_cpu_dispatch.h | h | 899 B |
torch/include/ATen/ops/angle_cuda_dispatch.h | h | 901 B |
torch/include/ATen/ops/angle_native.h | h | 728 B |
torch/include/ATen/ops/angle_ops.h | h | 1.52 KiB |
torch/include/ATen/ops/any.h | h | 3.33 KiB |
torch/include/ATen/ops/any_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/any_compositeexplicitautogradnonfunctional_dispatch.h | h | 996 B |
torch/include/ATen/ops/any_compositeimplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/any_cpu_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/any_cuda_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/any_meta.h | h | 918 B |
torch/include/ATen/ops/any_meta_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/any_native.h | h | 1.46 KiB |
torch/include/ATen/ops/any_ops.h | h | 5.53 KiB |
torch/include/ATen/ops/arange.h | h | 4.12 KiB |
torch/include/ATen/ops/arange_compositeexplicitautograd_dispatch.h | h | 1.85 KiB |
torch/include/ATen/ops/arange_cpu_dispatch.h | h | 947 B |
torch/include/ATen/ops/arange_cuda_dispatch.h | h | 949 B |
torch/include/ATen/ops/arange_meta_dispatch.h | h | 949 B |
torch/include/ATen/ops/arange_native.h | h | 1.46 KiB |
torch/include/ATen/ops/arange_ops.h | h | 5.03 KiB |
torch/include/ATen/ops/arccos.h | h | 1.16 KiB |
torch/include/ATen/ops/arccos_compositeimplicitautograd_dispatch.h | h | 998 B |
torch/include/ATen/ops/arccos_native.h | h | 628 B |
torch/include/ATen/ops/arccos_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/arccosh.h | h | 1.17 KiB |
torch/include/ATen/ops/arccosh_compositeimplicitautograd_dispatch.h | h | 1002 B |
torch/include/ATen/ops/arccosh_native.h | h | 631 B |
torch/include/ATen/ops/arccosh_ops.h | h | 2.03 KiB |
torch/include/ATen/ops/arcsin.h | h | 1.16 KiB |
torch/include/ATen/ops/arcsin_compositeimplicitautograd_dispatch.h | h | 998 B |
torch/include/ATen/ops/arcsin_native.h | h | 628 B |
torch/include/ATen/ops/arcsin_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/arcsinh.h | h | 1.17 KiB |
torch/include/ATen/ops/arcsinh_compositeimplicitautograd_dispatch.h | h | 1002 B |
torch/include/ATen/ops/arcsinh_native.h | h | 631 B |
torch/include/ATen/ops/arcsinh_ops.h | h | 2.03 KiB |
torch/include/ATen/ops/arctan.h | h | 1.16 KiB |
torch/include/ATen/ops/arctan2.h | h | 1.17 KiB |
torch/include/ATen/ops/arctan2_compositeimplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/arctan2_native.h | h | 709 B |
torch/include/ATen/ops/arctan2_ops.h | h | 2.29 KiB |
torch/include/ATen/ops/arctan_compositeimplicitautograd_dispatch.h | h | 998 B |
torch/include/ATen/ops/arctan_native.h | h | 628 B |
torch/include/ATen/ops/arctan_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/arctanh.h | h | 1.17 KiB |
torch/include/ATen/ops/arctanh_compositeimplicitautograd_dispatch.h | h | 1002 B |
torch/include/ATen/ops/arctanh_native.h | h | 631 B |
torch/include/ATen/ops/arctanh_ops.h | h | 2.03 KiB |
torch/include/ATen/ops/argmax.h | h | 1.33 KiB |
torch/include/ATen/ops/argmax_compositeexplicitautogradnonfunctional_dispatch.h | h | 878 B |
torch/include/ATen/ops/argmax_cpu_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/argmax_cuda_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/argmax_meta.h | h | 646 B |
torch/include/ATen/ops/argmax_meta_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/argmax_native.h | h | 663 B |
torch/include/ATen/ops/argmax_ops.h | h | 1.83 KiB |
torch/include/ATen/ops/argmin.h | h | 1.33 KiB |
torch/include/ATen/ops/argmin_compositeexplicitautogradnonfunctional_dispatch.h | h | 878 B |
torch/include/ATen/ops/argmin_cpu_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/argmin_cuda_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/argmin_meta.h | h | 646 B |
torch/include/ATen/ops/argmin_meta_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/argmin_native.h | h | 663 B |
torch/include/ATen/ops/argmin_ops.h | h | 1.83 KiB |
torch/include/ATen/ops/argsort.h | h | 1.90 KiB |
torch/include/ATen/ops/argsort_compositeimplicitautograd_dispatch.h | h | 1.26 KiB |
torch/include/ATen/ops/argsort_native.h | h | 864 B |
torch/include/ATen/ops/argsort_ops.h | h | 3.13 KiB |
torch/include/ATen/ops/argwhere.h | h | 652 B |
torch/include/ATen/ops/argwhere_compositeimplicitautograd_dispatch.h | h | 789 B |
torch/include/ATen/ops/argwhere_native.h | h | 499 B |
torch/include/ATen/ops/argwhere_ops.h | h | 968 B |
torch/include/ATen/ops/as_strided.h | h | 3.83 KiB |
torch/include/ATen/ops/as_strided_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/as_strided_copy.h | h | 5.86 KiB |
torch/include/ATen/ops/as_strided_copy_compositeexplicitautograd_dispatch.h | h | 1.47 KiB |
torch/include/ATen/ops/as_strided_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/as_strided_copy_native.h | h | 821 B |
torch/include/ATen/ops/as_strided_copy_ops.h | h | 2.23 KiB |
torch/include/ATen/ops/as_strided_cpu_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/as_strided_cuda_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/as_strided_meta_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/as_strided_native.h | h | 1.14 KiB |
torch/include/ATen/ops/as_strided_ops.h | h | 2.15 KiB |
torch/include/ATen/ops/as_strided_scatter.h | h | 6.36 KiB |
torch/include/ATen/ops/as_strided_scatter_compositeexplicitautograd_dispatch.h | h | 1.57 KiB |
torch/include/ATen/ops/as_strided_scatter_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/as_strided_scatter_native.h | h | 875 B |
torch/include/ATen/ops/as_strided_scatter_ops.h | h | 2.40 KiB |
torch/include/ATen/ops/asin.h | h | 1.14 KiB |
torch/include/ATen/ops/asin_compositeexplicitautogradnonfunctional_dispatch.h | h | 861 B |
torch/include/ATen/ops/asin_cpu_dispatch.h | h | 946 B |
torch/include/ATen/ops/asin_cuda_dispatch.h | h | 948 B |
torch/include/ATen/ops/asin_meta.h | h | 600 B |
torch/include/ATen/ops/asin_meta_dispatch.h | h | 948 B |
torch/include/ATen/ops/asin_native.h | h | 1.00 KiB |
torch/include/ATen/ops/asin_ops.h | h | 2.01 KiB |
torch/include/ATen/ops/asinh.h | h | 1.15 KiB |
torch/include/ATen/ops/asinh_compositeexplicitautogradnonfunctional_dispatch.h | h | 863 B |
torch/include/ATen/ops/asinh_cpu_dispatch.h | h | 950 B |
torch/include/ATen/ops/asinh_cuda_dispatch.h | h | 952 B |
torch/include/ATen/ops/asinh_meta.h | h | 601 B |
torch/include/ATen/ops/asinh_meta_dispatch.h | h | 952 B |
torch/include/ATen/ops/asinh_native.h | h | 1.01 KiB |
torch/include/ATen/ops/asinh_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/atan.h | h | 1.14 KiB |
torch/include/ATen/ops/atan2.h | h | 1.15 KiB |
torch/include/ATen/ops/atan2_compositeexplicitautogradnonfunctional_dispatch.h | h | 915 B |
torch/include/ATen/ops/atan2_cpu_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/atan2_cuda_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/atan2_meta.h | h | 627 B |
torch/include/ATen/ops/atan2_meta_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/atan2_native.h | h | 642 B |
torch/include/ATen/ops/atan2_ops.h | h | 2.27 KiB |
torch/include/ATen/ops/atan_compositeexplicitautogradnonfunctional_dispatch.h | h | 861 B |
torch/include/ATen/ops/atan_cpu_dispatch.h | h | 946 B |
torch/include/ATen/ops/atan_cuda_dispatch.h | h | 948 B |
torch/include/ATen/ops/atan_meta.h | h | 600 B |
torch/include/ATen/ops/atan_meta_dispatch.h | h | 948 B |
torch/include/ATen/ops/atan_native.h | h | 1.00 KiB |
torch/include/ATen/ops/atan_ops.h | h | 2.01 KiB |
torch/include/ATen/ops/atanh.h | h | 1.15 KiB |
torch/include/ATen/ops/atanh_compositeexplicitautogradnonfunctional_dispatch.h | h | 863 B |
torch/include/ATen/ops/atanh_cpu_dispatch.h | h | 950 B |
torch/include/ATen/ops/atanh_cuda_dispatch.h | h | 952 B |
torch/include/ATen/ops/atanh_meta.h | h | 601 B |
torch/include/ATen/ops/atanh_meta_dispatch.h | h | 952 B |
torch/include/ATen/ops/atanh_native.h | h | 1.01 KiB |
torch/include/ATen/ops/atanh_ops.h | h | 2.02 KiB |
torch/include/ATen/ops/atleast_1d.h | h | 854 B |
torch/include/ATen/ops/atleast_1d_compositeimplicitautograd_dispatch.h | h | 864 B |
torch/include/ATen/ops/atleast_1d_native.h | h | 574 B |
torch/include/ATen/ops/atleast_1d_ops.h | h | 1.53 KiB |
torch/include/ATen/ops/atleast_2d.h | h | 854 B |
torch/include/ATen/ops/atleast_2d_compositeimplicitautograd_dispatch.h | h | 864 B |
torch/include/ATen/ops/atleast_2d_native.h | h | 574 B |
torch/include/ATen/ops/atleast_2d_ops.h | h | 1.53 KiB |
torch/include/ATen/ops/atleast_3d.h | h | 854 B |
torch/include/ATen/ops/atleast_3d_compositeimplicitautograd_dispatch.h | h | 864 B |
torch/include/ATen/ops/atleast_3d_native.h | h | 574 B |
torch/include/ATen/ops/atleast_3d_ops.h | h | 1.53 KiB |
torch/include/ATen/ops/avg_pool1d.h | h | 961 B |
torch/include/ATen/ops/avg_pool1d_compositeimplicitautograd_dispatch.h | h | 925 B |
torch/include/ATen/ops/avg_pool1d_native.h | h | 635 B |
torch/include/ATen/ops/avg_pool1d_ops.h | h | 1.35 KiB |
torch/include/ATen/ops/avg_pool2d.h | h | 2.21 KiB |
torch/include/ATen/ops/avg_pool2d_backward.h | h | 2.44 KiB |
torch/include/ATen/ops/avg_pool2d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1019 B |
torch/include/ATen/ops/avg_pool2d_backward_cpu_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/avg_pool2d_backward_cuda_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/avg_pool2d_backward_meta.h | h | 808 B |
torch/include/ATen/ops/avg_pool2d_backward_meta_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/avg_pool2d_backward_native.h | h | 1.76 KiB |
torch/include/ATen/ops/avg_pool2d_backward_ops.h | h | 2.87 KiB |
torch/include/ATen/ops/avg_pool2d_compositeexplicitautogradnonfunctional_dispatch.h | h | 1009 B |
torch/include/ATen/ops/avg_pool2d_cpu_dispatch.h | h | 1.42 KiB |
torch/include/ATen/ops/avg_pool2d_cuda_dispatch.h | h | 1.43 KiB |
torch/include/ATen/ops/avg_pool2d_meta.h | h | 3.59 KiB |
torch/include/ATen/ops/avg_pool2d_meta_dispatch.h | h | 1.43 KiB |
torch/include/ATen/ops/avg_pool2d_native.h | h | 1.84 KiB |
torch/include/ATen/ops/avg_pool2d_ops.h | h | 2.62 KiB |
torch/include/ATen/ops/avg_pool3d.h | h | 2.21 KiB |
torch/include/ATen/ops/avg_pool3d_backward.h | h | 2.44 KiB |
torch/include/ATen/ops/avg_pool3d_backward_compositeexplicitautogradnonfunctional_dispatch.h | h | 1019 B |
torch/include/ATen/ops/avg_pool3d_backward_cpu_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/avg_pool3d_backward_cuda_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/avg_pool3d_backward_meta.h | h | 808 B |
torch/include/ATen/ops/avg_pool3d_backward_meta_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/avg_pool3d_backward_native.h | h | 1.76 KiB |
torch/include/ATen/ops/avg_pool3d_backward_ops.h | h | 2.87 KiB |
torch/include/ATen/ops/avg_pool3d_compositeexplicitautogradnonfunctional_dispatch.h | h | 1009 B |
torch/include/ATen/ops/avg_pool3d_cpu_dispatch.h | h | 1.42 KiB |
torch/include/ATen/ops/avg_pool3d_cuda_dispatch.h | h | 1.43 KiB |
torch/include/ATen/ops/avg_pool3d_meta.h | h | 767 B |
torch/include/ATen/ops/avg_pool3d_meta_dispatch.h | h | 1.43 KiB |
torch/include/ATen/ops/avg_pool3d_native.h | h | 1.84 KiB |
torch/include/ATen/ops/avg_pool3d_ops.h | h | 2.62 KiB |
torch/include/ATen/ops/baddbmm.h | h | 1.61 KiB |
torch/include/ATen/ops/baddbmm_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.06 KiB |
torch/include/ATen/ops/baddbmm_cpu_dispatch.h | h | 1.36 KiB |
torch/include/ATen/ops/baddbmm_cuda_dispatch.h | h | 1.36 KiB |
torch/include/ATen/ops/baddbmm_meta.h | h | 708 B |
torch/include/ATen/ops/baddbmm_meta_dispatch.h | h | 1.36 KiB |
torch/include/ATen/ops/baddbmm_native.h | h | 1.16 KiB |
torch/include/ATen/ops/baddbmm_ops.h | h | 3.07 KiB |
torch/include/ATen/ops/bartlett_window.h | h | 3.37 KiB |
torch/include/ATen/ops/bartlett_window_compositeexplicitautograd_dispatch.h | h | 1.70 KiB |
torch/include/ATen/ops/bartlett_window_native.h | h | 1.07 KiB |
torch/include/ATen/ops/bartlett_window_ops.h | h | 3.80 KiB |
torch/include/ATen/ops/batch_norm.h | h | 1.10 KiB |
torch/include/ATen/ops/batch_norm_backward.h | h | 1.33 KiB |
torch/include/ATen/ops/batch_norm_backward_cpu_dispatch.h | h | 1.10 KiB |
torch/include/ATen/ops/batch_norm_backward_cuda_dispatch.h | h | 1.10 KiB |
torch/include/ATen/ops/batch_norm_backward_elemt.h | h | 2.32 KiB |
torch/include/ATen/ops/batch_norm_backward_elemt_compositeexplicitautograd_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/batch_norm_backward_elemt_cuda_dispatch.h | h | 974 B |
torch/include/ATen/ops/batch_norm_backward_elemt_native.h | h | 1.01 KiB |
torch/include/ATen/ops/batch_norm_backward_elemt_ops.h | h | 2.98 KiB |
torch/include/ATen/ops/batch_norm_backward_native.h | h | 1.75 KiB |
torch/include/ATen/ops/batch_norm_backward_ops.h | h | 2.17 KiB |
torch/include/ATen/ops/batch_norm_backward_reduce.h | h | 2.69 KiB |
torch/include/ATen/ops/batch_norm_backward_reduce_compositeexplicitautograd_dispatch.h | h | 1.45 KiB |
torch/include/ATen/ops/batch_norm_backward_reduce_cuda_dispatch.h | h | 980 B |
torch/include/ATen/ops/batch_norm_backward_reduce_native.h | h | 1.09 KiB |
torch/include/ATen/ops/batch_norm_backward_reduce_ops.h | h | 3.29 KiB |
torch/include/ATen/ops/batch_norm_compositeimplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/batch_norm_elemt.h | h | 1.86 KiB |
torch/include/ATen/ops/batch_norm_elemt_cuda_dispatch.h | h | 1.35 KiB |
torch/include/ATen/ops/batch_norm_elemt_native.h | h | 908 B |
torch/include/ATen/ops/batch_norm_elemt_ops.h | h | 2.55 KiB |
torch/include/ATen/ops/batch_norm_gather_stats.h | h | 2.47 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_compositeexplicitautograd_dispatch.h | h | 1.38 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_cuda_dispatch.h | h | 983 B |
torch/include/ATen/ops/batch_norm_gather_stats_native.h | h | 1.05 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_ops.h | h | 3.12 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_with_counts.h | h | 2.64 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_with_counts_compositeexplicitautograd_dispatch.h | h | 1.42 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_with_counts_cuda_dispatch.h | h | 1007 B |
torch/include/ATen/ops/batch_norm_gather_stats_with_counts_native.h | h | 1.10 KiB |
torch/include/ATen/ops/batch_norm_gather_stats_with_counts_ops.h | h | 3.27 KiB |
torch/include/ATen/ops/batch_norm_native.h | h | 751 B |
torch/include/ATen/ops/batch_norm_ops.h | h | 1.74 KiB |
torch/include/ATen/ops/batch_norm_stats.h | h | 1.41 KiB |
torch/include/ATen/ops/batch_norm_stats_compositeexplicitautograd_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/batch_norm_stats_cuda_dispatch.h | h | 793 B |
torch/include/ATen/ops/batch_norm_stats_native.h | h | 699 B |
torch/include/ATen/ops/batch_norm_stats_ops.h | h | 1.92 KiB |
torch/include/ATen/ops/batch_norm_update_stats.h | h | 2.02 KiB |
torch/include/ATen/ops/batch_norm_update_stats_compositeexplicitautograd_dispatch.h | h | 1.22 KiB |
torch/include/ATen/ops/batch_norm_update_stats_cpu_dispatch.h | h | 902 B |
torch/include/ATen/ops/batch_norm_update_stats_cuda_dispatch.h | h | 904 B |
torch/include/ATen/ops/batch_norm_update_stats_native.h | h | 1.11 KiB |
torch/include/ATen/ops/batch_norm_update_stats_ops.h | h | 2.61 KiB |
torch/include/ATen/ops/bernoulli.h | h | 3.20 KiB |
torch/include/ATen/ops/bernoulli_compositeexplicitautograd_dispatch.h | h | 1.54 KiB |
torch/include/ATen/ops/bernoulli_compositeexplicitautogradnonfunctional_dispatch.h | h | 883 B |
torch/include/ATen/ops/bernoulli_cpu_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/bernoulli_cuda_dispatch.h | h | 1.19 KiB |
torch/include/ATen/ops/bernoulli_meta_dispatch.h | h | 950 B |
torch/include/ATen/ops/bernoulli_native.h | h | 1.46 KiB |
torch/include/ATen/ops/bernoulli_ops.h | h | 6.33 KiB |
torch/include/ATen/ops/bilinear.h | h | 828 B |
torch/include/ATen/ops/bilinear_compositeimplicitautograd_dispatch.h | h | 890 B |
torch/include/ATen/ops/bilinear_native.h | h | 600 B |
torch/include/ATen/ops/bilinear_ops.h | h | 1.26 KiB |
torch/include/ATen/ops/binary_cross_entropy.h | h | 1.71 KiB |
torch/include/ATen/ops/binary_cross_entropy_backward.h | h | 2.05 KiB |
torch/include/ATen/ops/binary_cross_entropy_backward_cpu_dispatch.h | h | 1.36 KiB |
torch/include/ATen/ops/binary_cross_entropy_backward_cuda_dispatch.h | h | 1.37 KiB |
torch/include/ATen/ops/binary_cross_entropy_backward_native.h | h | 1.34 KiB |
torch/include/ATen/ops/binary_cross_entropy_backward_ops.h | h | 2.50 KiB |
torch/include/ATen/ops/binary_cross_entropy_cpu_dispatch.h | h | 1.23 KiB |
torch/include/ATen/ops/binary_cross_entropy_cuda_dispatch.h | h | 1.23 KiB |
torch/include/ATen/ops/binary_cross_entropy_native.h | h | 1.16 KiB |
torch/include/ATen/ops/binary_cross_entropy_ops.h | h | 2.20 KiB |
torch/include/ATen/ops/binary_cross_entropy_with_logits.h | h | 2.08 KiB |
torch/include/ATen/ops/binary_cross_entropy_with_logits_compositeexplicitautograd_dispatch.h | h | 1.46 KiB |
torch/include/ATen/ops/binary_cross_entropy_with_logits_native.h | h | 930 B |
torch/include/ATen/ops/binary_cross_entropy_with_logits_ops.h | h | 2.58 KiB |
torch/include/ATen/ops/bincount.h | h | 1.41 KiB |
torch/include/ATen/ops/bincount_compositeexplicitautograd_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/bincount_cpu_dispatch.h | h | 814 B |
torch/include/ATen/ops/bincount_cuda_dispatch.h | h | 816 B |
torch/include/ATen/ops/bincount_native.h | h | 850 B |
torch/include/ATen/ops/bincount_ops.h | h | 1.95 KiB |
torch/include/ATen/ops/binomial.h | h | 1.44 KiB |
torch/include/ATen/ops/binomial_compositeexplicitautograd_dispatch.h | h | 1.02 KiB |
torch/include/ATen/ops/binomial_cpu_dispatch.h | h | 828 B |
torch/include/ATen/ops/binomial_cuda_dispatch.h | h | 830 B |
torch/include/ATen/ops/binomial_native.h | h | 886 B |
torch/include/ATen/ops/binomial_ops.h | h | 1.99 KiB |
torch/include/ATen/ops/bitwise_and.h | h | 2.79 KiB |
torch/include/ATen/ops/bitwise_and_compositeexplicitautograd_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/bitwise_and_compositeexplicitautogradnonfunctional_dispatch.h | h | 927 B |
torch/include/ATen/ops/bitwise_and_cpu_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_and_cuda_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_and_meta.h | h | 640 B |
torch/include/ATen/ops/bitwise_and_meta_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_and_native.h | h | 1.13 KiB |
torch/include/ATen/ops/bitwise_and_ops.h | h | 5.68 KiB |
torch/include/ATen/ops/bitwise_left_shift.h | h | 3.02 KiB |
torch/include/ATen/ops/bitwise_left_shift_compositeexplicitautograd_dispatch.h | h | 1.44 KiB |
torch/include/ATen/ops/bitwise_left_shift_compositeexplicitautogradnonfunctional_dispatch.h | h | 941 B |
torch/include/ATen/ops/bitwise_left_shift_cpu_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/bitwise_left_shift_cuda_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/bitwise_left_shift_meta.h | h | 647 B |
torch/include/ATen/ops/bitwise_left_shift_meta_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/bitwise_left_shift_native.h | h | 1.18 KiB |
torch/include/ATen/ops/bitwise_left_shift_ops.h | h | 5.91 KiB |
torch/include/ATen/ops/bitwise_not.h | h | 1.07 KiB |
torch/include/ATen/ops/bitwise_not_compositeexplicitautogradnonfunctional_dispatch.h | h | 875 B |
torch/include/ATen/ops/bitwise_not_cpu_dispatch.h | h | 974 B |
torch/include/ATen/ops/bitwise_not_cuda_dispatch.h | h | 976 B |
torch/include/ATen/ops/bitwise_not_meta.h | h | 607 B |
torch/include/ATen/ops/bitwise_not_meta_dispatch.h | h | 976 B |
torch/include/ATen/ops/bitwise_not_native.h | h | 634 B |
torch/include/ATen/ops/bitwise_not_ops.h | h | 2.07 KiB |
torch/include/ATen/ops/bitwise_or.h | h | 2.76 KiB |
torch/include/ATen/ops/bitwise_or_compositeexplicitautograd_dispatch.h | h | 1.39 KiB |
torch/include/ATen/ops/bitwise_or_compositeexplicitautogradnonfunctional_dispatch.h | h | 925 B |
torch/include/ATen/ops/bitwise_or_cpu_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_or_cuda_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_or_meta.h | h | 639 B |
torch/include/ATen/ops/bitwise_or_meta_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_or_native.h | h | 1.12 KiB |
torch/include/ATen/ops/bitwise_or_ops.h | h | 5.66 KiB |
torch/include/ATen/ops/bitwise_right_shift.h | h | 3.04 KiB |
torch/include/ATen/ops/bitwise_right_shift_compositeexplicitautograd_dispatch.h | h | 1.45 KiB |
torch/include/ATen/ops/bitwise_right_shift_compositeexplicitautogradnonfunctional_dispatch.h | h | 943 B |
torch/include/ATen/ops/bitwise_right_shift_cpu_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/bitwise_right_shift_cuda_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/bitwise_right_shift_meta.h | h | 648 B |
torch/include/ATen/ops/bitwise_right_shift_meta_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/bitwise_right_shift_native.h | h | 1.19 KiB |
torch/include/ATen/ops/bitwise_right_shift_ops.h | h | 5.93 KiB |
torch/include/ATen/ops/bitwise_xor.h | h | 2.79 KiB |
torch/include/ATen/ops/bitwise_xor_compositeexplicitautograd_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/bitwise_xor_compositeexplicitautogradnonfunctional_dispatch.h | h | 927 B |
torch/include/ATen/ops/bitwise_xor_cpu_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_xor_cuda_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_xor_meta.h | h | 640 B |
torch/include/ATen/ops/bitwise_xor_meta_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/bitwise_xor_native.h | h | 1.13 KiB |
torch/include/ATen/ops/bitwise_xor_ops.h | h | 5.68 KiB |
torch/include/ATen/ops/blackman_window.h | h | 3.37 KiB |
torch/include/ATen/ops/blackman_window_compositeexplicitautograd_dispatch.h | h | 1.70 KiB |
torch/include/ATen/ops/blackman_window_native.h | h | 1.07 KiB |
torch/include/ATen/ops/blackman_window_ops.h | h | 3.80 KiB |
torch/include/ATen/ops/block_diag.h | h | 1.08 KiB |
torch/include/ATen/ops/block_diag_compositeexplicitautograd_dispatch.h | h | 955 B |
torch/include/ATen/ops/block_diag_native.h | h | 582 B |
torch/include/ATen/ops/block_diag_ops.h | h | 1.55 KiB |
torch/include/ATen/ops/bmm.h | h | 1.12 KiB |
torch/include/ATen/ops/bmm_compositeexplicitautogradnonfunctional_dispatch.h | h | 835 B |
torch/include/ATen/ops/bmm_cpu_dispatch.h | h | 968 B |
torch/include/ATen/ops/bmm_cuda_dispatch.h | h | 970 B |
torch/include/ATen/ops/bmm_meta.h | h | 624 B |
torch/include/ATen/ops/bmm_meta_dispatch.h | h | 970 B |
torch/include/ATen/ops/bmm_native.h | h | 1.46 KiB |
torch/include/ATen/ops/bmm_ops.h | h | 1.67 KiB |
torch/include/ATen/ops/broadcast_tensors.h | h | 712 B |
torch/include/ATen/ops/broadcast_tensors_compositeimplicitautograd_dispatch.h | h | 812 B |
torch/include/ATen/ops/broadcast_tensors_native.h | h | 522 B |
torch/include/ATen/ops/broadcast_tensors_ops.h | h | 1.02 KiB |
torch/include/ATen/ops/broadcast_to.h | h | 1.44 KiB |
torch/include/ATen/ops/broadcast_to_compositeimplicitautograd_dispatch.h | h | 909 B |
torch/include/ATen/ops/broadcast_to_native.h | h | 536 B |
torch/include/ATen/ops/broadcast_to_ops.h | h | 1.05 KiB |
torch/include/ATen/ops/bucketize.h | h | 2.60 KiB |
torch/include/ATen/ops/bucketize_compositeexplicitautograd_dispatch.h | h | 1.00 KiB |
torch/include/ATen/ops/bucketize_cpu_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/bucketize_cuda_dispatch.h | h | 1.21 KiB |
torch/include/ATen/ops/bucketize_native.h | h | 1.38 KiB |
torch/include/ATen/ops/bucketize_ops.h | h | 3.55 KiB |
torch/include/ATen/ops/can_cast.h | h | 685 B |
torch/include/ATen/ops/can_cast_compositeimplicitautograd_dispatch.h | h | 799 B |
torch/include/ATen/ops/can_cast_native.h | h | 509 B |
torch/include/ATen/ops/can_cast_ops.h | h | 1012 B |
torch/include/ATen/ops/cartesian_prod.h | h | 683 B |
torch/include/ATen/ops/cartesian_prod_compositeimplicitautograd_dispatch.h | h | 794 B |
torch/include/ATen/ops/cartesian_prod_native.h | h | 504 B |
torch/include/ATen/ops/cartesian_prod_ops.h | h | 985 B |
torch/include/ATen/ops/cat.h | h | 1.79 KiB |
torch/include/ATen/ops/cat_compositeexplicitautogradnonfunctional_dispatch.h | h | 836 B |
torch/include/ATen/ops/cat_compositeimplicitautograd_dispatch.h | h | 985 B |
torch/include/ATen/ops/cat_cpu_dispatch.h | h | 969 B |
torch/include/ATen/ops/cat_cuda_dispatch.h | h | 971 B |
torch/include/ATen/ops/cat_meta.h | h | 4.92 KiB |
torch/include/ATen/ops/cat_meta_dispatch.h | h | 971 B |
torch/include/ATen/ops/cat_native.h | h | 1.55 KiB |
torch/include/ATen/ops/cat_ops.h | h | 2.88 KiB |
torch/include/ATen/ops/cauchy.h | h | 1.51 KiB |
torch/include/ATen/ops/cauchy_compositeexplicitautograd_dispatch.h | h | 1.17 KiB |
torch/include/ATen/ops/cauchy_cpu_dispatch.h | h | 830 B |
torch/include/ATen/ops/cauchy_cuda_dispatch.h | h | 832 B |
torch/include/ATen/ops/cauchy_meta_dispatch.h | h | 832 B |
torch/include/ATen/ops/cauchy_native.h | h | 879 B |
torch/include/ATen/ops/cauchy_ops.h | h | 2.76 KiB |
torch/include/ATen/ops/ccol_indices.h | h | 512 B |
torch/include/ATen/ops/ccol_indices_compositeexplicitautograd_dispatch.h | h | 793 B |
torch/include/ATen/ops/ccol_indices_copy.h | h | 1.13 KiB |
torch/include/ATen/ops/ccol_indices_copy_compositeexplicitautograd_dispatch.h | h | 913 B |
torch/include/ATen/ops/ccol_indices_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 824 B |
torch/include/ATen/ops/ccol_indices_copy_native.h | h | 598 B |
torch/include/ATen/ops/ccol_indices_copy_ops.h | h | 1.59 KiB |
torch/include/ATen/ops/ccol_indices_native.h | h | 583 B |
torch/include/ATen/ops/ccol_indices_ops.h | h | 986 B |
torch/include/ATen/ops/cdist.h | h | 790 B |
torch/include/ATen/ops/cdist_compositeimplicitautograd_dispatch.h | h | 873 B |
torch/include/ATen/ops/cdist_native.h | h | 583 B |
torch/include/ATen/ops/cdist_ops.h | h | 1.17 KiB |
torch/include/ATen/ops/ceil.h | h | 1.14 KiB |
torch/include/ATen/ops/ceil_compositeexplicitautogradnonfunctional_dispatch.h | h | 861 B |
torch/include/ATen/ops/ceil_cpu_dispatch.h | h | 946 B |
torch/include/ATen/ops/ceil_cuda_dispatch.h | h | 948 B |
torch/include/ATen/ops/ceil_meta.h | h | 600 B |
torch/include/ATen/ops/ceil_meta_dispatch.h | h | 948 B |
torch/include/ATen/ops/ceil_native.h | h | 1.00 KiB |
torch/include/ATen/ops/ceil_ops.h | h | 2.01 KiB |
torch/include/ATen/ops/celu.h | h | 1.35 KiB |
torch/include/ATen/ops/celu_compositeexplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/celu_native.h | h | 708 B |
torch/include/ATen/ops/celu_ops.h | h | 2.27 KiB |
torch/include/ATen/ops/chain_matmul.h | h | 1.11 KiB |
torch/include/ATen/ops/chain_matmul_compositeimplicitautograd_dispatch.h | h | 964 B |
torch/include/ATen/ops/chain_matmul_native.h | h | 588 B |
torch/include/ATen/ops/chain_matmul_ops.h | h | 1.57 KiB |
torch/include/ATen/ops/chalf.h | h | 505 B |
torch/include/ATen/ops/chalf_compositeimplicitautograd_dispatch.h | h | 850 B |
torch/include/ATen/ops/chalf_native.h | h | 560 B |
torch/include/ATen/ops/chalf_ops.h | h | 1.10 KiB |
torch/include/ATen/ops/channel_shuffle.h | h | 3.55 KiB |
torch/include/ATen/ops/channel_shuffle_compositeexplicitautograd_dispatch.h | h | 1.14 KiB |
torch/include/ATen/ops/channel_shuffle_cpu_dispatch.h | h | 859 B |
torch/include/ATen/ops/channel_shuffle_cuda_dispatch.h | h | 861 B |
torch/include/ATen/ops/channel_shuffle_native.h | h | 731 B |
torch/include/ATen/ops/channel_shuffle_ops.h | h | 1.71 KiB |
torch/include/ATen/ops/cholesky.h | h | 1.16 KiB |
torch/include/ATen/ops/cholesky_cpu_dispatch.h | h | 956 B |
torch/include/ATen/ops/cholesky_cuda_dispatch.h | h | 958 B |
torch/include/ATen/ops/cholesky_inverse.h | h | 1.24 KiB |
torch/include/ATen/ops/cholesky_inverse_cpu_dispatch.h | h | 980 B |
torch/include/ATen/ops/cholesky_inverse_cuda_dispatch.h | h | 982 B |
torch/include/ATen/ops/cholesky_inverse_native.h | h | 626 B |
torch/include/ATen/ops/cholesky_inverse_ops.h | h | 1.68 KiB |
torch/include/ATen/ops/cholesky_native.h | h | 610 B |
torch/include/ATen/ops/cholesky_ops.h | h | 1.63 KiB |
torch/include/ATen/ops/cholesky_solve.h | h | 1.36 KiB |
torch/include/ATen/ops/cholesky_solve_compositeexplicitautograd_dispatch.h | h | 1.07 KiB |
torch/include/ATen/ops/cholesky_solve_native.h | h | 676 B |
torch/include/ATen/ops/cholesky_solve_ops.h | h | 1.84 KiB |
torch/include/ATen/ops/choose_qparams_optimized.h | h | 902 B |
torch/include/ATen/ops/choose_qparams_optimized_compositeimplicitautograd_dispatch.h | h | 895 B |
torch/include/ATen/ops/choose_qparams_optimized_native.h | h | 605 B |
torch/include/ATen/ops/choose_qparams_optimized_ops.h | h | 1.29 KiB |
torch/include/ATen/ops/chunk.h | h | 735 B |
torch/include/ATen/ops/chunk_compositeimplicitautograd_dispatch.h | h | 832 B |
torch/include/ATen/ops/chunk_native.h | h | 656 B |
torch/include/ATen/ops/chunk_ops.h | h | 1.09 KiB |
torch/include/ATen/ops/clamp.h | h | 2.92 KiB |
torch/include/ATen/ops/clamp_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.31 KiB |
torch/include/ATen/ops/clamp_cpu_dispatch.h | h | 1.89 KiB |
torch/include/ATen/ops/clamp_cuda_dispatch.h | h | 1.89 KiB |
torch/include/ATen/ops/clamp_max.h | h | 2.28 KiB |
torch/include/ATen/ops/clamp_max_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/clamp_max_cpu_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/clamp_max_cuda_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/clamp_max_meta.h | h | 786 B |
torch/include/ATen/ops/clamp_max_meta_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/clamp_max_native.h | h | 840 B |
torch/include/ATen/ops/clamp_max_ops.h | h | 4.19 KiB |
torch/include/ATen/ops/clamp_meta.h | h | 838 B |
torch/include/ATen/ops/clamp_meta_dispatch.h | h | 1.89 KiB |
torch/include/ATen/ops/clamp_min.h | h | 2.28 KiB |
torch/include/ATen/ops/clamp_min_compositeexplicitautogradnonfunctional_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/clamp_min_cpu_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/clamp_min_cuda_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/clamp_min_meta.h | h | 786 B |
torch/include/ATen/ops/clamp_min_meta_dispatch.h | h | 1.40 KiB |
torch/include/ATen/ops/clamp_min_native.h | h | 840 B |
torch/include/ATen/ops/clamp_min_ops.h | h | 4.19 KiB |
torch/include/ATen/ops/clamp_native.h | h | 1.04 KiB |
torch/include/ATen/ops/clamp_ops.h | h | 5.26 KiB |
torch/include/ATen/ops/clip.h | h | 2.90 KiB |
torch/include/ATen/ops/clip_compositeimplicitautograd_dispatch.h | h | 1.92 KiB |
torch/include/ATen/ops/clip_native.h | h | 1.33 KiB |
torch/include/ATen/ops/clip_ops.h | h | 5.24 KiB |
torch/include/ATen/ops/clone.h | h | 1.33 KiB |
torch/include/ATen/ops/clone_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/clone_native.h | h | 1.29 KiB |
torch/include/ATen/ops/clone_ops.h | h | 1.85 KiB |
torch/include/ATen/ops/coalesce.h | h | 508 B |
torch/include/ATen/ops/coalesce_compositeimplicitautograd_dispatch.h | h | 789 B |
torch/include/ATen/ops/coalesce_native.h | h | 499 B |
torch/include/ATen/ops/coalesce_ops.h | h | 974 B |
torch/include/ATen/ops/col2im.h | h | 5.77 KiB |
torch/include/ATen/ops/col2im_cpu_dispatch.h | h | 1.90 KiB |
torch/include/ATen/ops/col2im_cuda_dispatch.h | h | 1.90 KiB |
torch/include/ATen/ops/col2im_native.h | h | 1.23 KiB |
torch/include/ATen/ops/col2im_ops.h | h | 2.41 KiB |
torch/include/ATen/ops/col_indices.h | h | 511 B |
torch/include/ATen/ops/col_indices_compositeexplicitautograd_dispatch.h | h | 792 B |
torch/include/ATen/ops/col_indices_copy.h | h | 1.12 KiB |
torch/include/ATen/ops/col_indices_copy_compositeexplicitautograd_dispatch.h | h | 911 B |
torch/include/ATen/ops/col_indices_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 823 B |
torch/include/ATen/ops/col_indices_copy_native.h | h | 596 B |
torch/include/ATen/ops/col_indices_copy_ops.h | h | 1.58 KiB |
torch/include/ATen/ops/col_indices_native.h | h | 581 B |
torch/include/ATen/ops/col_indices_ops.h | h | 983 B |
torch/include/ATen/ops/column_stack.h | h | 1.10 KiB |
torch/include/ATen/ops/column_stack_compositeimplicitautograd_dispatch.h | h | 961 B |
torch/include/ATen/ops/column_stack_native.h | h | 586 B |
torch/include/ATen/ops/column_stack_ops.h | h | 1.56 KiB |
torch/include/ATen/ops/combinations.h | h | 769 B |
torch/include/ATen/ops/combinations_compositeimplicitautograd_dispatch.h | h | 835 B |
torch/include/ATen/ops/combinations_native.h | h | 545 B |
torch/include/ATen/ops/combinations_ops.h | h | 1.08 KiB |
torch/include/ATen/ops/complex.h | h | 1.16 KiB |
torch/include/ATen/ops/complex_compositeexplicitautograd_dispatch.h | h | 813 B |
torch/include/ATen/ops/complex_cpu_dispatch.h | h | 899 B |
torch/include/ATen/ops/complex_cuda_dispatch.h | h | 901 B |
torch/include/ATen/ops/complex_native.h | h | 628 B |
torch/include/ATen/ops/complex_ops.h | h | 1.69 KiB |
torch/include/ATen/ops/concat.h | h | 1.81 KiB |
torch/include/ATen/ops/concat_compositeimplicitautograd_dispatch.h | h | 1.22 KiB |
torch/include/ATen/ops/concat_native.h | h | 768 B |
torch/include/ATen/ops/concat_ops.h | h | 2.85 KiB |
torch/include/ATen/ops/concatenate.h | h | 1.91 KiB |
torch/include/ATen/ops/concatenate_compositeimplicitautograd_dispatch.h | h | 1.25 KiB |
torch/include/ATen/ops/concatenate_native.h | h | 788 B |
torch/include/ATen/ops/concatenate_ops.h | h | 2.91 KiB |
torch/include/ATen/ops/conj.h | h | 653 B |
torch/include/ATen/ops/conj_compositeimplicitautograd_dispatch.h | h | 785 B |
torch/include/ATen/ops/conj_native.h | h | 495 B |
torch/include/ATen/ops/conj_ops.h | h | 962 B |
torch/include/ATen/ops/conj_physical.h | h | 1.25 KiB |
torch/include/ATen/ops/conj_physical_compositeexplicitautograd_dispatch.h | h | 791 B |
torch/include/ATen/ops/conj_physical_compositeimplicitautograd_dispatch.h | h | 794 B |
torch/include/ATen/ops/conj_physical_cpu_dispatch.h | h | 861 B |
torch/include/ATen/ops/conj_physical_cuda_dispatch.h | h | 863 B |
torch/include/ATen/ops/conj_physical_native.h | h | 909 B |
torch/include/ATen/ops/conj_physical_ops.h | h | 2.09 KiB |
torch/include/ATen/ops/constant_pad_nd.h | h | 4.21 KiB |
torch/include/ATen/ops/constant_pad_nd_compositeexplicitautograd_dispatch.h | h | 1.50 KiB |
torch/include/ATen/ops/constant_pad_nd_native.h | h | 701 B |
torch/include/ATen/ops/constant_pad_nd_ops.h | h | 1.92 KiB |
torch/include/ATen/ops/contiguous.h | h | 510 B |
torch/include/ATen/ops/contiguous_compositeimplicitautograd_dispatch.h | h | 853 B |
torch/include/ATen/ops/contiguous_native.h | h | 563 B |
torch/include/ATen/ops/contiguous_ops.h | h | 1.08 KiB |
torch/include/ATen/ops/conv1d.h | h | 4.46 KiB |
torch/include/ATen/ops/conv1d_compositeimplicitautograd_dispatch.h | h | 1.68 KiB |
torch/include/ATen/ops/conv1d_native.h | h | 1018 B |
torch/include/ATen/ops/conv1d_ops.h | h | 2.63 KiB |
torch/include/ATen/ops/conv2d.h | h | 4.46 KiB |
torch/include/ATen/ops/conv2d_compositeimplicitautograd_dispatch.h | h | 1.68 KiB |
torch/include/ATen/ops/conv2d_native.h | h | 1018 B |
torch/include/ATen/ops/conv2d_ops.h | h | 2.63 KiB |
torch/include/ATen/ops/conv3d.h | h | 4.46 KiB |
torch/include/ATen/ops/conv3d_compositeimplicitautograd_dispatch.h | h | 1.68 KiB |
torch/include/ATen/ops/conv3d_native.h | h | 1018 B |
torch/include/ATen/ops/conv3d_ops.h | h | 2.63 KiB |
torch/include/ATen/ops/conv_depthwise3d.h | h | 7.19 KiB |
torch/include/ATen/ops/conv_depthwise3d_compositeexplicitautograd_dispatch.h | h | 1.79 KiB |
torch/include/ATen/ops/conv_depthwise3d_cuda_dispatch.h | h | 1.16 KiB |
torch/include/ATen/ops/conv_depthwise3d_native.h | h | 970 B |
torch/include/ATen/ops/conv_depthwise3d_ops.h | h | 2.81 KiB |
torch/include/ATen/ops/conv_tbc.h | h | 1.40 KiB |
torch/include/ATen/ops/conv_tbc_backward.h | h | 910 B |
torch/include/ATen/ops/conv_tbc_backward_compositeimplicitautograd_dispatch.h | h | 925 B |
torch/include/ATen/ops/conv_tbc_backward_native.h | h | 635 B |
torch/include/ATen/ops/conv_tbc_backward_ops.h | h | 1.39 KiB |
torch/include/ATen/ops/conv_tbc_compositeexplicitautograd_dispatch.h | h | 1.12 KiB |
torch/include/ATen/ops/conv_tbc_native.h | h | 712 B |
torch/include/ATen/ops/conv_tbc_ops.h | h | 1.96 KiB |
torch/include/ATen/ops/conv_transpose1d.h | h | 2.96 KiB |
torch/include/ATen/ops/conv_transpose1d_compositeimplicitautograd_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/conv_transpose1d_native.h | h | 792 B |
torch/include/ATen/ops/conv_transpose1d_ops.h | h | 1.67 KiB |
torch/include/ATen/ops/conv_transpose2d.h | h | 3.00 KiB |
torch/include/ATen/ops/conv_transpose2d_compositeimplicitautograd_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/conv_transpose2d_native.h | h | 792 B |
torch/include/ATen/ops/conv_transpose2d_ops.h | h | 1.69 KiB |
torch/include/ATen/ops/conv_transpose3d.h | h | 3.00 KiB |
torch/include/ATen/ops/conv_transpose3d_compositeimplicitautograd_dispatch.h | h | 1.32 KiB |
torch/include/ATen/ops/conv_transpose3d_native.h | h | 792 B |
torch/include/ATen/ops/conv_transpose3d_ops.h | h | 1.69 KiB |
torch/include/ATen/ops/convolution.h | h | 7.96 KiB |
torch/include/ATen/ops/convolution_backward.h | h | 11.28 KiB |
torch/include/ATen/ops/convolution_backward_compositeexplicitautograd_dispatch.h | h | 3.26 KiB |
torch/include/ATen/ops/convolution_backward_cuda_dispatch.h | h | 1.44 KiB |
torch/include/ATen/ops/convolution_backward_native.h | h | 1.26 KiB |
torch/include/ATen/ops/convolution_backward_ops.h | h | 3.90 KiB |
torch/include/ATen/ops/convolution_backward_overrideable.h | h | 10.50 KiB |
torch/include/ATen/ops/convolution_backward_overrideable_compositeexplicitautograd_dispatch.h | h | 3.12 KiB |
torch/include/ATen/ops/convolution_backward_overrideable_native.h | h | 1.21 KiB |
torch/include/ATen/ops/convolution_backward_overrideable_ops.h | h | 3.76 KiB |
torch/include/ATen/ops/convolution_compositeexplicitautograd_dispatch.h | h | 2.47 KiB |
torch/include/ATen/ops/convolution_native.h | h | 1.01 KiB |
torch/include/ATen/ops/convolution_ops.h | h | 3.04 KiB |
torch/include/ATen/ops/convolution_overrideable.h | h | 8.36 KiB |
torch/include/ATen/ops/convolution_overrideable_compositeexplicitautograd_dispatch.h | h | 2.55 KiB |
torch/include/ATen/ops/convolution_overrideable_native.h | h | 1.03 KiB |
torch/include/ATen/ops/convolution_overrideable_ops.h | h | 3.12 KiB |
torch/include/ATen/ops/copy.h | h | 1.30 KiB |
torch/include/ATen/ops/copy_compositeexplicitautograd_dispatch.h | h | 1.05 KiB |
torch/include/ATen/ops/copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 860 B |
torch/include/ATen/ops/copy_meta_dispatch.h | h | 792 B |
torch/include/ATen/ops/copy_native.h | h | 1.28 KiB |
torch/include/ATen/ops/copy_ops.h | h | 2.44 KiB |
torch/include/ATen/ops/copy_sparse_to_sparse.h | h | 1.76 KiB |
torch/include/ATen/ops/copy_sparse_to_sparse_compositeexplicitautograd_dispatch.h | h | 1.11 KiB |
torch/include/ATen/ops/copy_sparse_to_sparse_meta_dispatch.h | h | 806 B |
torch/include/ATen/ops/copy_sparse_to_sparse_native.h | h | 804 B |
torch/include/ATen/ops/copy_sparse_to_sparse_ops.h | h | 2.59 KiB |
torch/include/ATen/ops/copysign.h | h | 1.91 KiB |
torch/include/ATen/ops/copysign_compositeexplicitautograd_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/copysign_compositeexplicitautogradnonfunctional_dispatch.h | h | 921 B |
torch/include/ATen/ops/copysign_cpu_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/copysign_cuda_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/copysign_meta.h | h | 637 B |
torch/include/ATen/ops/copysign_meta_dispatch.h | h | 1.04 KiB |
torch/include/ATen/ops/copysign_native.h | h | 928 B |
torch/include/ATen/ops/copysign_ops.h | h | 4.25 KiB |
torch/include/ATen/ops/corrcoef.h | h | 652 B |
torch/include/ATen/ops/corrcoef_compositeimplicitautograd_dispatch.h | h | 789 B |
torch/include/ATen/ops/corrcoef_native.h | h | 499 B |
torch/include/ATen/ops/corrcoef_ops.h | h | 968 B |
torch/include/ATen/ops/cos.h | h | 1.12 KiB |
torch/include/ATen/ops/cos_compositeexplicitautogradnonfunctional_dispatch.h | h | 859 B |
torch/include/ATen/ops/cos_cpu_dispatch.h | h | 942 B |
torch/include/ATen/ops/cos_cuda_dispatch.h | h | 944 B |
torch/include/ATen/ops/cos_meta.h | h | 599 B |
torch/include/ATen/ops/cos_meta_dispatch.h | h | 944 B |
torch/include/ATen/ops/cos_native.h | h | 675 B |
torch/include/ATen/ops/cos_ops.h | h | 2.00 KiB |
torch/include/ATen/ops/cosh.h | h | 1.14 KiB |
torch/include/ATen/ops/cosh_compositeexplicitautogradnonfunctional_dispatch.h | h | 861 B |
torch/include/ATen/ops/cosh_cpu_dispatch.h | h | 946 B |
torch/include/ATen/ops/cosh_cuda_dispatch.h | h | 948 B |
torch/include/ATen/ops/cosh_meta.h | h | 600 B |
torch/include/ATen/ops/cosh_meta_dispatch.h | h | 948 B |
torch/include/ATen/ops/cosh_native.h | h | 613 B |
torch/include/ATen/ops/cosh_ops.h | h | 2.01 KiB |
torch/include/ATen/ops/cosine_embedding_loss.h | h | 925 B |
torch/include/ATen/ops/cosine_embedding_loss_compositeimplicitautograd_dispatch.h | h | 916 B |
torch/include/ATen/ops/cosine_embedding_loss_native.h | h | 626 B |
torch/include/ATen/ops/cosine_embedding_loss_ops.h | h | 1.28 KiB |
torch/include/ATen/ops/cosine_similarity.h | h | 791 B |
torch/include/ATen/ops/cosine_similarity_compositeimplicitautograd_dispatch.h | h | 852 B |
torch/include/ATen/ops/cosine_similarity_native.h | h | 562 B |
torch/include/ATen/ops/cosine_similarity_ops.h | h | 1.13 KiB |
torch/include/ATen/ops/count_nonzero.h | h | 2.03 KiB |
torch/include/ATen/ops/count_nonzero_compositeexplicitautograd_dispatch.h | h | 1.27 KiB |
torch/include/ATen/ops/count_nonzero_cpu_dispatch.h | h | 771 B |
torch/include/ATen/ops/count_nonzero_cuda_dispatch.h | h | 773 B |
torch/include/ATen/ops/count_nonzero_native.h | h | 959 B |
torch/include/ATen/ops/count_nonzero_ops.h | h | 3.10 KiB |
torch/include/ATen/ops/cov.h | h | 851 B |
torch/include/ATen/ops/cov_compositeimplicitautograd_dispatch.h | h | 904 B |
torch/include/ATen/ops/cov_native.h | h | 614 B |
torch/include/ATen/ops/cov_ops.h | h | 1.30 KiB |
torch/include/ATen/ops/cross.h | h | 1.32 KiB |
torch/include/ATen/ops/cross_compositeimplicitautograd_dispatch.h | h | 1.11 KiB |
torch/include/ATen/ops/cross_entropy_loss.h | h | 2.40 KiB |
torch/include/ATen/ops/cross_entropy_loss_compositeimplicitautograd_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cross_entropy_loss_native.h | h | 688 B |
torch/include/ATen/ops/cross_entropy_loss_ops.h | h | 1.44 KiB |
torch/include/ATen/ops/cross_native.h | h | 701 B |
torch/include/ATen/ops/cross_ops.h | h | 1.89 KiB |
torch/include/ATen/ops/crow_indices.h | h | 512 B |
torch/include/ATen/ops/crow_indices_compositeexplicitautograd_dispatch.h | h | 793 B |
torch/include/ATen/ops/crow_indices_copy.h | h | 1.13 KiB |
torch/include/ATen/ops/crow_indices_copy_compositeexplicitautograd_dispatch.h | h | 913 B |
torch/include/ATen/ops/crow_indices_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 824 B |
torch/include/ATen/ops/crow_indices_copy_native.h | h | 598 B |
torch/include/ATen/ops/crow_indices_copy_ops.h | h | 1.59 KiB |
torch/include/ATen/ops/crow_indices_native.h | h | 583 B |
torch/include/ATen/ops/crow_indices_ops.h | h | 986 B |
torch/include/ATen/ops/ctc_loss.h | h | 1.56 KiB |
torch/include/ATen/ops/ctc_loss_compositeimplicitautograd_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/ctc_loss_native.h | h | 918 B |
torch/include/ATen/ops/ctc_loss_ops.h | h | 2.49 KiB |
torch/include/ATen/ops/cudnn_affine_grid_generator.h | h | 1.48 KiB |
torch/include/ATen/ops/cudnn_affine_grid_generator_backward.h | h | 1.57 KiB |
torch/include/ATen/ops/cudnn_affine_grid_generator_backward_compositeexplicitautograd_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/cudnn_affine_grid_generator_backward_cuda_dispatch.h | h | 819 B |
torch/include/ATen/ops/cudnn_affine_grid_generator_backward_native.h | h | 724 B |
torch/include/ATen/ops/cudnn_affine_grid_generator_backward_ops.h | h | 2.01 KiB |
torch/include/ATen/ops/cudnn_affine_grid_generator_compositeexplicitautograd_dispatch.h | h | 1023 B |
torch/include/ATen/ops/cudnn_affine_grid_generator_cuda_dispatch.h | h | 811 B |
torch/include/ATen/ops/cudnn_affine_grid_generator_native.h | h | 716 B |
torch/include/ATen/ops/cudnn_affine_grid_generator_ops.h | h | 1.96 KiB |
torch/include/ATen/ops/cudnn_batch_norm.h | h | 2.97 KiB |
torch/include/ATen/ops/cudnn_batch_norm_backward.h | h | 3.16 KiB |
torch/include/ATen/ops/cudnn_batch_norm_backward_compositeexplicitautograd_dispatch.h | h | 1.65 KiB |
torch/include/ATen/ops/cudnn_batch_norm_backward_cuda_dispatch.h | h | 1.08 KiB |
torch/include/ATen/ops/cudnn_batch_norm_backward_native.h | h | 1.30 KiB |
torch/include/ATen/ops/cudnn_batch_norm_backward_ops.h | h | 3.94 KiB |
torch/include/ATen/ops/cudnn_batch_norm_compositeexplicitautograd_dispatch.h | h | 1.57 KiB |
torch/include/ATen/ops/cudnn_batch_norm_cuda_dispatch.h | h | 1.01 KiB |
torch/include/ATen/ops/cudnn_batch_norm_native.h | h | 1.20 KiB |
torch/include/ATen/ops/cudnn_batch_norm_ops.h | h | 3.63 KiB |
torch/include/ATen/ops/cudnn_convolution.h | h | 7.52 KiB |
torch/include/ATen/ops/cudnn_convolution_add_relu.h | h | 8.09 KiB |
torch/include/ATen/ops/cudnn_convolution_add_relu_compositeexplicitautograd_dispatch.h | h | 2.03 KiB |
torch/include/ATen/ops/cudnn_convolution_add_relu_cuda_dispatch.h | h | 1.28 KiB |
torch/include/ATen/ops/cudnn_convolution_add_relu_native.h | h | 1.06 KiB |
torch/include/ATen/ops/cudnn_convolution_add_relu_ops.h | h | 3.20 KiB |
torch/include/ATen/ops/cudnn_convolution_cuda_dispatch.h | h | 2.23 KiB |
torch/include/ATen/ops/cudnn_convolution_native.h | h | 940 B |
torch/include/ATen/ops/cudnn_convolution_ops.h | h | 2.81 KiB |
torch/include/ATen/ops/cudnn_convolution_relu.h | h | 6.94 KiB |
torch/include/ATen/ops/cudnn_convolution_relu_compositeexplicitautograd_dispatch.h | h | 1.76 KiB |
torch/include/ATen/ops/cudnn_convolution_relu_cuda_dispatch.h | h | 1.15 KiB |
torch/include/ATen/ops/cudnn_convolution_relu_native.h | h | 951 B |
torch/include/ATen/ops/cudnn_convolution_relu_ops.h | h | 2.76 KiB |
torch/include/ATen/ops/cudnn_convolution_transpose.h | h | 8.71 KiB |
torch/include/ATen/ops/cudnn_convolution_transpose_compositeexplicitautograd_dispatch.h | h | 1.95 KiB |
torch/include/ATen/ops/cudnn_convolution_transpose_cuda_dispatch.h | h | 1.25 KiB |
torch/include/ATen/ops/cudnn_convolution_transpose_native.h | h | 1.03 KiB |
torch/include/ATen/ops/cudnn_convolution_transpose_ops.h | h | 3.10 KiB |
torch/include/ATen/ops/cudnn_grid_sampler.h | h | 1.27 KiB |
torch/include/ATen/ops/cudnn_grid_sampler_backward.h | h | 1.76 KiB |
torch/include/ATen/ops/cudnn_grid_sampler_backward_compositeexplicitautograd_dispatch.h | h | 1.11 KiB |
torch/include/ATen/ops/cudnn_grid_sampler_backward_cuda_dispatch.h | h | 848 B |
torch/include/ATen/ops/cudnn_grid_sampler_backward_native.h | h | 804 B |
torch/include/ATen/ops/cudnn_grid_sampler_backward_ops.h | h | 2.28 KiB |
torch/include/ATen/ops/cudnn_grid_sampler_compositeexplicitautograd_dispatch.h | h | 965 B |
torch/include/ATen/ops/cudnn_grid_sampler_cuda_dispatch.h | h | 782 B |
torch/include/ATen/ops/cudnn_grid_sampler_native.h | h | 658 B |
torch/include/ATen/ops/cudnn_grid_sampler_ops.h | h | 1.77 KiB |
torch/include/ATen/ops/cudnn_is_acceptable.h | h | 688 B |
torch/include/ATen/ops/cudnn_is_acceptable_compositeimplicitautograd_dispatch.h | h | 794 B |
torch/include/ATen/ops/cudnn_is_acceptable_native.h | h | 504 B |
torch/include/ATen/ops/cudnn_is_acceptable_ops.h | h | 981 B |
torch/include/ATen/ops/cummax.h | h | 2.32 KiB |
torch/include/ATen/ops/cummax_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/cummax_compositeimplicitautograd_dispatch.h | h | 1.10 KiB |
torch/include/ATen/ops/cummax_native.h | h | 924 B |
torch/include/ATen/ops/cummax_ops.h | h | 3.44 KiB |
torch/include/ATen/ops/cummaxmin_backward.h | h | 819 B |
torch/include/ATen/ops/cummaxmin_backward_compositeimplicitautograd_dispatch.h | h | 866 B |
torch/include/ATen/ops/cummaxmin_backward_native.h | h | 576 B |
torch/include/ATen/ops/cummaxmin_backward_ops.h | h | 1.19 KiB |
torch/include/ATen/ops/cummin.h | h | 2.32 KiB |
torch/include/ATen/ops/cummin_compositeexplicitautograd_dispatch.h | h | 1.09 KiB |
torch/include/ATen/ops/cummin_compositeimplicitautograd_dispatch.h | h | 1.10 KiB |
torch/include/ATen/ops/cummin_native.h | h | 924 B |
torch/include/ATen/ops/cummin_ops.h | h | 3.44 KiB |
torch/include/ATen/ops/cumprod.h | h | 2.27 KiB |
torch/include/ATen/ops/cumprod_backward.h | h | 808 B |
torch/include/ATen/ops/cumprod_backward_compositeimplicitautograd_dispatch.h | h | 863 B |
torch/include/ATen/ops/cumprod_backward_native.h | h | 573 B |
torch/include/ATen/ops/cumprod_backward_ops.h | h | 1.18 KiB |
torch/include/ATen/ops/cumprod_compositeexplicitautogradnonfunctional_dispatch.h | h | 1001 B |
torch/include/ATen/ops/cumprod_compositeimplicitautograd_dispatch.h | h | 1.24 KiB |
torch/include/ATen/ops/cumprod_cpu_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cumprod_cuda_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cumprod_meta.h | h | 655 B |
torch/include/ATen/ops/cumprod_meta_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cumprod_native.h | h | 1.04 KiB |
torch/include/ATen/ops/cumprod_ops.h | h | 4.80 KiB |
torch/include/ATen/ops/cumsum.h | h | 2.25 KiB |
torch/include/ATen/ops/cumsum_compositeexplicitautogradnonfunctional_dispatch.h | h | 999 B |
torch/include/ATen/ops/cumsum_compositeimplicitautograd_dispatch.h | h | 1.24 KiB |
torch/include/ATen/ops/cumsum_cpu_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cumsum_cuda_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cumsum_meta.h | h | 654 B |
torch/include/ATen/ops/cumsum_meta_dispatch.h | h | 1.18 KiB |
torch/include/ATen/ops/cumsum_native.h | h | 1.03 KiB |
torch/include/ATen/ops/cumsum_ops.h | h | 4.78 KiB |
torch/include/ATen/ops/cumulative_trapezoid.h | h | 1023 B |
torch/include/ATen/ops/cumulative_trapezoid_compositeimplicitautograd_dispatch.h | h | 943 B |
torch/include/ATen/ops/cumulative_trapezoid_native.h | h | 653 B |
torch/include/ATen/ops/cumulative_trapezoid_ops.h | h | 1.76 KiB |
torch/include/ATen/ops/data.h | h | 504 B |
torch/include/ATen/ops/data_compositeimplicitautograd_dispatch.h | h | 785 B |
torch/include/ATen/ops/data_native.h | h | 495 B |
torch/include/ATen/ops/data_ops.h | h | 956 B |
torch/include/ATen/ops/deg2rad.h | h | 1.17 KiB |
torch/include/ATen/ops/deg2rad_compositeexplicitautograd_dispatch.h | h | 1002 B |
torch/include/ATen/ops/deg2rad_native.h | h | 1.04 KiB |
torch/include/ATen/ops/deg2rad_ops.h | h | 2.03 KiB |
torch/include/ATen/ops/dense_dim.h | h | 509 B |
torch/include/ATen/ops/dense_dim_compositeexplicitautograd_dispatch.h | h | 787 B |
torch/include/ATen/ops/dense_dim_native.h | h | 633 B |
torch/include/ATen/ops/dense_dim_ops.h | h | 959 B |
torch/include/ATen/ops/dequantize.h | h | 1.71 KiB |
torch/include/ATen/ops/dequantize_compositeexplicitautograd_dispatch.h | h | 1.03 KiB |
torch/include/ATen/ops/dequantize_cpu_dispatch.h | h | 747 B |
torch/include/ATen/ops/dequantize_cuda_dispatch.h | h | 749 B |
torch/include/ATen/ops/dequantize_native.h | h | 849 B |
torch/include/ATen/ops/dequantize_ops.h | h | 2.75 KiB |
torch/include/ATen/ops/det.h | h | 632 B |
torch/include/ATen/ops/det_compositeimplicitautograd_dispatch.h | h | 784 B |
torch/include/ATen/ops/det_native.h | h | 494 B |
torch/include/ATen/ops/det_ops.h | h | 953 B |
torch/include/ATen/ops/detach.h | h | 797 B |
torch/include/ATen/ops/detach_compositeexplicitautograd_dispatch.h | h | 839 B |
torch/include/ATen/ops/detach_copy.h | h | 1.07 KiB |
torch/include/ATen/ops/detach_copy_compositeexplicitautograd_dispatch.h | h | 901 B |
torch/include/ATen/ops/detach_copy_compositeexplicitautogradnonfunctional_dispatch.h | h | 818 B |
torch/include/ATen/ops/detach_copy_native.h | h | 586 B |
torch/include/ATen/ops/detach_copy_ops.h | h | 1.56 KiB |
torch/include/ATen/ops/detach_native.h | h | 549 B |
torch/include/ATen/ops/detach_ops.h | h | 1.44 KiB |
torch/include/ATen/ops/diag.h | h | 1.13 KiB |
torch/include/ATen/ops/diag_compositeimplicitautograd_dispatch.h | h | 998 B |
torch/include/ATen/ops/diag_embed.h | h | 1.38 KiB |
torch/include/ATen/ops/diag_embed_compositeexplicitautograd_dispatch.h | h | 995 B |
torch/include/ATen/ops/diag_embed_compositeexplicitautogradnonfunctional_dispatch.h | h | 869 B |
torch/include/ATen/ops/diag_embed_native.h | h | 680 B |
File types
File type | Files | Total size |
---|---|---|
pyd | 9 | 443.44 MiB |
py | 2,319 | 36.14 MiB |
dll | 112 | 11.05 GiB |
pyi | 33 | 2.77 MiB |
yaml | 1 | 8.37 KiB |
cpp | 5 | 19.94 KiB |
h | 8,589 | 26.81 MiB |
exe | 10 | 26.22 MiB |
cuh | 54 | 623.23 KiB |
hpp | 40 | 305.96 KiB |
typed | 1 | 0 B |
cmake | 29 | 263.03 KiB |
js | 1 | 19.47 KiB |
mjs | 2 | 11.05 KiB |
html | 1 | 405 B |
categories | 14 | 39.33 KiB |
bat | 1 | 66 B |
json | 1 | 396 B |
Total | 11,222 | 11.57 GiB |
Packages
SubID | Name | Billing Type | Last Update | |
---|---|---|---|---|
630589 | SVFI - Advanced-GPU Optimization Pack | Free on Demand |
Apps that reference this depot
AppID | Type | Name | Last Update | |
---|---|---|---|---|
1758570 | DLC | SVFI - Advanced-GPU Optimization Pack |
Previously seen manifests
Copy format:
Seen Date | Relative Date | ManifestID | |
---|---|---|---|
12 February 2025 – 07:37:10 UTC | 5138775820992547222 | ||
12 February 2025 – 04:20:31 UTC | 7111158572143106672 | ||
14 October 2023 – 12:34:03 UTC | 3804877537452420472 | ||
1 February 2023 – 02:31:47 UTC | 115690711392845611 | ||
18 November 2021 – 11:09:49 UTC | 9121642250966709537 |
Displaying change Go back
History
Sign in to view older history
Crunching history, just for you…