Searched refs:IsLeakyRelu (Results 1 – 3 of 3) sorted by relevance
373 return IsRelu(node) || IsRelu6(node) || IsElu(node) || IsLeakyRelu(node) || in IsCpuCompatibleDataType()376 return IsRelu(node) || IsRelu6(node) || IsElu(node) || IsLeakyRelu(node); in IsCpuCompatibleDataType()476 IsLeakyRelu(*node_def)) in IsCpuCompatibleDataType()761 if (!IsConv2D(*contraction_node_def) && IsLeakyRelu(*node_def)) return false; in IsCpuCompatibleDataType()980 if (activation != nullptr && IsLeakyRelu(*activation)) { in IsCpuCompatibleDataType()1001 if (activation != nullptr && IsLeakyRelu(*activation)) { in IsCpuCompatibleDataType()1044 if (activation != nullptr && IsLeakyRelu(*activation)) { in IsCpuCompatibleDataType()
104 bool IsLeakyRelu(const NodeDef& node);
345 bool IsLeakyRelu(const NodeDef& node) { return node.op() == "LeakyRelu"; } in IsLeakyRelu() function