Home
last modified time | relevance | path

Searched refs:metadata_cache (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/nested/_internal/
Dnested_tensor.py261 metadata_cache = {}
263 metadata_cache["min_seqlen"] = min_seqlen_tensor
265 metadata_cache["max_seqlen"] = max_seqlen_tensor
281 _metadata_cache=metadata_cache,
327 ctx.metadata_cache = x._metadata_cache
337 _metadata_cache=ctx.metadata_cache,
349 metadata_cache: Optional[Dict[str, Any]] = None,
353 if metadata_cache is not None:
354 min_seqlen = metadata_cache.get("min_seqlen", None)
355 max_seqlen = metadata_cache.get("max_seqlen", None)
[all …]
Dops.py1573 metadata_cache = {}
1575 metadata_cache["min_seqlen"] = min_seqlen
1577 metadata_cache["max_seqlen"] = max_seqlen
1584 _metadata_cache=metadata_cache,
/external/minigbm/
Dvirtgpu_cross_domain.c37 struct drv_array *metadata_cache; member
61 if (priv->metadata_cache) in cross_domain_release_private()
62 drv_array_destroy(priv->metadata_cache); in cross_domain_release_private()
158 for (uint32_t i = 0; i < drv_array_size(priv->metadata_cache); i++) { in cross_domain_metadata_query()
159 cached_data = (struct bo_metadata *)drv_array_at_idx(priv->metadata_cache, i); in cross_domain_metadata_query()
212 drv_array_append(priv->metadata_cache, metadata); in cross_domain_metadata_query()
269 priv->metadata_cache = drv_array_init(sizeof(struct bo_metadata)); in cross_domain_init()
270 if (!priv->metadata_cache) { in cross_domain_init()
/external/pytorch/test/
Dtest_nestedtensor.py246 metadata_cache = {"max_seqlen": values.shape[1], "min_seqlen": 1}
248 values.view(-1, values.shape[-1]), offsets, metadata_cache
258 metadata_cache = {"max_seqlen": max_length, "min_seqlen": 1}
259 nt = ViewNestedFromBuffer.apply(values, offsets, metadata_cache)