Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dcompare_and_bitpack_op_test.py42 def _testBasic(self, dtype): member in CompareAndBitpackTest
57 self._testBasic(np.float32)
60 self._testBasic(np.float64)
63 self._testBasic(np.float16)
66 self._testBasic(np.bool)
69 self._testBasic(np.int8)
72 self._testBasic(np.int16)
75 self._testBasic(np.int32)
78 self._testBasic(np.int64)
Dreverse_sequence_op_test.py65 def _testBasic(self, dtype, len_dtype=np.int64): member in ReverseSequenceTest
91 self._testBasic(np.float32, np.int32)
94 self._testBasic(np.float32)
97 self._testBasic(np.float64)
100 self._testBasic(np.int32)
103 self._testBasic(np.int64)
106 self._testBasic(np.complex64)
109 self._testBasic(np.complex128)
Dargmax_op_test.py59 def _testBasic(self, dtype): member in ArgMaxTest
75 self._testBasic(np.float32)
96 self._testBasic(np.float64)
100 self._testBasic(np.int32)
104 self._testBasic(np.int64)
Done_hot_op_test.py55 def _testBasic(self, dtype): member in OneHotTest
102 self._testBasic(np.float32)
106 self._testBasic(np.float64)
110 self._testBasic(np.int32)
114 self._testBasic(np.int64)
118 self._testBasic(np.complex64)
122 self._testBasic(np.complex128)
Dsparse_tensor_dense_matmul_op_test.py92 def _testBasic(self, value_dtype, indices_dtype=np.int64): member in SparseTensorDenseMatMulTest
103 self._testBasic(np.int32)
104 self._testBasic(np.float32)
105 self._testBasic(np.float64)
106 self._testBasic(np.complex64)
107 self._testBasic(np.complex128)
108 self._testBasic(np.int32, indices_dtype=np.int32)
109 self._testBasic(np.float32, indices_dtype=np.int32)
Dconfusion_matrix_test.py59 def _testBasic(self, dtype): member in ConfusionMatrixTest
75 self._testBasic(dtype=np.int32)
79 self._testBasic(dtype=np.int64)
/external/tensorflow/tensorflow/compiler/tests/
Dreverse_sequence_op_test.py61 def _testBasic(self, dtype, len_dtype): member in ReverseSequenceTest
89 self._testBasic(dtype, seq_dtype)