Searched refs:test_log_softmax (Results 1 – 12 of 12) sorted by relevance
39 bool test_log_softmax();
827 bool test_log_softmax() { function
152 def test_log_softmax(self): member in _TestJITIRToONNX
401 def test_log_softmax(self): member in TestNNAPI
1270 def test_log_softmax(x, y): function1277 for test in (test_softmax, test_log_softmax, test_softmax_neg_index):
4999 def test_log_softmax(self): member in TestMPS
641 // CHECK-LABEL: test_log_softmax647 func.func @test_log_softmax(%arg0: tensor<13x21x3xf32>) -> tensor<13x21x3xf32> {
991 // CHECK-LABEL: test_log_softmax997 func.func @test_log_softmax(%arg0: tensor<13x21x3xf32>) -> tensor<13x21x3xf32> {
762 def test_log_softmax(self): member in NNTest
1763 def test_log_softmax(self, device): member in TestVmapOfGrad
3728 def test_log_softmax(self, device): member in TestVmapBatchedGradient
2710 def test_log_softmax(self): member in CommonTemplate