Searched refs:set_assigned_device_name_index (Results 1 – 4 of 4) sorted by relevance
77 send_node->set_assigned_device_name_index(send_dev); in ReplaceReduce()92 recv_node->set_assigned_device_name_index(recv_dev); in ReplaceReduce()182 send_node->set_assigned_device_name_index(send_dev); in ReplaceBroadcast()211 shape_node->set_assigned_device_name_index(send_dev); in ReplaceBroadcast()226 shape_node->set_assigned_device_name_index(recv_dev); in ReplaceBroadcast()233 recv_node->set_assigned_device_name_index(recv_dev); in ReplaceBroadcast()
121 void set_assigned_device_name_index(int index);621 void set_assigned_device_name_index(Node* node, int device_name_index) { in set_assigned_device_name_index() function880 inline void Node::set_assigned_device_name_index(int index) { in set_assigned_device_name_index() function
3410 new_node->set_assigned_device_name_index(mklop->assigned_device_name_index()); in FuseTransposeMklOpTranspose()
70 node->set_assigned_device_name_index(assigned_device); in AssignAndLog()