Home
last modified time | relevance | path

Searched refs:IsLeakyRelu (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc373 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()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h104 bool IsLeakyRelu(const NodeDef& node);
Dop_types.cc345 bool IsLeakyRelu(const NodeDef& node) { return node.op() == "LeakyRelu"; } in IsLeakyRelu() function