Searched refs:index_shape (Results 1 – 1 of 1) sorted by relevance
56 EmbeddingLookupOpModel(std::initializer_list<uint32_t> index_shape, in EmbeddingLookupOpModel() argument65 OperandType LookupTy(Type::TENSOR_INT32, index_shape); in EmbeddingLookupOpModel()