Searched refs:OneHotOpModel (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot_test.cc | 30 class OneHotOpModel : public SingleOpModel { class
32 OneHotOpModel(std::initializer_list<int> input_shape, int depth_value, in OneHotOpModel() function in tflite::__anon6d43443f0111::OneHotOpModel
67 OneHotOpModel<float> model({3}, depth, TensorType_FLOAT32); in TEST()
78 OneHotOpModel<int> model({3}, depth, TensorType_INT32); in TEST()
88 OneHotOpModel<bool> model({3}, depth, TensorType_BOOL); in TEST()
100 OneHotOpModel<int> model({3}, depth, TensorType_INT32); in TEST()
110 OneHotOpModel<int> model({2}, depth, TensorType_INT32); in TEST()
123 OneHotOpModel<int> model({4}, depth, TensorType_INT32, axis, on, off); in TEST()
137 OneHotOpModel<int> model({4}, depth, TensorType_INT32, axis, on, off); in TEST()
151 OneHotOpModel<float> model({2, 2}, depth, TensorType_FLOAT32, axis, on, off); in TEST()
[all …]
|