Searched refs:elements_after (Results 1 – 2 of 2) sorted by relevance
160 int64 elements_after = xla::ShapeUtil::ElementsIn(shape); in ReshapeLiteral() local161 if (elements_before != elements_after) { in ReshapeLiteral()
753 int64 elements_after = ShapeUtil::ElementsIn(output->shape()); in Reshape() local754 if (elements_before != elements_after) { in Reshape()