Searched refs:out_unknown_idx (Results 1 – 1 of 1) sorted by relevance
195 int32 out_unknown_idx = -1; in SetOutputShapeForReshape() local203 if (out_unknown_idx >= 0) { in SetOutputShapeForReshape()207 out_unknown_idx = i; in SetOutputShapeForReshape()233 if (in_unknown_idx < 0 && out_unknown_idx < 0) { in SetOutputShapeForReshape()241 } else if (in_unknown_idx < 0 && out_unknown_idx >= 0 && in SetOutputShapeForReshape()249 c->ReplaceDim(out, out_unknown_idx, inferred_dim, &out)); in SetOutputShapeForReshape()251 } else if (in_unknown_idx >= 0 && out_unknown_idx < 0 && in SetOutputShapeForReshape()261 } else if (in_unknown_idx >= 0 && out_unknown_idx >= 0) { in SetOutputShapeForReshape()267 c->ReplaceDim(out, out_unknown_idx, unknown_in_dim, &out)); in SetOutputShapeForReshape()