Searched defs:kv_len (Results 1 – 2 of 2) sorted by relevance
| /external/icing/icing/file/ |
| D | persistent-hash-map.cc | 189 int32_t kv_len = key.length() + 1 + info().value_type_size; in Put() local
289 int32_t kv_len = key.length() + 1 + info().value_type_size; in Delete() local
669 int32_t kv_len = key.size() + 1 + info().value_type_size; in Insert() local
|
| /external/pytorch/test/ |
| D | test_transformers.py | 2641 … def _get_mem_eff_drop_mask(batch_size, n_heads, q_len, kv_len, p, seed, offset, device=device): argument
2751 … def _get_mem_eff_drop_mask(batch_size, n_heads, q_len, kv_len, p, seed, offset, device=device): argument
2995 …def _get_mem_eff_drop_mask(batch_size, n_heads, q_len, kv_len, dropout_p, seed, offset, device=dev… argument
3001 …def get_dropout_mask(output, fused_kernel, batch_size, n_heads, q_len, kv_len, dropout_p, device=d… argument
|