Lines Matching refs:arg

17              py::arg("stage"), py::arg("var"), py::arg("align"))  in add_schedule_methods()
19 py::arg("stage"), py::arg("var"), py::arg("align") = LoopAlignStrategy::Auto) in add_schedule_methods()
22 py::arg("var")) in add_schedule_methods()
24 py::arg("var"), py::arg("factor"), py::arg("tail") = TailStrategy::Auto) in add_schedule_methods()
27 …py::arg("old"), py::arg("outer"), py::arg("inner"), py::arg("factor"), py::arg("tail") = TailStrat… in add_schedule_methods()
30 py::arg("inner"), py::arg("outer"), py::arg("fused")) in add_schedule_methods()
33 py::arg("var")) in add_schedule_methods()
36 …py::arg("x"), py::arg("y"), py::arg("xo"), py::arg("yo"), py::arg("xi"), py::arg("yi"), py::arg("x… in add_schedule_methods()
38 …py::arg("x"), py::arg("y"), py::arg("xi"), py::arg("yi"), py::arg("xfactor"), py::arg("yfactor"), … in add_schedule_methods()
40 …py::arg("previous"), py::arg("outers"), py::arg("inners"), py::arg("factors"), py::arg("tail") = T… in add_schedule_methods()
42 … py::arg("previous"), py::arg("inners"), py::arg("factors"), py::arg("tail") = TailStrategy::Auto) in add_schedule_methods()
43 .def("reorder", (T & (T::*)(const std::vector<VarOrRVar> &)) & T::reorder, py::arg("vars")) in add_schedule_methods()
48 .def("parallel", (T & (T::*)(const VarOrRVar &)) & T::parallel, py::arg("var")) in add_schedule_methods()
49 …RVar &, const Expr &, TailStrategy)) & T::parallel, py::arg("var"), py::arg("task_size"), py::arg(… in add_schedule_methods()
51 .def("vectorize", (T & (T::*)(const VarOrRVar &)) & T::vectorize, py::arg("var")) in add_schedule_methods()
52 …RVar &, const Expr &, TailStrategy)) & T::vectorize, py::arg("var"), py::arg("factor"), py::arg("t… in add_schedule_methods()
54 … (T & (T::*)(const VarOrRVar &, DeviceAPI)) & T::gpu_blocks, py::arg("block_x"), py::arg("device_a… in add_schedule_methods()
55 …&, const VarOrRVar &, DeviceAPI)) & T::gpu_blocks, py::arg("block_x"), py::arg("block_y"), py::arg in add_schedule_methods()
56 …arOrRVar &, DeviceAPI)) & T::gpu_blocks, py::arg("block_x"), py::arg("block_y"), py::arg("block_z"… in add_schedule_methods()
58 …rRVar &, const VarOrRVar &, DeviceAPI)) & T::gpu, py::arg("block_x"), py::arg("thread_x"), py::arg in add_schedule_methods()
59 …RVar &, DeviceAPI)) & T::gpu, py::arg("block_x"), py::arg("block_y"), py::arg("thread_x"), py::arg in add_schedule_methods()
60 … T::gpu, py::arg("block_x"), py::arg("block_y"), py::arg("block_z"), py::arg("thread_x"), py::arg(… in add_schedule_methods()
62 …T & (T::*)(const VarOrRVar &, DeviceAPI)) & T::gpu_threads, py::arg("thread_x"), py::arg("device_a… in add_schedule_methods()
63 … const VarOrRVar &, DeviceAPI)) & T::gpu_threads, py::arg("thread_x"), py::arg("thread_y"), py::ar… in add_schedule_methods()
64 …rRVar &, DeviceAPI)) & T::gpu_threads, py::arg("thread_x"), py::arg("thread_y"), py::arg("thread_z… in add_schedule_methods()
65 ….def("gpu_single_thread", (T & (T::*)(DeviceAPI)) & T::gpu_single_thread, py::arg("device_api") = … in add_schedule_methods()
67 …PI)) & T::gpu_tile, py::arg("x"), py::arg("bx"), py::arg("tx"), py::arg("x_size"), py::arg("tail")… in add_schedule_methods()
69 … DeviceAPI)) & T::gpu_tile, py::arg("x"), py::arg("tx"), py::arg("x_size"), py::arg("tail") = Tail… in add_schedule_methods()
71arg("x"), py::arg("y"), py::arg("bx"), py::arg("by"), py::arg("tx"), py::arg("ty"), py::arg("x_siz… in add_schedule_methods()
73 …le, py::arg("x"), py::arg("y"), py::arg("tx"), py::arg("ty"), py::arg("x_size"), py::arg("y_size")… in add_schedule_methods()
75arg("x"), py::arg("y"), py::arg("z"), py::arg("bx"), py::arg("by"), py::arg("bz"), py::arg("tx"), … in add_schedule_methods()
76arg("x"), py::arg("y"), py::arg("z"), py::arg("tx"), py::arg("ty"), py::arg("tz"), py::arg("x_size… in add_schedule_methods()
78 .def("rename", &T::rename, py::arg("old_name"), py::arg("new_name")) in add_schedule_methods()
80 .def("specialize", &T::specialize, py::arg("condition")) in add_schedule_methods()
81 .def("specialize_fail", &T::specialize_fail, py::arg("message")) in add_schedule_methods()
85 .def("atomic", &T::atomic, py::arg("override_associativity_test") = false) in add_schedule_methods()
87 .def("hexagon", &T::hexagon, py::arg("x") = Var::outermost()) in add_schedule_methods()
89 …pr, PrefetchBoundStrategy)) & T::prefetch, py::arg("func"), py::arg("var"), py::arg("offset") = 1,… in add_schedule_methods()
95 …py::arg("image"), py::arg("var"), py::arg("offset") = 1, py::arg("strategy") = PrefetchBoundStrate… in add_schedule_methods()