Searched refs:tiled_inputs (Results 1 – 1 of 1) sorted by relevance
148 llvm::SmallVectorImpl<mlir::Value>* tiled_inputs) { in HandleTileShardedInputs() argument195 tiled_inputs->reserve(input_sharding.tile_assignment_devices_size()); in HandleTileShardedInputs()197 tiled_inputs->append(split_op.getResults().begin(), in HandleTileShardedInputs()287 llvm::SmallVector<mlir::Value, 4> tiled_inputs; in ExtractInputsForLogicalDevices() local289 cluster_func.getLoc(), sharding, input_value, builder, &tiled_inputs); in ExtractInputsForLogicalDevices()292 const int64 tiled_inputs_size = tiled_inputs.size(); in ExtractInputsForLogicalDevices()294 return tiled_sharding_mismatched(tiled_inputs.size()); in ExtractInputsForLogicalDevices()298 (*input_list)[assigned_logical_device].emplace_back(tiled_inputs[i]); in ExtractInputsForLogicalDevices()