Searched refs:cpu_embedding_lookup (Results 1 – 3 of 3) sorted by relevance
119 results = tpu_embedding_v2.cpu_embedding_lookup(137 tpu_embedding_v2.cpu_embedding_lookup(161 results = tpu_embedding_v2.cpu_embedding_lookup(184 results = tpu_embedding_v2.cpu_embedding_lookup(213 tpu_embedding_v2.cpu_embedding_lookup(235 results = tpu_embedding_v2.cpu_embedding_lookup(261 tpu_embedding_v2.cpu_embedding_lookup(274 tpu_embedding_v2.cpu_embedding_lookup(299 result = tpu_embedding_v2.cpu_embedding_lookup(324 result = tpu_embedding_v2.cpu_embedding_lookup(
1506 def cpu_embedding_lookup(inputs, weights, tables, feature_config): function
279 features = tpu_embedding_v2.cpu_embedding_lookup(