Searched refs:my_identity (Results 1 – 2 of 2) sorted by relevance
5 // CHECK-LABEL: my_identity6 func @my_identity(%arg0: tensor<2x3xf32>) -> tensor<2x3xf32> {
1324 def my_identity(x): function1331 self.assertAllEqual(backprop.gradients_function(my_identity)(1.0)[0], 2.0)