Searched refs:tiled_sharding_mismatched (Results 1 – 1 of 1) sorted by relevance
246 auto tiled_sharding_mismatched = [&](int tiled_input_size) { in ExtractInputsForLogicalDevices() local274 return tiled_sharding_mismatched(partitioned_input.inputs().size()); in ExtractInputsForLogicalDevices()294 return tiled_sharding_mismatched(tiled_inputs.size()); in ExtractInputsForLogicalDevices()