Searched refs:tiled_inputs (Results 1 – 1 of 1) sorted by relevance
142 llvm::SmallVectorImpl<mlir::Value>* tiled_inputs) { in HandleTileShardedInputs() argument189 tiled_inputs->reserve(input_sharding.tile_assignment_devices_size()); in HandleTileShardedInputs()191 tiled_inputs->append(split_op.getResults().begin(), in HandleTileShardedInputs()281 llvm::SmallVector<mlir::Value, 4> tiled_inputs; in ExtractInputsForLogicalDevices() local283 cluster_func.getLoc(), sharding, input_value, builder, &tiled_inputs); in ExtractInputsForLogicalDevices()286 const int64_t tiled_inputs_size = tiled_inputs.size(); in ExtractInputsForLogicalDevices()288 return tiled_sharding_mismatched(tiled_inputs.size()); in ExtractInputsForLogicalDevices()292 (*input_list)[assigned_logical_device].emplace_back(tiled_inputs[i]); in ExtractInputsForLogicalDevices()