Caffe2 - C++ API
A deep learning, cross platform ML framework
anomaly_mode.h
1 #pragma once
2 
3 #include <torch/csrc/WindowsTorchApiMacro.h>
4 
5 namespace torch { namespace autograd {
6 
7 struct AnomalyMode {
8  static bool is_enabled() {
9  return _enabled;
10  }
11  static void set_enabled(bool enabled) {
12  _enabled = enabled;
13  }
14 
15 private:
16  TORCH_API static bool _enabled;
17 };
18 
19 
21  virtual ~AnomalyMetadata() = default;
22  virtual void store_stack() = 0;
23  virtual void print_stack() = 0;
24 };
25 
26 }}
Definition: jit_type.h:17