Searched refs:log_softmax (Results 1 – 3 of 3) sorted by relevance
/frameworks/ml/nn/runtime/test/generated/tests/ |
D | log_softmax.mod.py.cpp | 5 namespace log_softmax { namespace 12 TEST_F(GeneratedTests, log_softmax) { in TEST_F() argument 13 execute(log_softmax::CreateModel, in TEST_F() 14 log_softmax::is_ignored, in TEST_F() 15 log_softmax::get_examples()); in TEST_F() 17 TEST_AVAILABLE_SINCE(V1_2, log_softmax, log_softmax::CreateModel) in TEST_AVAILABLE_SINCE() argument 20 execute(log_softmax::CreateModel_relaxed, in TEST_AVAILABLE_SINCE() 21 log_softmax::is_ignored_relaxed, in TEST_AVAILABLE_SINCE() 22 log_softmax::get_examples_relaxed()); in TEST_AVAILABLE_SINCE() 26 execute(log_softmax::CreateModel_float16, in TEST_F() [all …]
|
/frameworks/ml/nn/common/operations/ |
D | LogSoftmax.cpp | 28 namespace log_softmax { namespace 118 NN_REGISTER_OPERATION(LOG_SOFTMAX, log_softmax::kOperationName, log_softmax::validate, 119 log_softmax::prepare, log_softmax::execute);
|
/frameworks/ml/nn/runtime/test/generated/ |
D | all_generated_V1_2_vts_tests.cpp | 47121 namespace log_softmax { namespace 47128 TEST_F(NeuralnetworksHidlTest, log_softmax) { in TEST_F() argument 47130 log_softmax::createTestModel, in TEST_F() 47131 log_softmax::is_ignored, in TEST_F() 47132 log_softmax::get_examples()); in TEST_F() 47135 TEST_F(ValidationTest, log_softmax) { in TEST_F() argument 47136 const Model model = log_softmax::createTestModel(); in TEST_F() 47137 const std::vector<Request> requests = createRequests(log_softmax::get_examples()); in TEST_F() 47144 log_softmax::createTestModel_relaxed, in TEST_F() 47145 log_softmax::is_ignored_relaxed, in TEST_F() [all …]
|