Searched refs:in_unknown_idx (Results 1 – 1 of 1) sorted by relevance
214 int32 in_unknown_idx = -1; in SetOutputShapeForReshape() local221 if (in_unknown_idx >= 0) { in SetOutputShapeForReshape()225 in_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()251 } else if (in_unknown_idx >= 0 && out_unknown_idx < 0 && in SetOutputShapeForReshape()258 DimensionHandle unknown_in_dim = c->Dim(in, in_unknown_idx); in SetOutputShapeForReshape()261 } else if (in_unknown_idx >= 0 && out_unknown_idx >= 0) { in SetOutputShapeForReshape()265 DimensionHandle unknown_in_dim = c->Dim(in, in_unknown_idx); in SetOutputShapeForReshape()