Searched refs:mklop (Results 1 – 1 of 1) sorted by relevance
3347 Node* mklop = nodes[1]; in FuseTransposeMklOpTranspose() local3357 const int mklop_num_inputs = mklop->num_inputs(); in FuseTransposeMklOpTranspose()3360 FillInputs(mklop, &mklop_control_edges, &mklop_in); in FuseTransposeMklOpTranspose()3371 NodeBuilder nb(mklop->name(), mklop->type_string()); in FuseTransposeMklOpTranspose()3384 copy_attrs(const_cast<const Node*>(mklop), &nb, true); in FuseTransposeMklOpTranspose()3388 nb.Device(mklop->def().device()); in FuseTransposeMklOpTranspose()3406 new_node->set_assigned_device_name(mklop->assigned_device_name()); in FuseTransposeMklOpTranspose()3409 new_node->set_requested_device(mklop->requested_device()); in FuseTransposeMklOpTranspose()3410 new_node->set_assigned_device_name_index(mklop->assigned_device_name_index()); in FuseTransposeMklOpTranspose()3413 (*g)->RemoveNode(mklop); in FuseTransposeMklOpTranspose()