jax (definition) in projects: aosp_15_r20

Searched defs:jax (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A D	transfer_guard_lib.cc	`32 namespace jax { namespace`
H A D	sharded_device_array.cc	`28 namespace jax { namespace`
H A D	weakref_lru_cache.cc	`26 namespace jax { namespace`
H A D	pmap_lib.cc	`48 namespace jax { namespace`
H A D	jax_jit.cc	`67 namespace jax { namespace`
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/
H A D	tokenizer.json	`17270 "jax": 14858, number`
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/
H A D	tokenizer.json	`17270 "jax": 14858, number`
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/
H A D	tokenizer.json	`17211 "jax": 14858, number`