Searched refs:assigned_device_name_index (Results 1 – 5 of 5) sorted by relevance
175 assigned_device = input->assigned_device_name_index(); in Run()205 int output_device_name = output->assigned_device_name_index(); in Run()210 return e->dst()->assigned_device_name_index() == output_device_name; in Run()
407 if (node.assigned_device_name_index() == assigned_device_name_index_) { in AssignDevice()434 assigned_device_name_index_ = node.assigned_device_name_index(); in AssignDevice()674 if (node.assigned_device_name_index() < 0) { in LimitToAssignedDevice()
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()
120 int assigned_device_name_index() const { return assigned_device_name_index_; } in assigned_device_name_index() function618 return device_names_[node.assigned_device_name_index()]; in get_assigned_device_name()
3410 new_node->set_assigned_device_name_index(mklop->assigned_device_name_index()); in FuseTransposeMklOpTranspose()