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