Eg: input is (128, 16) dtype=fp32, i wann obtain the fp32 data in fp16 mode that output is (128, 16, 2) dtype=fp16, but i don’t know how to deal these. res = tvm.compute((128, 16, 2), func(), name=“fp322fp16”) while input is (128,16) dtype=fp32 res = tvm.compute((128, 16), func(), name=“fp162fp32”) while input is (128,16,2) dtype=fp16

Is there anyone meet the same problem or can help me figure out it. I would appreciate it very much~