Searched refs:bitcast_shape (Results 1 – 1 of 1) sorted by relevance
4870 auto bitcast_shape = ShapeUtil::MakeShape(F32, {kFeature, kOutput}); in TEST_P() local4882 builder.AddInstruction(HloInstruction::CreateBitcast(bitcast_shape, rhs)); in TEST_P()