Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for biased_dense (0.11 sec)

  1. tensorflow/compiler/mlir/tfr/integration/graph_decompose_test.py

      def testBiasedDense(self):
        biased_dense = def_function.function(gen_composite_ops.my_biased_dense)
        t1 = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])
        t2 = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])
        t3 = constant_op.constant([[-10.0, -10.0], [-10.0, -10.0]])
        sq = biased_dense(t1, t2, t3)
        self.assertAllEqual(sq.numpy().reshape(-1), [-3, 0, 5, 12])
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Sep 28 21:37:05 UTC 2021
    - 3.2K bytes
    - Viewed (0)
Back to top