Searched refs:cluster_a (Results 1 – 3 of 3) sorted by relevance
46 mlir::LogicalResult ShouldMergeClusters(mlir::tf_device::ClusterOp cluster_a, in ShouldMergeClusters() argument49 if (cluster_a->getParentRegion() != cluster_b->getParentRegion()) { in ShouldMergeClusters()54 auto mesh_a_or_status = ExtractDeviceMeshFromOp(cluster_a.getOperation()); in ShouldMergeClusters()56 return cluster_a.emitOpError(mesh_a_or_status.status().error_message()); in ShouldMergeClusters()65 return !mesh_a ? cluster_a.emitOpError(kMissingMeshAttributeErrorMessage) in ShouldMergeClusters()
92 const uint32_t cluster_a = processor_a->package_leader_id; in cmp_arm_linux_processor() local94 if (cluster_a != cluster_b) { in cmp_arm_linux_processor()95 return cluster_a > cluster_b ? -1 : 1; in cmp_arm_linux_processor()
331 StatusOr<bool> AreDevicesCompatible(const Cluster& cluster_a,1721 const Cluster& cluster_a, const Cluster& cluster_b) { in AreDevicesCompatible() argument1722 DeviceSet devices = cluster_a.devices(); in AreDevicesCompatible()1746 return resource_op_device_ok(cluster_a.resource_op_device()) && in AreDevicesCompatible()