Searched refs:elem_shape_U8 (Results 1 – 1 of 1) sorted by relevance
129 const Shape elem_shape_U8 = in RunTest1D() local134 HloInstruction::CreateParameter(1, elem_shape_U8, "1")); in RunTest1D()137 HloInstruction::CreateConvert(elem_shape_U8, param0)); in RunTest1D()144 elem_shape_U8, HloOpcode::kSubtract, param0_U8, param1)); in RunTest1D()164 HloInstruction::CreateGetTupleElement(elem_shape_U8, tuple, 0)); in RunTest1D()