Searched refs:logical_core_0_sharding (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_sharding_identification_pass.cc | 182 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationInputs() argument
236 sharding_for_args.push_back(logical_core_0_sharding); in IdentifyXlaShardingForComputationInputs()
352 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationOutputs() argument
406 sharding_for_rets.push_back(logical_core_0_sharding); in IdentifyXlaShardingForComputationOutputs()
421 const std::string logical_core_0_sharding = in IdentifyXlaShardingForTPUComputation() local
429 IdentifyXlaShardingForComputationInputs(logical_core_0_sharding, use_spmd, in IdentifyXlaShardingForTPUComputation()
436 logical_core_0_sharding, use_spmd, /*infer_from_computation=*/true, in IdentifyXlaShardingForTPUComputation()
461 logical_core_0_sharding, in IdentifyXlaShardingForTPUComputation()
465 logical_core_0_sharding, in IdentifyXlaShardingForTPUComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | set_default_sharding.cc | 37 const std::string logical_core_0_sharding = in SetDefaultSharding() local
41 logical_core_0_sharding); in SetDefaultSharding()
43 cluster.getNumResults(), logical_core_0_sharding); in SetDefaultSharding()
|