py::class_<Split, Expr, std::unique_ptr<Split, py::nodelete>> split(

nvfuser, "Split");

split.def(

"__str__",

[](Split* self) { return self->toString(/*indent_size=*/0); },

"Convert the Split to a string representation.");

split.def(

"outer",

&Split::outer,

R"(

)");

split.def(

"inner",

&Split::inner,

R"(

)");

py::class_<Merge, Expr, std::unique_ptr<Merge, py::nodelete>> merge(

nvfuser, "Merge");

merge.def(

"__str__",

[](Merge* self) { return self->toString(/*indent_size=*/0); },

"Convert the Merge to a string representation.");

merge.def(

"outer",

&Merge::outer,

R"(

)");

merge.def(

"inner",

&Merge::inner,

R"(

)");

py::class_<BroadcastOp, Expr, std::unique_ptr<BroadcastOp, py::nodelete>>

broadcast(nvfuser, "BroadcastOp");

broadcast.def(

"__str__",

[](BroadcastOp* self) { return self->toString(/*indent_size=*/0); },

"Convert the BroadcastOp to a string representation.");

py::class_<ReshapeOp, Expr, std::unique_ptr<ReshapeOp, py::nodelete>> reshape(

nvfuser, "ReshapeOp");

reshape.def(

"__str__",

[](ReshapeOp* self) { return self->toString(/*indent_size=*/0); },

"Convert the ReshapeOp to a string representation.");

py::class_<SqueezeOp, Expr, std::unique_ptr<SqueezeOp, py::nodelete>> squeeze(

nvfuser, "SqueezeOp");

squeeze.def(

"__str__",

[](SqueezeOp* self) { return self->toString(/*indent_size=*/0); },

"Convert the SqueezeOp to a string representation.");