Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dreverse_sequence_op_test.py67 def _testBasic(self, dtype, len_dtype=np.int64): member in ReverseSequenceTest
99 self._testBasic(np.float32, np.int32)
102 self._testBasic(np.float32)
105 self._testBasic(np.float64)
108 self._testBasic(np.int32)
111 self._testBasic(np.int64)
114 self._testBasic(np.complex64)
117 self._testBasic(np.complex128)
120 self._testBasic(np.dtype("S1"))
Done_hot_op_test.py54 def _testBasic(self, dtype): member in OneHotTest
104 self._testBasic(np.float32)
108 self._testBasic(np.float64)
112 self._testBasic(np.int32)
116 self._testBasic(np.int64)
120 self._testBasic(np.complex64)
124 self._testBasic(np.complex128)
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dargmax_op_test.py59 def _testBasic(self, dtype): member in ArgMaxTest
95 self._testBasic(np.float32)
117 self._testBasic(np.float64)
122 self._testBasic(np.int16)
127 self._testBasic(np.int32)
132 self._testBasic(np.int64)
137 self._testBasic(np.bool_)
Dconfusion_matrix_test.py54 def _testBasic(self, dtype): member in ConfusionMatrixTest
70 self._testBasic(dtype=np.int32)
74 self._testBasic(dtype=np.int64)
Dcwise_ops_test.py458 def _testBasic(self, fn): member in SelectOpTest
474 self._testBasic(array_ops.where)
475 self._testBasic(array_ops.where_v2)
/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_tensor_dense_matmul_op_test.py90 def _testBasic(self, value_dtype, indices_dtype=np.int64): member in SparseTensorDenseMatMulTest
100 self._testBasic(np.int32)
101 self._testBasic(np.float16)
102 self._testBasic(np.float32)
103 self._testBasic(np.float64)
104 self._testBasic(np.complex64)
105 self._testBasic(np.complex128)
106 self._testBasic(np.int32, indices_dtype=np.int32)
107 self._testBasic(np.float32, indices_dtype=np.int32)
/external/tensorflow/tensorflow/compiler/tests/
Dreverse_sequence_op_test.py57 def _testBasic(self, dtype, len_dtype): member in ReverseSequenceTest
85 self._testBasic(dtype, seq_dtype)