Searched refs:lookup_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 39 std::initializer_list<int> lookup_shape, in EmbeddingLookupSparseOpModel() argument
52 BuildInterpreter({lookup_shape, indices_shape, dense_shape_shape, in EmbeddingLookupSparseOpModel()
53 lookup_shape, value_shape}); in EmbeddingLookupSparseOpModel()
|
| D | hashtable_lookup_test.cc | 41 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, in HashtableLookupOpModel() argument
51 BuildInterpreter({lookup_shape, key_shape, value_shape}); in HashtableLookupOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 4811 HashtableLookupOpModel(std::initializer_list<int> lookup_shape, in HashtableLookupOpModel() argument
4821 BuildInterpreterWithNNAPI({lookup_shape, key_shape, value_shape}); in HashtableLookupOpModel()
|