Searched defs:output_sharding_config (Results 1 – 2 of 2) sorted by relevance
375 llvm::ArrayRef<xla::OpSharding> output_sharding_config, const int core_id, in MapClusterOutputIndexWithRegionOutputIndex()501 const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, in GetOutputTypesForLogicalDeviceComputation()538 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in RemapOutputsFromLogicalDevices()
450 const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, in BuildExecuteOp()471 llvm::ArrayRef<xla::OpSharding> output_sharding_config, in BuildParallelExecuteOp()