Searched refs:prepend_rank (Results 1 – 1 of 1) sorted by relevance
690 int64 prepend_rank = c->Value(c->Dim(c->input(2), 0)); in __anona2f81a441702() local692 prepend_rank != InferenceContext::kUnknownDim) { in __anona2f81a441702()695 dims.reserve(prepend_rank + input_rank); in __anona2f81a441702()696 for (int i = 0; i < prepend_rank; ++i) { in __anona2f81a441702()