Searched refs:elements_after (Results 1 – 2 of 2) sorted by relevance
72 int64 elements_after = xla::ShapeUtil::ElementsIn(shape); in ReshapeLiteral() local73 if (elements_before != elements_after) { in ReshapeLiteral()
665 int64 elements_after = ShapeUtil::ElementsIn(output.shape()); in Reshape() local666 if (elements_before != elements_after) { in Reshape()