Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h100 bool inline Check5DFormat(const NodeDef& ndef) { in Check5DFormat() function
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dmkl_eager_op_rewrite.cc234 if (Check5DFormat(ndef)) { in RewriteFusedBatchNormV3()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1694 if (Check5DFormat(n->def())) { in FusedBatchNormV3Rewrite()