Searched refs:RNNOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn_test.cc | 174 class RNNOpModel : public SingleOpModel { class
176 RNNOpModel(int batches, int units, int size, in RNNOpModel() function in tflite::__anonaa6237b70111::RNNOpModel
234 class HybridRNNOpModel : public RNNOpModel {
237 : RNNOpModel(batches, units, size, tensor_type, tensor_type) { in HybridRNNOpModel()
259 RNNOpModel rnn(2, 16, 8); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1901 class RNNOpModel : public SingleOpModelWithNNAPI { class
1903 RNNOpModel(int batches, int units, int size, in RNNOpModel() function in tflite::__anon0c68fcb70111::RNNOpModel
1961 RNNOpModel rnn(2, 16, 8); in TEST()
|