Searched defs:block_table_ (Results 1 – 2 of 2) sorted by relevance
550 std::optional<at::Tensor> &block_table_, // batch_size x max_num_blocks_per_seq in mha_varlen_fwd()1262 std::optional<at::Tensor> &block_table_, // batch_size x max_num_blocks_per_seq in mha_fwd_kvcache()
772 DiskBlockTable *block_table_; // Disk Block Table, shared by all disk variable