Searched refs:k_caches_read (Results 1 – 2 of 2) sorted by relevance
525 io_info_.k_caches_read.tensor_meta = in set_tensor_meta()549 size_t k_buffer = io_info_.k_caches_read.size / io_info_.k_caches_write.size; in compute_total_nbytes()552 total_nbytes_ += num_heads * io_info_.k_caches_read.size + k_buffer; in compute_total_nbytes()581 cur_pos += io_info_.k_caches_read.size; in init_tensors()584 cur_pos += io_info_.k_caches_read.size / io_info_.k_caches_write.size; in init_tensors()620 i * io_info_.k_caches_read.element_size, in set_all_shifted_ptrs()621 io_info_.k_caches_read); in set_all_shifted_ptrs()
129 InfoAttrs k_caches_read; member138 &k_caches_read,166 DEFINE_IOMEMMGR_VEC_ACCESSOR(k_caches_read);