Searched refs:fused_batch_norm_node (Results 1 – 1 of 1) sorted by relevance
1534 NodeDef* fused_batch_norm_node = in FindMklLayerNorm() local1538 if (!TryGetNodeAttr(*fused_batch_norm_node, kIsTraining, &is_training) || in FindMklLayerNorm()1542 if (!NodeIsOnCpu(fused_batch_norm_node)) return false; in FindMklLayerNorm()