def func():
    with backprop.GradientTape() as gt:
        gt.watch(m)
        y = f(m, m, transpose_b=transpose_b)
    _ = gt.gradient(y, m)
def func():
    with backprop.GradientTape() as gt:
        gt.watch(m)
        y = f(m, m, transpose_b=transpose_b)
    _ = gt.gradient(y, m)