Searched refs:testSoftmaxCrossEntropyWithLogits (Results 1 – 2 of 2) sorted by relevance
75 def testSoftmaxCrossEntropyWithLogits(self): member in TestUpgrade
1637 // CHECK-LABEL: func @testSoftmaxCrossEntropyWithLogits1638 func @testSoftmaxCrossEntropyWithLogits(%arg0: tensor<2x3xf32>, %arg1: tensor<3xf32>) -> (tensor<3x…1645 func @testSoftmaxCrossEntropyWithLogits(%arg0: tensor<2xf32>, %arg1: tensor<3xf32>) -> (tensor<3xf3…1653 func @testSoftmaxCrossEntropyWithLogits(%arg0: tensor<3xf32>, %arg1: tensor<3xf32>) -> (tensor<3xf3…