Searched refs:elem_shapes (Results 1 – 1 of 1) sorted by relevance
363 std::vector<xla::Shape> elem_shapes; in BuildComputation() local367 elem_shapes.push_back(elem_shape); in BuildComputation()369 xla::Shape shape = xla::ShapeUtil::MakeTupleShape(elem_shapes); in BuildComputation()378 if (elem_shapes[i].IsTuple()) { in BuildComputation()383 xla::ShapeUtil::GetLeafCount(elem_shapes[i])); in BuildComputation()