Searched refs:RankOpModel (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | rank_test.cc | 29 class RankOpModel : public SingleOpModel { class
31 RankOpModel(std::initializer_list<int> input_shape, TensorType input_type) { in RankOpModel() function in tflite::__anon7767a8340111::RankOpModel
53 RankOpModel model({1, 3, 1, 3, 5}, TensorType_FLOAT32); in TEST()
61 RankOpModel model({1, 3, 1, 3, 5}, TensorType_INT32); in TEST()
69 RankOpModel model({}, TensorType_FLOAT32); in TEST()
77 RankOpModel model({1, 0}, TensorType_FLOAT32); in TEST()
|