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