Searched refs:IsNodeType (Results 1 – 1 of 1) sorted by relevance
231 bool IsNodeType(const string& node_name, const string& type) { in IsNodeType() function237 return IsNodeType(node_name, kTransposeNHWCToNCHW); in IsTransposeNHWCToNCHW()241 return IsNodeType(node_name, kTransposeNCHWToNHWC); in IsTransposeNCHWToNHWC()245 return IsNodeType(node_name, kDimMapNHWCToNCHW); in IsDimMapNHWCToNCHW()249 return IsNodeType(node_name, kDimMapNCHWToNHWC); in IsDimMapNCHWToNHWC()253 return IsNodeType(node_name, kVecPermuteNHWCToNCHW); in IsVecPermuteNHWCToNCHW()257 return IsNodeType(node_name, kVecPermuteNCHWToNHWC); in IsVecPermuteNCHWToNHWC()