Searched refs:tiled_input_size (Results 1 – 1 of 1) sorted by relevance
246 auto tiled_sharding_mismatched = [&](int tiled_input_size) { in ExtractInputsForLogicalDevices() argument251 input_index, tiled_input_size, num_cores_per_replica)); in ExtractInputsForLogicalDevices()