Searched defs:SoftmaxTest (Results 1 – 5 of 5) sorted by relevance
35 TEST(SoftmaxTest, Softmax) { in TEST() argument57 TEST(SoftmaxTest, DoesNotWorkForHeightAxis) { in TEST() argument77 TEST(SoftmaxTest, DoesNotWorkForWidthAxis) { in TEST() argument97 TEST(SoftmaxTest, Softmax1x1) { in TEST() argument124 TEST(SoftmaxTest, SoftmaxBigNumber) { in TEST() argument159 TEST(SoftmaxTest, Softmax1x1BigNumber) { in TEST() argument
101 void SoftmaxTest(tflite::BuiltinOperator softmaxOperatorCode, in SoftmaxTest() function
29 absl::Status SoftmaxTest(TestExecutionEnvironment* env) { in SoftmaxTest() function
36 class SoftmaxTest(test.TestCase): class
100 class SoftmaxTest(test_lib.TestCase, parameterized.TestCase): class