Searched refs:lookup_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 35 std::initializer_list<int> lookup_shape, in EmbeddingLookupSparseOpModel() argument
48 BuildInterpreter({lookup_shape, indices_shape, dense_shape_shape, in EmbeddingLookupSparseOpModel()
49 lookup_shape, value_shape}); in EmbeddingLookupSparseOpModel()
|
| D | hashtable_lookup_test.cc | 35 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, in HashtableLookupOpModel() argument
45 BuildInterpreter({lookup_shape, key_shape, value_shape}); in HashtableLookupOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3520 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, in HashtableLookupOpModel() argument
3530 BuildInterpreter({lookup_shape, key_shape, value_shape}); in HashtableLookupOpModel()
|