src/blocksparse_conv_op.cc [158:190]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    .Input("fprop_grid: int32")
    .Input("bprop_grid: int32")
    .Input("updat_grid: int32")
    .Input("mpq_lut: int32")
    .Input("dhw_lut: int32")
    .Input("ck_lut: int32")
    .Input("a: a_type")
    .Input("b: b_type")
    .Output("c: c_type")
    .Attr("a_type: {half, float, bfloat16}")
    .Attr("b_type: {half, float, bfloat16}")
    .Attr("c_type: {half, float, bfloat16}")
    .Attr("mode: int >=0 =0")
    .Attr("overlapC: bool = false")
    .Attr("overlapK: bool = false")
    .Attr("C: int >=0")
    .Attr("K: int >=0")
    .Attr("DHW: list(int) >= 3")
    .Attr("MPQ: list(int) >= 3")
    .Attr("dimF: list(int)")
    .Attr("trs: int >=0")
    .Attr("magic_trs: int >= 0")
    .Attr("shift_trs: int >= 0")
    .Attr("fshare: int >= 0")
    .Attr("bshare: int >= 0")
    .Attr("debug: bool = false")
    .SetShapeFn(BlocksparseConvShape)
    .Doc(R"doc(
Blocksparse convolution.
)doc");


template <typename AT, typename BT, typename CT>
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/blocksparse_conv_op.cc [388:419]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    .Input("fprop_grid: int32")
    .Input("bprop_grid: int32")
    .Input("updat_grid: int32")
    .Input("mpq_lut: int32")
    .Input("dhw_lut: int32")
    .Input("ck_lut: int32")
    .Input("a: a_type")
    .Input("b: b_type")
    .Output("c: c_type")
    .Attr("a_type: {half, float, bfloat16}")
    .Attr("b_type: {half, float, bfloat16}")
    .Attr("c_type: {half, float, bfloat16}")
    .Attr("mode: int >=0 =0")
    .Attr("overlapC: bool = false")
    .Attr("overlapK: bool = false")
    .Attr("C: int >=0")
    .Attr("K: int >=0")
    .Attr("DHW: list(int) >= 3")
    .Attr("MPQ: list(int) >= 3")
    .Attr("dimF: list(int)")
    .Attr("trs: int >=0")
    .Attr("magic_trs: int >= 0")
    .Attr("shift_trs: int >= 0")
    .Attr("fshare: int >= 0")
    .Attr("bshare: int >= 0")
    .Attr("debug: bool = false")
    .SetShapeFn(BlocksparseConvShape)
    .Doc(R"doc(
Blocksparse convolution.
)doc");

template <typename AT, typename BT, typename CT>
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -