Lines Matching defs:args

56 TVM_REGISTER_GLOBAL("topi.TEST_create_target").set_body([](TVMArgs args, TVMRetValue* rv) {  in __anonbd8a05830102()
61 TVM_REGISTER_GLOBAL("topi.generic.default_schedule").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830202()
69 TVM_REGISTER_GLOBAL("topi.generic.schedule_extern").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830302()
73 TVM_REGISTER_GLOBAL("topi.generic.schedule_injective").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830402()
78 .set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830502()
83 TVM_REGISTER_GLOBAL("topi.x86.schedule_binarize_pack").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830602()
87 TVM_REGISTER_GLOBAL("topi.x86.schedule_binary_dense").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830702()
91 TVM_REGISTER_GLOBAL("topi.x86.default_schedule").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830802()
99 TVM_REGISTER_GLOBAL("topi.x86.schedule_injective").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830902()
104 .set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830a02()
109 TVM_REGISTER_GLOBAL("topi.rocm.dense_cuda").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830b02()
113 TVM_REGISTER_GLOBAL("topi.rocm.schedule_dense").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830c02()
117 TVM_REGISTER_GLOBAL("topi.rocm.schedule_injective").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830d02()
122 .set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830e02()
126 TVM_REGISTER_GLOBAL("topi.rocm.schedule_pool").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05830f02()
130 TVM_REGISTER_GLOBAL("topi.rocm.schedule_global_pool").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831002()
134 TVM_REGISTER_GLOBAL("topi.rocm.schedule_reduce").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831102()
138 TVM_REGISTER_GLOBAL("topi.rocm.schedule_softmax").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831202()
142 TVM_REGISTER_GLOBAL("topi.rocm.schedule_lrn").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831302()
147 TVM_REGISTER_GLOBAL("topi.cuda.dense_cuda").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831402()
151 TVM_REGISTER_GLOBAL("topi.cuda.schedule_dense").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831502()
155 TVM_REGISTER_GLOBAL("topi.cuda.schedule_injective").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831602()
160 .set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831702()
164 TVM_REGISTER_GLOBAL("topi.cuda.schedule_pool").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831802()
168 TVM_REGISTER_GLOBAL("topi.cuda.schedule_global_pool").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831902()
172 TVM_REGISTER_GLOBAL("topi.cuda.schedule_reduce").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831a02()
176 TVM_REGISTER_GLOBAL("topi.cuda.schedule_softmax").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831b02()
180 TVM_REGISTER_GLOBAL("topi.cuda.schedule_lrn").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831c02()
185 TVM_REGISTER_GLOBAL("topi.util.is_empty_shape").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831d02()
189 TVM_REGISTER_GLOBAL("topi.util.bilinear_sample_nchw").set_body([](TVMArgs args, TVMRetValue* rv) { in __anonbd8a05831e02()
207 return PackedFunc([builder](TVMArgs args, TVMRetValue* ret) { in WrapSchedule()
277 [builder](TVMArgs args, TVMRetValue* ret) { *ret = builder(args[0], args[1]); }); in WrapScheduleFromExisting()
301 return PackedFunc([builder](TVMArgs args, TVMRetValue* ret) { in WrapDenseOp()