tf.compat.v1.test.StubOutForTesting