Searched refs:tile_sharded_outputs (Results 1 – 1 of 1) sorted by relevance
603 llvm::SmallVector<mlir::Value, 4> tile_sharded_outputs; in RemapOutputsFromLogicalDevices() local607 &tile_sharded_outputs))) in RemapOutputsFromLogicalDevices()610 llvm::zip(partitioned_output.output(), tile_sharded_outputs)) in RemapOutputsFromLogicalDevices()