/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/variadic_ops.h (993B)
#pragma once
#include
namespace torch {
namespace jit {
// Replaces the `aten::cat` ops in the given graph with variadic cat ops.
// Returns true if the graph is modified.
TORCH_API bool UseVariadicCat(const std::shared_ptr& graph);
TORCH_API bool RemoveListMutationAndUseVariadicCat(
const std::shared_ptr& graph);
// Replaces the `aten::stack` ops in the given graph with variadic cat ops.
// Returns true if the graph is modified.
TORCH_API bool UseVariadicStack(const std::shared_ptr& graph);
TORCH_API bool RemoveListMutationAndUseVariadicStack(
const std::shared_ptr& graph);
TORCH_API bool UseVariadicOp(
const std::shared_ptr& graph,
NodeKind op,
NodeKind variadic_op,
size_t list_idx = 0);
TORCH_API bool RemoveListMutationAndUseVariadicOp(
const std::shared_ptr& graph,
NodeKind op,
NodeKind variadic_op,
size_t list_idx = 0);
} // namespace jit
} // namespace torch