Searched refs:op_sharding (Results 1 – 2 of 2) sorted by relevance
103 absl::optional<xla::OpSharding> op_sharding = in Compute() local109 xla::XlaScopedShardingAssignment assign_sharding(b, op_sharding); in Compute()
3985 OpSharding op_sharding; in ParseShardingOnly() local3986 if (!ParseSharding(&op_sharding)) { in ParseShardingOnly()3992 return HloSharding::FromProto(op_sharding); in ParseShardingOnly()