#pragma once
|
|
#include <torch/csrc/jit/ir.h>
|
|
namespace torch {
|
namespace jit {
|
|
// removes tuples where TupleConstruct and TupleUnpack are matched
|
// but leaves tuples in place across if statements, loops, and as inputs/outputs
|
TORCH_API void LowerSimpleTuples(std::shared_ptr<Graph>& graph);
|
|
// removes _all_ tuples and raises an error if some cannot be removed
|
// this is used by ONNX to ensure there are not tuples before conversion,
|
// but will not work on graphs whose inputs contain tuples.
|
TORCH_API void LowerAllTuples(std::shared_ptr<Graph>& graph);
|
|
TORCH_API void LowerSimpleTuples(Block* block);
|
|
} // namespace jit
|
} // namespace torch
|
