Searched refs:cache_shape (Results 1 – 7 of 7) 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() argument
401 if self.cache_shape:402 x.cache_shape = self.cache_shape449 def cache_shape(self): member in Parameter451 return self.param_info.cache_shape453 @cache_shape.setter454 def cache_shape(self, value): member in Parameter457 self.param_info.cache_shape = value
48 auto cache_shape = param_info->cache_shape(); in AddCacheParameters() local50 auto new_tensor = std::make_shared<tensor::Tensor>(type_id, cache_shape); in AddCacheParameters()70 auto cache_shape = host_param_info->cache_shape(); in CheckHostCacheParamSize() local71 if (cache_shape.empty()) { in CheckHostCacheParamSize()74 auto tmp_cache_size = cache_shape[0]; in CheckHostCacheParamSize()184 auto cache_shape = cache_tensor->shape_c(); in BindAndInitCacheTensor() local185 if (host_shape.size() != 2 && host_shape.size() != 2 && host_shape[1] != cache_shape[1]) { in BindAndInitCacheTensor()187 << "host shape:" << host_shape << ", cache shape:" << cache_shape; in BindAndInitCacheTensor()
39 … .def_property("cache_shape", &ParamInfo::cache_shape, &ParamInfo::set_cache_shape) in __anon2c0500310102()
343 auto cache_shape = cache_tensor_ptr->shape_c(); in FlushFromCache() local344 if (host_shape.size() != 2 && host_shape.size() != 2 && host_shape[1] != cache_shape[1]) { in FlushFromCache()346 << "host shape:" << host_shape << ", cache shape:" << cache_shape; in FlushFromCache()
332 self.embedding_table.cache_shape = (self.vocab_cache_size, self.embedding_size)
862 self.embedding_table.cache_shape = (self.vocab_cache_size, self.embedding_size)