Searched refs:relugrad_node_view (Results 1 – 1 of 1) sorted by relevance
1840 const utils::MutableNodeView* relugrad_node_view = in FindFusedBatchNormGradEx() local1842 const NodeDef* relugrad_node_def = relugrad_node_view->node(); in FindFusedBatchNormGradEx()1845 if (!is_relugrad || HasControlFaninOrFanout(*relugrad_node_view) || in FindFusedBatchNormGradEx()1849 if (relugrad_node_view->NumRegularFanins() < 1) return false; in FindFusedBatchNormGradEx()1853 relugrad_node_view->GetRegularFanin(1); in FindFusedBatchNormGradEx()1863 if (fwd_bn_act_used && relugrad_node_view->GetRegularFanout(0).size() == 1) { in FindFusedBatchNormGradEx()1872 relugrad_node_view->GetRegularFanout(0).size() == 2) { in FindFusedBatchNormGradEx()1884 relugrad_node_view->GetRegularFanouts()[0]; in FindFusedBatchNormGradEx()