Searched refs:core_annotation (Results 1 – 2 of 2) sorted by relevance
513 const int core_annotation = sharding.value().tile_assignment_devices(0); in SetNodeShardingFromNeighbors() local514 if (core == -1 || core > core_annotation) { in SetNodeShardingFromNeighbors()515 core = core_annotation; in SetNodeShardingFromNeighbors()
1200 int64 core_annotation = in ParseAndValidateSharding() local1203 ValidateCoreNumber(core_annotation, num_cores_per_replica)); in ParseAndValidateSharding()1204 if (*inferred_core_id == -1 || *inferred_core_id > core_annotation) { in ParseAndValidateSharding()1205 *inferred_core_id = core_annotation; in ParseAndValidateSharding()