Searched defs:cache_shape (Results 1 – 9 of 9) sorted by relevance
118 std::vector<int64_t> cache_shape() const { return cache_shape_; } in cache_shape() function119 void set_cache_shape(const std::vector<int64_t> &cache_shape) { cache_shape_ = cache_shape; } in set_cache_shape()
106 static void SetDimsValue(PKVTilingInfo *info, const gert::StorageShape *cache_shape, in SetDimsValue()215 const gert::StorageShape *cache_shape = context->GetInputShape(index0); in TilingFunc() local
56 const gert::StorageShape *cache_shape = context->GetInputShape(index0); in TilingFunc() local
691 def cache_shape(self): member in Parameter707 def cache_shape(self, value): member in Parameter
56 auto cache_shape = param_info->cache_shape(); in AddCacheParameters() local76 auto cache_shape = host_param_info->cache_shape(); in CheckHostCacheParamSize() local190 auto cache_shape = cache_tensor->shape_c(); in BindAndInitCacheTensor() local
473 auto cache_shape = cache_tensor_ptr->shape_c(); in FlushFromCache() local
936 auto cache_shape = p->cache_shape(); in ToStringAttr() local