when I create a topi operation as
conv_out_t = topi.nn.conv2d(inp_t, weight_t, stride, padding, dilation, layout=layout, out_dtype='float32') s1 = topi.generic.schedule_conv2d_nchw([conv_out_t])
When I don’t specify
out_dtype='float32' it will look up with
0 in place of
'float32'. If one wants lookups to succeed it might be nicer to use if properties would be derived when the arguments are not specified.
(I previously thought I also had funny effects w.r.t. workload names (conv2d vs. topi_nn_conv2d) - both show up in the tophub files but that might have been the above actually and only the conv2d is actually used.)