Template Function tvm::tirx::ContainsNode

Template Function tvm::tirx::ContainsNode#

Function Documentation#

template<typename Node, typename = std::enable_if_t<std::is_base_of_v<StmtObj, Node>>>
bool tvm::tirx::ContainsNode(const Stmt &stmt)#

Check if the statement contains the specified node type.

This utility potentially walks the entire statement, and should therefore not be used if it could otherwise be merged with another pass.

Parameters:

stmt – The statement to be searched

Returns:

Whether stmt contains Node