Searched refs:xla_program_shape (Results 1 – 1 of 1) sorted by relevance
847 xla::ProgramShape xla_program_shape = in TEST() local851 xla::ShapeUtil::GetSubshape(xla_program_shape.parameters(0), {0}) in TEST()855 xla::ShapeUtil::GetSubshape(xla_program_shape.parameters(0), {1}) in TEST()858 xla_program_shape.result().layout())); in TEST()