/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes/lower_grad_of.h (348B)
#pragma once
#include
namespace torch {
namespace jit {
// This pass removes 'grad_of' nodes, replacing them with conditionals of
// the form:
// if any_defined(inputs):
// outputs =
// else:
// outputs = undefineds
TORCH_API void LowerGradOf(Graph& g);
} // namespace jit
} // namespace torch