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
|
#pragma once
#include <string>
#include <memory>
#include <torch/csrc/WindowsTorchApiMacro.h>
namespace torch { namespace autograd {
// forward declaration of Node from function.h
struct Node;
struct TORCH_API AnomalyMode {
static bool is_enabled() {
return _enabled;
}
static void set_enabled(bool enabled) {
_enabled = enabled;
}
private:
static bool _enabled;
};
struct TORCH_API AnomalyMetadata {
virtual ~AnomalyMetadata();
virtual void store_stack() = 0;
virtual void print_stack(const std::string& current_node_name) = 0;
virtual void assign_parent(const std::shared_ptr<Node>& parent_node) = 0;
};
}}
|
