Searched refs:Set3DWeightMatrix (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/ | ||
| H A D | embedding_lookup_sparse_test.cc | 62 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { function in class:tflite::__anon25930::EmbeddingLookupSparseOpModel 90 m.Set3DWeightMatrix( 106 m.Set3DWeightMatrix( 122 m.Set3DWeightMatrix( 142 m.Set3DWeightMatrix( |
| H A D | embedding_lookup_test.cc | 47 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { function in class:tflite::__anon25931::EmbeddingLookupOpModel 74 m.Set3DWeightMatrix( |
Completed in 149 milliseconds