_create_ragged_dataset (reference) in projects: aosp_15_r20

Searched refs:_create_ragged_dataset (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A D	tpu_embedding_v2_valid_input_test.py	`97 ragged = self._create_ragged_dataset(strategy) 173 dataset = self._create_ragged_dataset(strategy, include_weights=True,`
H A D	tpu_embedding_v1_correctness_test.py	`129 dataset = self._create_ragged_dataset(strategy) 272 dataset = self._create_ragged_dataset(`
H A D	tpu_embedding_v2_correctness_sequence_feature_test.py	`59 dataset = self._create_ragged_dataset(strategy)`
H A D	tpu_embedding_v2_sequence_feature_test.py	`38 dataset = self._create_ragged_dataset(strategy)`
H A D	tpu_embedding_v2_correctness_base_test.py	`42 dataset = self._create_ragged_dataset(strategy)`
H A D	tpu_embedding_v2_invalid_input_test.py	`103 ragged = self._create_ragged_dataset(strategy, include_weights=True)`
H A D	tpu_embedding_base_test.py	`293 def _create_ragged_dataset(self, strategy, include_weights=False, weight=0.5): member in TPUEmbeddingBaseTest`