def benchmark_tf_gradient_forward_identity(self):
    with backprop.GradientTape() as tape:
        m = self._m_2
        tape.watch(m)
        self._run((lambda : gen_array_ops.identity(m)), 30000)
def benchmark_tf_gradient_forward_identity(self):
    with backprop.GradientTape() as tape:
        m = self._m_2
        tape.watch(m)
        self._run((lambda : gen_array_ops.identity(m)), 30000)