Searched refs:lookup_size (Results 1 – 4 of 4) sorted by relevance
157 size_t lookup_size = ext_context.graph.vertices_[p.lookup_index].table_size (); in _promote_extensions_if_needed() local159 …t subtables_size = ext_context.graph.find_subgraph_size (p.lookup_index, visited, 1) - lookup_size; in _promote_extensions_if_needed()160 size_t remaining_size = p.size - subtables_size - lookup_size; in _promote_extensions_if_needed()162 l2_l3_size += lookup_size; in _promote_extensions_if_needed()163 l3_l4_size += lookup_size + subtables_size; in _promote_extensions_if_needed()
11 int lookup_size = 1;14 lookup_size *= dim;24 Both `embedding_size` and `lookup_size` are products of values provided by the user. Hence, a malic…
181 size_t lookup_size = 1; in Eval() local186 MultiplyAndCheckOverflow(lookup_size, dim, &lookup_size) == kTfLiteOk, in Eval()199 const size_t output_size = lookup_size * embedding_size; in Eval()
788 int lookup_size) { in HashtableFindOpModel() argument793 lookup_ = AddInput({key_type, {lookup_size}}); in HashtableFindOpModel()796 output_ = AddOutput({value_type, {lookup_size}}); in HashtableFindOpModel()