Searched refs:known_in_elems (Results 1 – 1 of 1) sorted by relevance
215 DimensionHandle known_in_elems = c->NumElements(in); in SetOutputShapeForReshape() local216 if (!c->ValueKnown(known_in_elems)) { in SetOutputShapeForReshape()217 known_in_elems = c->MakeDim(1); in SetOutputShapeForReshape()227 TF_RETURN_IF_ERROR(c->Multiply(known_in_elems, dim, &known_in_elems)); in SetOutputShapeForReshape()235 if (c->Value(known_in_elems) != c->Value(known_out_elems)) { in SetOutputShapeForReshape()237 "Cannot reshape a tensor with ", c->DebugString(known_in_elems), in SetOutputShapeForReshape()245 TF_RETURN_IF_ERROR(c->Divide(known_in_elems, c->Value(known_out_elems), in SetOutputShapeForReshape()252 c->Value(known_in_elems) != 0) { in SetOutputShapeForReshape()255 TF_RETURN_IF_ERROR(c->Divide(known_out_elems, c->Value(known_in_elems), in SetOutputShapeForReshape()264 if (c->Value(known_in_elems) == c->Value(known_out_elems)) { in SetOutputShapeForReshape()