Searched refs:explicit_sharding (Results 1 – 2 of 2) sorted by relevance
54 absl::optional<xla::OpSharding> explicit_sharding) { in ParseShardingFromDevice() argument64 if (explicit_sharding.has_value()) { in ParseShardingFromDevice()65 return explicit_sharding; in ParseShardingFromDevice()
38 absl::optional<xla::OpSharding> explicit_sharding = absl::nullopt);