Function random_int_array

axlearn/common/text_dual_encoder_test.py:110–111 · view source on GitHub ↗

(*, shape: tuple)

Source from the content-addressed store, hash-verified

108
109
110	def random_int_array(*, shape: tuple) -> Tensor:
111	return jax.random.randint(jax.random.PRNGKey(0), shape=shape, minval=1, maxval=VOCAB_SIZE)
112
113
114	class TestTextEmbeddingStreamEncoder(TestCase):

test_positive_inputs_onlyMethod · 0.85

test_positive_inputs_only_with_projectionMethod · 0.85

test_positive_and_negative_inputsMethod · 0.85

test_no_negative_embeddingsMethod · 0.85

test_positive_and_negative_embeddingsMethod · 0.85

test_shape_assertionMethod · 0.85

test_complex_encoder_nameMethod · 0.85

test_simpleMethod · 0.85

no outgoing calls

no test coverage detected