Searched refs:IsRelu6 (Results 1 – 4 of 4) sorted by relevance
152 bool IsRelu6(const NodeDef& node);
478 bool IsRelu6(const NodeDef& node) { return node.op() == "Relu6"; } in IsRelu6() function
373 return IsRelu(node) || IsRelu6(node) || IsElu(node) || IsLeakyRelu(node) || in IsCpuCompatibleDataType()376 return IsRelu(node) || IsRelu6(node) || IsElu(node) || IsLeakyRelu(node); in IsCpuCompatibleDataType()
1658 if (IsRelu(*op) || IsRelu6(*op)) { in OpsAreSafeToHoist()2902 if (IsRelu(consumer) || IsRelu6(consumer)) { in TrySimplify()