Searched refs:badd (Results 1 – 3 of 3) sorted by relevance
3318 Node* badd = m->type_string() == csinfo_.bias_add_grad ? m : n; in MergeConv2DBackpropFilterWithBiasAddGrad() local3324 TF_CHECK_OK(GetNodeAttr(badd->def(), "T", &T_b)); in MergeConv2DBackpropFilterWithBiasAddGrad()3326 TF_CHECK_OK(GetNodeAttr(badd->def(), "data_format", &data_format_b)); in MergeConv2DBackpropFilterWithBiasAddGrad()3329 badd->assigned_device_name() != fltr->assigned_device_name() || in MergeConv2DBackpropFilterWithBiasAddGrad()3330 badd->def().device() != fltr->def().device()) { in MergeConv2DBackpropFilterWithBiasAddGrad()3371 for (const Edge* e : badd->in_edges()) { in MergeConv2DBackpropFilterWithBiasAddGrad()3400 for (const Edge* e : badd->out_edges()) { in MergeConv2DBackpropFilterWithBiasAddGrad()3431 new_node->set_assigned_device_name(badd->assigned_device_name()); in MergeConv2DBackpropFilterWithBiasAddGrad()3433 VLOG(1) << "MklLayoutRewritePass: Merged old node:" << badd->DebugString() in MergeConv2DBackpropFilterWithBiasAddGrad()3437 (*g)->RemoveNode(badd); in MergeConv2DBackpropFilterWithBiasAddGrad()
11782 badd 5 1553 1163