Fix adjoint hessian bug

Author: jaeyoo

tensorflow_quantum/core/ops/math_ops/BUILD

@@ -17,6 +17,7 @@ cc_binary(
     srcs = [
         "tfq_inner_product.cc",
         "tfq_inner_product_grad.cc",
+        "tfq_inner_product_hessian.cc",
     ],
     copts = select({
         ":windows": [
@@ -62,6 +63,7 @@ cc_binary(
         # cirq cc proto
         "//tensorflow_quantum/core/ops:parse_context",
         "//tensorflow_quantum/core/ops:tfq_simulate_utils",
+        "//tensorflow_quantum/core/src:adj_hessian_util",
         "//tensorflow_quantum/core/src:adj_util",
         "//tensorflow_quantum/core/src:circuit_parser_qsim",
         "//tensorflow_quantum/core/src:util_qsim",
@@ -100,3 +102,13 @@ py_test(
         "//tensorflow_quantum/python:util",
     ],
 )
+
+py_test(
+    name = "inner_product_hessian_test",
+    srcs = ["inner_product_hessian_test.py"],
+    python_version = "PY3",
+    deps = [
+        ":inner_product_op_py",
+        "//tensorflow_quantum/python:util",
+    ],
+)

tensorflow_quantum/core/ops/math_ops/inner_product_grad_test.py

@@ -27,7 +27,7 @@ class InnerProductAdjGradTest(tf.test.TestCase, parameterized.TestCase):
     """Tests tfq_inner_product_grad."""
 
     def test_inner_product_grad_inputs(self):
-        """Makes sure that inner_product_adj_grad fails on bad inputs."""
+        """Makes sure that inner_product_grad fails on bad inputs."""
         n_qubits = 5
         batch_size = 5
         n_other_programs = 3
@@ -232,7 +232,7 @@ def test_inner_product_grad_inputs(self):
     ])
     def test_correctness_with_symbols(self, n_qubits, batch_size,
                                       inner_dim_size):
-        """Tests that inner_product works with symbols."""
+        """Tests that inner_product_grad works with symbols."""
         symbol_names = ['alpha', 'beta', 'gamma']
         n_params = len(symbol_names)
         qubits = cirq.GridQubit.rect(1, n_qubits)
@@ -242,7 +242,7 @@ def test_correctness_with_symbols(self, n_qubits, batch_size,
 
         other_batch = [
             util.random_circuit_resolver_batch(qubits, inner_dim_size)[0]
-            for i in range(batch_size)
+            for _ in range(batch_size)
         ]
 
         symbol_values_array = np.array(
@@ -312,15 +312,15 @@ def test_correctness_with_symbols(self, n_qubits, batch_size,
     ])
     def test_correctness_without_symbols(self, n_qubits, batch_size,
                                          inner_dim_size):
-        """Tests that inner_product_adj_grad works without symbols."""
+        """Tests that inner_product_grad works without symbols."""
         qubits = cirq.GridQubit.rect(1, n_qubits)
         circuit_batch, _ = \
           util.random_circuit_resolver_batch(
               qubits, batch_size)
 
         other_batch = [
             util.random_circuit_resolver_batch(qubits, inner_dim_size)[0]
-            for i in range(batch_size)
+            for _ in range(batch_size)
         ]
 
         programs = util.convert_to_tensor(circuit_batch)