Searched defs:logits_dtype (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | sparse_xent_op_deterministic_test.py | 92 def _generateInputs(self, labels_dtype, logits_dtype, seed=123): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_xent_op_d9m_test.py | 86 def _generateInputs(self, labels_dtype, logits_dtype, seed): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/ |
| H A D | stateless_random_ops_test.py | 157 def wrap(op, logits, logits_dtype, output_dtype, seed): argument
|