Searched refs:in_unknown_idx (Results 1 – 1 of 1) sorted by relevance
220 int32 in_unknown_idx = -1; in SetOutputShapeForReshape() local227 if (in_unknown_idx >= 0) { in SetOutputShapeForReshape()231 in_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()257 } else if (in_unknown_idx >= 0 && out_unknown_idx < 0 && in SetOutputShapeForReshape()264 DimensionHandle unknown_in_dim = c->Dim(in, in_unknown_idx); in SetOutputShapeForReshape()267 } else if (in_unknown_idx >= 0 && out_unknown_idx >= 0) { in SetOutputShapeForReshape()271 DimensionHandle unknown_in_dim = c->Dim(in, in_unknown_idx); in SetOutputShapeForReshape()