Searched defs:include_weights (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_base_test.py | 194 def _create_sparse_data(self, include_weights, weight=0.5): argument
220 def _create_sparse_dataset(self, strategy, include_weights=False, weight=0.5): argument
232 include_weights=False, argument
266 include_weights=False, argument
293 def _create_ragged_dataset(self, strategy, include_weights=False, weight=0.5): argument
305 def _create_dense_dataset(self, strategy, include_weights=False, weight=0.5): argument
328 include_weights=False, argument
|