Searched refs:out_unknown_idx (Results 1 – 1 of 1) sorted by relevance
201 int32 out_unknown_idx = -1; in SetOutputShapeForReshape() local209 if (out_unknown_idx >= 0) { in SetOutputShapeForReshape()213 out_unknown_idx = i; in SetOutputShapeForReshape()239 if (in_unknown_idx < 0 && out_unknown_idx < 0) { in SetOutputShapeForReshape()247 } else if (in_unknown_idx < 0 && out_unknown_idx >= 0 && in SetOutputShapeForReshape()255 c->ReplaceDim(out, out_unknown_idx, inferred_dim, &out)); in SetOutputShapeForReshape()257 } else if (in_unknown_idx >= 0 && out_unknown_idx < 0 && in SetOutputShapeForReshape()267 } else if (in_unknown_idx >= 0 && out_unknown_idx >= 0) { in SetOutputShapeForReshape()273 c->ReplaceDim(out, out_unknown_idx, unknown_in_dim, &out)); in SetOutputShapeForReshape()