Searched refs:output_sub_shape (Results 1 – 1 of 1) sorted by relevance
333 xla::Shape* output_sub_shape = xla::ShapeUtil::GetMutableSubshape( in BuildComputation() local335 *output_sub_shape->mutable_layout() = index_and_layout.second; in BuildComputation()