xref: /aosp_15_r20/external/pytorch/tools/autograd/templates/TraceType.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #define TORCH_ASSERT_ONLY_METHOD_OPERATORS
2 #include "torch/csrc/jit/frontend/tracer.h"
3 
4 #include <torch/library.h>
5 
6 #include "torch/csrc/autograd/function.h"
7 
8 #include "ATen/quantized/Quantizer.h"
9 
10 // ${generated_comment}
11 
12 // See the `Tracer` section in `torch/csrc/jit/OVERVIEW.md`.
13 // NOTE See [Sharded File] comment in VariableType
14 
15 #ifndef AT_PER_OPERATOR_HEADERS
16 #include <ATen/Operators.h>
17 #else
18 $ops_headers
19 #endif
20 
21 using namespace at;
22 
23 namespace torch {
24 
25 namespace TraceType {
26 
27 namespace {
28 ${trace_method_definitions}
29 }  // namespace
30 }  // namespace TraceType
31 
32 namespace {
33 
TORCH_LIBRARY_IMPL(aten,Tracer,m)34 TORCH_LIBRARY_IMPL(aten, Tracer, m) {
35   ${trace_wrapper_registrations};
36 }
37 
38 }  // namespace
39 
40 } // namespace torch
41