Home
last modified time | relevance | path

Searched defs:SoftmaxOpModel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsoftmax_test.cc32 class SoftmaxOpModel : public SingleOpModel { class
34 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel() function in tflite::__anonbc7c082a0111::SoftmaxOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc624 class SoftmaxOpModel : public SingleOpModelWithNNAPI { class
626 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel() function in tflite::__anon0c68fcb70111::SoftmaxOpModel