Searched refs:mklop (Results 1 – 1 of 1) sorted by relevance
3778 Node* mklop = nodes[1]; in FuseTransposeMklOpTranspose() local3788 const int mklop_num_inputs = mklop->num_inputs(); in FuseTransposeMklOpTranspose()3791 FillInputs(mklop, &mklop_control_edges, &mklop_in); in FuseTransposeMklOpTranspose()3802 NodeBuilder nb(mklop->name(), mklop->type_string()); in FuseTransposeMklOpTranspose()3815 copy_attrs(const_cast<const Node*>(mklop), &nb, true); in FuseTransposeMklOpTranspose()3819 nb.Device(mklop->def().device()); in FuseTransposeMklOpTranspose()3838 new_node->set_assigned_device_name(mklop->assigned_device_name()); in FuseTransposeMklOpTranspose()3841 new_node->set_requested_device(mklop->requested_device()); in FuseTransposeMklOpTranspose()3842 new_node->set_assigned_device_name_index(mklop->assigned_device_name_index()); in FuseTransposeMklOpTranspose()3845 (*g)->RemoveNode(mklop); in FuseTransposeMklOpTranspose()