Searched refs:assigned_device_name_index (Results 1 – 5 of 5) sorted by relevance
55 int recv_dev = node->assigned_device_name_index(); in ReplaceReduce()65 int send_dev = src_node->assigned_device_name_index(); in ReplaceReduce()120 int send_dev = node->assigned_device_name_index(); in ReplaceBroadcast()129 : edge->dst()->assigned_device_name_index(); in ReplaceBroadcast()
237 assigned_device = input->assigned_device_name_index(); in Run()267 int output_device_name = output->assigned_device_name_index(); in Run()272 return e->dst()->assigned_device_name_index() == output_device_name; in Run()
468 if (node.assigned_device_name_index() == assigned_device_name_index_) { in AssignDevice()502 assigned_device_name_index_ = node.assigned_device_name_index(); in AssignDevice()1095 if (node.assigned_device_name_index() < 0) { in LimitToAssignedDevice()
3842 new_node->set_assigned_device_name_index(mklop->assigned_device_name_index()); in FuseTransposeMklOpTranspose()
130 int assigned_device_name_index() const { return assigned_device_name_index_; } in assigned_device_name_index() function669 return device_names_[node.assigned_device_name_index()]; in get_assigned_device_name()