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