Functions — github.com/UCSC-VLAA/OpenVision

CodeHub MCPcopy Create free account

hub / github.com/UCSC-VLAA/OpenVision / functions

Functions716 in github.com/UCSC-VLAA/OpenVision

⨍ Functions 716◇ Types & classes 90 ↳ Endpoints 1

Functiontranslate_x

Equivalent of PIL Translate in X dimension.

src/transforms/autoaugment.py:262

Functiontranslate_y

Equivalent of PIL Translate in Y dimension.

src/transforms/autoaugment.py:268

Functiontree_apply

Apply a pytree of functions to the pytree.

src/helpers/utils.py:470

Functiontree_compare

Returns `(tree1_only, tree2_only, dtype_shape_mismatch)`.

src/helpers/utils.py:670

Functiontree_map_with_regex

Apply jax-style tree_map based on regex rules. Args: f: a function that is being applied to every variable. tree: jax tree of arrays. r

src/helpers/utils.py:564

Functiontree_replace

Renames/removes (nested) keys. Example usage: tree = {'a': {'b': 2, 'c': 3}, 'c': 4} replacements = { 'a/b': 'a/b/x', # replaces

src/helpers/utils.py:620

Methodtruncate_img_tokens

src/models/text_decoder.py:513

Functionunbatched_gather

src/losses/common.py:34

src/helpers/utils.py:111

Functionupdate_sharding

(tree_leaf1, tree_leaf2)

src/main_clip.py:278

Functionweighted_softmax_xent

Compute weighted cross entropy. Args: logits: [batch, length, num_classes] float array. labels: categorical targets [batch, length] int array

src/losses/common.py:281

Methodwrap_flash_attention

(query, key, value, decoder_segment_ids, head_dim)

src/models/text_transformer.py:348

Methodwrap_flash_attention

src/models/text_decoder_v2.py:192

Methodwrap_flash_attention

(query, key, value, decoder_segment_ids)

src/models/common.py:507

Methodwrap_flash_attention

(query, key, value, decoder_segment_ids)

src/models/text_decoder.py:213

Functionzeros_like_spec

src/datasets/input_pipeline.py:171

← previous701–716 of 716, ranked by callers