[Eager] bilinear_tensor_product yaml

paddle/phi/api/yaml/legacy_api.yaml

@@ -272,6 +272,16 @@
     func : bce_loss
   backward : bce_loss_grad
 
+- api : bilinear_tensor_product
+  args : (Tensor x, Tensor y, Tensor weight, Tensor bias)
+  output : Tensor
+  infer_meta :
+    func : BilinearTensorProductInferMeta
+  kernel :
+    func : bilinear_tensor_product
+  optional : bias
+  backward : bilinear_tensor_product_grad
+
 # bitwise_and
 - api : bitwise_and
   args : (Tensor x, Tensor y)

paddle/phi/api/yaml/legacy_backward.yaml

@@ -239,6 +239,15 @@
     func : bce_loss_grad
   inplace : (out_grad -> input_grad)
 
+- backward_api : bilinear_tensor_product_grad
+  forward : bilinear_tensor_product (Tensor x, Tensor y, Tensor weight, Tensor bias) -> Tensor(out)
+  args : (Tensor x, Tensor y, Tensor weight, Tensor out_grad)
+  output : Tensor(x_grad), Tensor(y_grad), Tensor(weight_grad), Tensor(bias_grad)
+  infer_meta :
+    func : BilinearTensorProductGradInferMeta
+  kernel :
+    func : bilinear_tensor_product_grad
+
 - backward_api : brelu_grad
   forward : brelu (Tensor x, float t_min, float t_max) -> Tensor(out)
   args : (Tensor x, Tensor out_grad, float t_min, float t_max)

python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py

@@ -18,6 +18,7 @@
 import numpy as np
 import paddle.fluid as fluid
 from op_test import OpTest
+import paddle
 
 
 class TestDygraphBilinearTensorProductAPIError(unittest.TestCase):
@@ -41,6 +42,7 @@ class TestBilinearTensorProductOp(OpTest):
 
     def setUp(self):
         self.op_type = "bilinear_tensor_product"
+        self.python_api = paddle.nn.functional.bilinear
         batch_size = 6
         size0 = 5
         size1 = 4
@@ -63,10 +65,10 @@ def setUp(self):
         self.outputs = {'Out': output + bias}
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_eager=True)
 
     def test_check_grad_normal(self):
-        self.check_grad(['X', 'Y', 'Weight', 'Bias'], 'Out')
+        self.check_grad(['X', 'Y', 'Weight', 'Bias'], 'Out', check_eager=True)
 
 
 if __name__ == "__main__":

python/paddle/nn/functional/common.py

@@ -850,7 +850,9 @@ def bilinear(x1, x2, weight, bias=None, name=None):
 
     """
 
-    if in_dynamic_mode():
+    if in_dygraph_mode():
+        return _C_ops.final_state_bilinear_tensor_product(x1, x2, weight, bias)
+    elif _non_static_mode():
         return _C_ops.bilinear_tensor_product(x1, x2, weight, bias)
 
     check_variable_and_dtype(x1, 'x1', ['float32', 'float64'], 'bilinear')