/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/runtime
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/runtime/symbolic_script.h (598B)
#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
#include
#include
namespace torch {
namespace jit {
struct GradientPair {
std::shared_ptr forward;
std::shared_ptr backward;
};
TORCH_API c10::optional gradientInfoForSchema(
const FunctionSchema& schema);
TORCH_API bool hasGradientInfoForSchema(const FunctionSchema& schema);
} // namespace jit
} // namespace torch