tensor_compute_op.cc
9.78 KB
-
[TIR] Enhance Substitute, python bindings for Substitute/PostOrderVisit/IRTransform. (#5400) · 6cb5b882Substitute now takes a std::function to customize more replacing behaviors. Co-authored-by: Siyuan Feng <hzfengsy@sjtu.edu.cn> Co-authored-by: Siyuan Feng <hzfengsy@sjtu.edu.cn>
Tianqi Chen committed