Lines Matching refs:in_idx
154 int64 in_idx = data[i];155 if (in_idx >= rank) {156 return errors::InvalidArgument("perm dim ", in_idx,159 dims[i] = c->Dim(input, in_idx);