Home
last modified time | relevance | path

Searched refs:Set3DWeightMatrix (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse_test.cc66 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { in Set3DWeightMatrix() function in tflite::__anonc8d704f40111::EmbeddingLookupSparseOpModel
95 m.Set3DWeightMatrix( in TEST()
111 m.Set3DWeightMatrix( in TEST()
127 m.Set3DWeightMatrix( in TEST()
147 m.Set3DWeightMatrix( in TEST()
Dembedding_lookup_test.cc72 void Set3DWeightMatrix(const std::function<T(int, int, int)>& function) { in Set3DWeightMatrix() function in tflite::__anon08d5d9270111::EmbeddingLookupOpModel
109 m.Set3DWeightMatrix<float>( in TEST()
251 m.Set3DWeightMatrix<uint8_t>( in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc4778 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { in Set3DWeightMatrix() function in tflite::__anonde5f765e0111::EmbeddingLookupOpModel
4796 m.Set3DWeightMatrix( in TEST()