Searched refs:original_input_shape (Results 1 – 1 of 1) sorted by relevance
538 const ArrayRef<int64_t> &original_input_shape = in RewriteNewAxisMask() local542 const int original_input_rank = original_input_shape.size(); in RewriteNewAxisMask()547 revised_shape.emplace_back(original_input_shape[index++]); in RewriteNewAxisMask()