Searched refs:HashtableLookupOpModel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | hashtable_lookup_test.cc | 39 class HashtableLookupOpModel : public SingleOpModel { class
41 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, in HashtableLookupOpModel() function in tflite::__anon07e193630111::HashtableLookupOpModel
110 HashtableLookupOpModel m({4}, {3}, {3, 2}, TensorType_FLOAT32); in TEST()
133 HashtableLookupOpModel m({4}, {3}, {3}, TensorType_FLOAT32); in TEST()
156 HashtableLookupOpModel m({4}, {3}, {3}, TensorType_STRING); in TEST()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HashtableLookupTest.cpp | 59 class HashtableLookupOpModel { class
61 HashtableLookupOpModel(std::initializer_list<uint32_t> lookup_shape, in HashtableLookupOpModel() function in android::nn::wrapper::HashtableLookupOpModel
170 HashtableLookupOpModel m({4}, {3}, {3, 2}); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 4886 class HashtableLookupOpModel : public SingleOpModelWithNNAPI { class
4888 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, in HashtableLookupOpModel() function in tflite::__anon1a74836d0111::HashtableLookupOpModel
4955 HashtableLookupOpModel m({4}, {3}, {3, 2}, TensorType_FLOAT32); in TEST()
4978 HashtableLookupOpModel m({4}, {3}, {3}, TensorType_FLOAT32); in TEST()
|