tvm icon indicating copy to clipboard operation
tvm copied to clipboard

Published 20 hours ago • verified publisher icon apache

[SME][TOPI] Add conv2d NHWC SME fp16->fp32 schedule

Open Anndrey24 opened this issue 1 year ago • 0 comments

This commit extends the SME conv2d NHWC schedule to support convolutions with float16 inputs (data and kernel) and a float32 output using the tensor intrinsics added in #16981.

cc @ekalda @lhutton1

Anndrey24 avatar May 30 '24 16:05 Anndrey24