Searched defs:logical_core_0_sharding (Results 1 – 2 of 2) sorted by relevance
182 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationInputs()352 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationOutputs()421 const std::string logical_core_0_sharding = in IdentifyXlaShardingForTPUComputation() local
37 const std::string logical_core_0_sharding = in SetDefaultSharding() local