Searched refs:fwd_matched (Results 1 – 1 of 1) sorted by relevance
1855 FusedBatchNormEx fwd_matched; in FindFusedBatchNormGradEx() local1856 FindFusedBatchNormEx(ctx, fwd_node_view->node_index(), &fwd_matched); in FindFusedBatchNormGradEx()1857 bool fwd_bn_act_used = fwd_matched.activation != kMissingIndex && in FindFusedBatchNormGradEx()1858 fwd_matched.side_input == kMissingIndex; in FindFusedBatchNormGradEx()1859 bool fwd_bn_add_act_used = fwd_matched.activation != kMissingIndex && in FindFusedBatchNormGradEx()1860 fwd_matched.side_input != kMissingIndex; in FindFusedBatchNormGradEx()1866 matched->fwd_fused_batch_norm = fwd_matched.fused_batch_norm; in FindFusedBatchNormGradEx()1879 if (fwd_matched.fused_batch_norm != fwd_batch_norm_node.node_index()) { in FindFusedBatchNormGradEx()1895 matched->fwd_fused_batch_norm = fwd_matched.fused_batch_norm; in FindFusedBatchNormGradEx()