Searched refs:cache_table_shape (Results 1 – 2 of 2) sorted by relevance
170 def check_shape(self, cache_table_shape, swap_cache_idx_shape, miss_value_shape): argument171 if len(cache_table_shape) != 2:173 "cache table shape must be 2, but got %d" % len(cache_table_shape))
479 auto cache_table_shape = cache_table_shp->shape(); in InferImplCacheSwapTable() local483 shape.emplace_back(cache_table_shape[1]); in InferImplCacheSwapTable()489 max_shape.emplace_back(cache_table_shape[1]); in InferImplCacheSwapTable()