Searched defs:cache_shape (Results 1 – 4 of 4) sorted by relevance
94 std::vector<int64_t> cache_shape() const { return cache_shape_; } in cache_shape() function95 void set_cache_shape(const std::vector<int64_t> &cache_shape) { cache_shape_ = cache_shape; } in set_cache_shape()
449 def cache_shape(self): member in Parameter454 def cache_shape(self, value): member in Parameter
48 auto cache_shape = param_info->cache_shape(); in AddCacheParameters() local70 auto cache_shape = host_param_info->cache_shape(); in CheckHostCacheParamSize() local184 auto cache_shape = cache_tensor->shape_c(); in BindAndInitCacheTensor() local
343 auto cache_shape = cache_tensor_ptr->shape_c(); in FlushFromCache() local