Searched defs:output_sharding_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 373 llvm::ArrayRef<xla::OpSharding> output_sharding_config, const int core_id, in MapClusterOutputIndexWithRegionOutputIndex()
391 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in GetTileShardedOutputsToMerge()
413 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in HandleTileShardedOutputs()
501 const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, in GetOutputTypesForLogicalDeviceComputation()
538 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in RemapOutputsFromLogicalDevices()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 435 const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, in BuildExecuteOp()
456 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in BuildParallelExecuteOp()
|