Searched defs:sharding_proto (Results 1 – 2 of 2) sorted by relevance
498 for (const auto& sharding_proto : proto.spmd_parameters_shardings()) { in CreateFromProto() local
459 xla::OpSharding sharding_proto; in CreateOpShardingFromStringRef() local