Searched defs:logical_core_0_sharding (Results 1 – 1 of 1) sorted by relevance
131 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationInputs()246 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationOutputs()299 const std::string logical_core_0_sharding = in IdentifyXlaShardingForTPUComputation() local