Searched refs:SoftmaxOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | softmax_test.cc | 32 class SoftmaxOpModel : public SingleOpModel { class
34 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel() function in tflite::__anonbc7c082a0111::SoftmaxOpModel
63 SoftmaxOpModel m(/*batches=*/2, /*size=*/5, /*beta=*/1.0); in TEST()
88 SoftmaxOpModel m(batch_size, input_size, beta); in TEST()
117 SoftmaxOpModel m(batch_size, input_size, beta); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 624 class SoftmaxOpModel : public SingleOpModelWithNNAPI { class
626 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel() function in tflite::__anon0c68fcb70111::SoftmaxOpModel
655 SoftmaxOpModel m(/*batches=*/2, /*size=*/5, /*beta=*/1.0); in TEST()
|