lookup_table_op.cc | 395 const auto key_matrix = key.shaped<K, 2>({num_elements, key_size}); local 411 const uint64 key_hash = HashKey(key_matrix, i); 413 IsEqualKey(empty_key_matrix, 0, key_matrix, i)) { 418 IsEqualKey(deleted_key_matrix, 0, key_matrix, i)) { 425 if (IsEqualKey(key_buckets_matrix, bucket_index, key_matrix, i)) { 573 const auto key_matrix = key.shaped<K, 2>({num_elements, key_size}); local 586 const uint64 key_hash = HashKey(key_matrix, i); 588 IsEqualKey(empty_key_tensor, 0, key_matrix, i)) { 596 IsEqualKey(deleted_key_tensor, 0, key_matrix, i)) { 606 if (IsEqualKey(key_buckets_matrix, bucket_index, key_matrix, i)) 643 const auto key_matrix = key.shaped<K, 2>({num_elements, key_size}); local [all...] |