Searched refs:known_out_elems (Results 1 – 1 of 1) sorted by relevance
197 DimensionHandle known_out_elems = c->NumElements(out); in SetOutputShapeForReshape() local198 if (!c->ValueKnown(known_out_elems)) { in SetOutputShapeForReshape()199 known_out_elems = c->MakeDim(1); in SetOutputShapeForReshape()210 c->Multiply(known_out_elems, dim, &known_out_elems)); in SetOutputShapeForReshape()235 if (c->Value(known_in_elems) != c->Value(known_out_elems)) { in SetOutputShapeForReshape()239 c->DebugString(known_out_elems), " elements)"); in SetOutputShapeForReshape()242 c->Value(known_out_elems) > 0) { in SetOutputShapeForReshape()245 TF_RETURN_IF_ERROR(c->Divide(known_in_elems, c->Value(known_out_elems), 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()