1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#pragma once
// This file is temporary until native_functions.yaml and derivatives.yaml are
// merged. Ideally this should all go into native_functions.yaml
#include <c10/util/StringUtil.h>
#include <torch/csrc/jit/api/module.h>
#include <optional>
namespace torch::jit {
struct GradientPair {
std::shared_ptr<Graph> forward;
std::shared_ptr<Graph> backward;
};
TORCH_API std::optional<GradientPair> gradientInfoForSchema(
const FunctionSchema& schema);
TORCH_API bool hasGradientInfoForSchema(const FunctionSchema& schema);
} // namespace torch::jit
|
