Searched refs:cache_tensor_ptr (Results 1 – 2 of 2) sorted by relevance
337 auto cache_tensor_ptr = tensor.cache_tensor_ptr(); in FlushFromCache() local338 if (hashmap_tensor_ptr != nullptr && cache_tensor_ptr != nullptr) { in FlushFromCache()340 cache_tensor_ptr->data_sync(); in FlushFromCache()343 auto cache_shape = cache_tensor_ptr->shape_c(); in FlushFromCache()349 auto cache_data_max_size = static_cast<size_t>(cache_tensor_ptr->Size()); in FlushFromCache()352 …FromCacheToHost<int32_t>(hashmap_tensor_ptr->data_c(), tensor.data_c(), cache_tensor_ptr->data_c(), in FlushFromCache()355 …FromCacheToHost<int32_t>(hashmap_tensor_ptr->data_c(), tensor.data_c(), cache_tensor_ptr->data_c(), in FlushFromCache()
318 std::shared_ptr<Tensor> cache_tensor_ptr() const { return cache_tensor_ptr_; } in cache_tensor_ptr() function319 void set_cache_tensor_ptr(std::shared_ptr<Tensor> cache_tensor_ptr = nullptr) {320 cache_tensor_ptr_ = cache_tensor_ptr;