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.cc62 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { in Set3DWeightMatrix() function in tflite::__anonb811566d0111::EmbeddingLookupSparseOpModel
90 m.Set3DWeightMatrix( in TEST()
106 m.Set3DWeightMatrix( in TEST()
122 m.Set3DWeightMatrix( in TEST()
142 m.Set3DWeightMatrix( in TEST()
Dembedding_lookup_test.cc63 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { in Set3DWeightMatrix() function in tflite::__anon62f3d0c00111::EmbeddingLookupOpModel
99 m.Set3DWeightMatrix( in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3487 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { in Set3DWeightMatrix() function in tflite::__anon0c68fcb70111::EmbeddingLookupOpModel
3505 m.Set3DWeightMatrix( in TEST()