Searched refs:known_out_elems (Results 1 – 1 of 1) sorted by relevance
203 DimensionHandle known_out_elems = c->NumElements(out); in SetOutputShapeForReshape() local204 if (!c->ValueKnown(known_out_elems)) { in SetOutputShapeForReshape()205 known_out_elems = c->MakeDim(1); in SetOutputShapeForReshape()216 c->Multiply(known_out_elems, dim, &known_out_elems)); in SetOutputShapeForReshape()241 if (c->Value(known_in_elems) != c->Value(known_out_elems)) { in SetOutputShapeForReshape()245 c->DebugString(known_out_elems), " elements)"); in SetOutputShapeForReshape()248 c->Value(known_out_elems) > 0) { in SetOutputShapeForReshape()251 TF_RETURN_IF_ERROR(c->Divide(known_in_elems, c->Value(known_out_elems), 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()