Searched defs:output_sharding_config (Results 1 – 2 of 2) sorted by relevance
391 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in GetTileShardedOutputsToMerge()418 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in HandleTileShardedOutputs()512 const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, in GetOutputTypesForLogicalDeviceComputation()555 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in RemapOutputsFromLogicalDevices()
438 const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, in BuildExecuteOp()508 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in AddToParallelExecuteOp()