Home
last modified time | relevance | path

Searched refs:_testBasic (Results 1 – 8 of 8) 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.py63 def _testBasic(self, dtype): member in ArgMaxTest
93 self._testBasic(np.float32)
115 self._testBasic(np.float64)
120 self._testBasic(np.int32)
125 self._testBasic(np.int64)
130 self._testBasic(np.bool_)
Dsparse_tensor_dense_matmul_op_test.py92 def _testBasic(self, value_dtype, indices_dtype=np.int64): member in SparseTensorDenseMatMulTest
102 self._testBasic(np.int32)
103 self._testBasic(np.float32)
104 self._testBasic(np.float64)
105 self._testBasic(np.complex64)
106 self._testBasic(np.complex128)
107 self._testBasic(np.int32, indices_dtype=np.int32)
108 self._testBasic(np.float32, indices_dtype=np.int32)
Done_hot_op_test.py58 def _testBasic(self, dtype): member in OneHotTest
108 self._testBasic(np.float32)
112 self._testBasic(np.float64)
116 self._testBasic(np.int32)
120 self._testBasic(np.int64)
124 self._testBasic(np.complex64)
128 self._testBasic(np.complex128)
Dconfusion_matrix_test.py59 def _testBasic(self, dtype): member in ConfusionMatrixTest
75 self._testBasic(dtype=np.int32)
79 self._testBasic(dtype=np.int64)
Dcwise_ops_test.py464 def _testBasic(self, fn): member in SelectOpTest
480 self._testBasic(array_ops.where)
481 self._testBasic(array_ops.where_v2)
/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)