Searched refs:known_in_elems (Results 1 – 1 of 1) sorted by relevance
221 DimensionHandle known_in_elems = c->NumElements(in); in SetOutputShapeForReshape() local222 if (!c->ValueKnown(known_in_elems)) { in SetOutputShapeForReshape()223 known_in_elems = c->MakeDim(1); in SetOutputShapeForReshape()233 TF_RETURN_IF_ERROR(c->Multiply(known_in_elems, dim, &known_in_elems)); in SetOutputShapeForReshape()241 if (c->Value(known_in_elems) != c->Value(known_out_elems)) { in SetOutputShapeForReshape()243 "Cannot reshape a tensor with ", c->DebugString(known_in_elems), in SetOutputShapeForReshape()251 TF_RETURN_IF_ERROR(c->Divide(known_in_elems, c->Value(known_out_elems), in SetOutputShapeForReshape()258 c->Value(known_in_elems) != 0) { in SetOutputShapeForReshape()261 TF_RETURN_IF_ERROR(c->Divide(known_out_elems, c->Value(known_in_elems), in SetOutputShapeForReshape()270 if (c->Value(known_in_elems) == c->Value(known_out_elems)) { in SetOutputShapeForReshape()