MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsSupportedActivation

Function IsSupportedActivation

tensorflow/core/grappler/optimizers/remapper.cc:417–423  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

415}
416
417bool IsSupportedActivation(const NodeDef& node) {
418 if (!DisableMKL()) {
419 return IsRelu(node) || IsRelu6(node) || IsElu(node) || IsGelu(node) ||
420 IsLeakyRelu(node) || IsTanh(node);
421 }
422 return IsRelu(node) || IsRelu6(node) || IsElu(node) || IsLeakyRelu(node);
423}
424
425inline bool HasControlFaninOrFanout(const utils::MutableNodeView& node_view) {
426 return node_view.NumControllingFanins() > 0 ||

Calls 7

IsReluFunction · 0.85
IsRelu6Function · 0.85
IsEluFunction · 0.85
IsGeluFunction · 0.85
IsLeakyReluFunction · 0.85
IsTanhFunction · 0.85
DisableMKLFunction · 0.70

Tested by

no test coverage detected