Searched refs:SoftmaxTest (Results 1 – 7 of 7) 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
31 auto status = SoftmaxTest(&exec_env_); in TEST_F()
25 absl::Status SoftmaxTest(TestExecutionEnvironment* env);
29 absl::Status SoftmaxTest(TestExecutionEnvironment* env) { in SoftmaxTest() function
30 auto status = SoftmaxTest(&exec_env_);
36 class SoftmaxTest(test.TestCase): class
100 class SoftmaxTest(test_lib.TestCase, parameterized.TestCase): class