Searched refs:log_softmax (Results 1 – 2 of 2) 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);
|