Home
last modified time | relevance | path

Searched refs:BaseEmbeddingLookupOpModel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_test.cc35 class BaseEmbeddingLookupOpModel : public SingleOpModel { class
37 BaseEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in BaseEmbeddingLookupOpModel() function in tflite::__anon62f3d0c00111::BaseEmbeddingLookupOpModel
59 class EmbeddingLookupOpModel : public BaseEmbeddingLookupOpModel {
61 using BaseEmbeddingLookupOpModel::BaseEmbeddingLookupOpModel;
78 class HybridEmbeddingLookupOpModel : public BaseEmbeddingLookupOpModel {
83 : BaseEmbeddingLookupOpModel(index_shape, weight_shape, type) {} in HybridEmbeddingLookupOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3459 class BaseEmbeddingLookupOpModel : public SingleOpModelWithNNAPI { class
3461 BaseEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in BaseEmbeddingLookupOpModel() function in tflite::__anon0c68fcb70111::BaseEmbeddingLookupOpModel
3483 class EmbeddingLookupOpModel : public BaseEmbeddingLookupOpModel {
3485 using BaseEmbeddingLookupOpModel::BaseEmbeddingLookupOpModel;