Searched refs:elem_shape_F32 (Results 1 – 1 of 1) sorted by relevance
127 const Shape elem_shape_F32 = in RunTest1D() local132 HloInstruction::CreateParameter(0, elem_shape_F32, "0")); in RunTest1D()139 HloInstruction::CreateConvert(elem_shape_F32, param1)); in RunTest1D()141 elem_shape_F32, HloOpcode::kAdd, param0, param1_F32)); in RunTest1D()146 HloInstruction::CreateConvert(elem_shape_F32, sub_U8)); in RunTest1D()166 HloInstruction::CreateGetTupleElement(elem_shape_F32, tuple, 1)); in RunTest1D()