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

Function GetMklOpName

tensorflow/core/graph/mkl_graph_util.h:128–130  ·  view source on GitHub ↗

Get the name of OneDNN op from original TensorFlow op We prefix 'OneDNN' to the original op to get OneDNN op.

Source from the content-addressed store, hash-verified

126// Get the name of OneDNN op from original TensorFlow op
127// We prefix 'OneDNN' to the original op to get OneDNN op.
128inline string GetMklOpName(const string& name) {
129 return string(kMklOpPrefix) + name;
130}
131
132// Get the name of OneDNN Eager op from original TensorFlow op
133// We prefix 'MklEager' to the original op to get OneDNN Eager op.

Callers 11

CreateGenericMklOpMethod · 0.85
ShouldRewriteOpMethod · 0.85
IsMklElementWiseOpFunction · 0.85
MklLayoutRewritePassMethod · 0.85
CheckForMklOpMethod · 0.85
LrnGradRewriteMethod · 0.85
MaxpoolGradRewriteMethod · 0.85
CheckForNodeRewriteMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected