Searched refs:explicit_sharding (Results 1 – 2 of 2) sorted by relevance
58 absl::optional<xla::OpSharding> explicit_sharding, in ParseShardingFromDevice() argument61 return explicit_sharding; in ParseShardingFromDevice()69 if (explicit_sharding.has_value()) { in ParseShardingFromDevice()70 return explicit_sharding; in ParseShardingFromDevice()
38 absl::optional<xla::OpSharding> explicit_sharding = absl::nullopt,