Home
last modified time | relevance | path

Searched defs:logical_core_0_sharding (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/
Dset_default_sharding.cc37 const std::string logical_core_0_sharding = in SetDefaultSharding() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_sharding_identification_pass.cc182 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