Searched refs:device_assignment_attr_size (Results 1 – 2 of 2) sorted by relevance
359 const int device_assignment_attr_size = device_assignment_attr.size(); in GetGeneralTPUExecutionDeviceAssignment() local360 if (device_assignment_attr_size != expected_device_assignment_size) in GetGeneralTPUExecutionDeviceAssignment()
1448 const int64 device_assignment_attr_size = in ParseDeviceAssignmentAttr() local1450 if (device_assignment_attr.size() != device_assignment_attr_size) { in ParseDeviceAssignmentAttr()