forked from jiuyuan/InfiniTensor
16 lines
420 B
C
16 lines
420 B
C
|
#pragma once
|
||
|
|
#include "nnet/visitor.h"
|
||
|
|
|
||
|
|
namespace nnet {
|
||
|
|
|
||
|
|
// Clone ExprNodes in a stage except Tensor, Var, and Constant nodes.
|
||
|
|
class CloneMutator : public Mutator {
|
||
|
|
public:
|
||
|
|
CloneMutator() : Mutator(false) {}
|
||
|
|
Expr visit_(const Constant &c) override;
|
||
|
|
Expr visit_(const Var &c) override;
|
||
|
|
Expr visit_(const Tensor &c) override;
|
||
|
|
Expr clone(const Expr &c) { return dispatch(c); }
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace nnet
|