Searched refs:original_input_rank (Results 1 – 1 of 1) sorted by relevance
542 const int original_input_rank = original_input_shape.size(); in RewriteNewAxisMask() local543 while (index < original_input_rank || new_axis_mask) { in RewriteNewAxisMask()