| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | manip_ops_test.py | 60 def _testAll(self, np_input, shift, axis): member in RollTest
68 self._testAll(np.random.randint(-100, 100, (5)).astype(t), 3, 0)
70 self._testAll(
73 self._testAll(
80 self._testAll(np.random.rand(5).astype(t), 2, 0)
82 self._testAll(np.random.rand(3, 4).astype(t), [1, 2], [1, 0])
83 self._testAll(np.random.rand(1, 3, 4).astype(t), [1, 0, -3], [0, 1, 2])
89 self._testAll(x + 1j * x, 2, 0)
92 self._testAll(x + 1j * x, [1, 2], [1, 0])
94 self._testAll(x + 1j * x, [2, 1, 1, 0], [0, 3, 1, 2])
[all …]
|
| D | constant_op_eager_test.py | 60 def _testAll(self, x): member in ConstantTest
65 self._testAll(np.arange(-15, 15).reshape([2, 3, 5]).astype(np.float32))
66 self._testAll(
68 self._testAll(np.empty((2, 0, 5)).astype(np.float32))
93 self._testAll(np.arange(-15, 15).reshape([2, 3, 5]).astype(np.float64))
94 self._testAll(
96 self._testAll(np.empty((2, 0, 5)).astype(np.float64))
113 self._testAll(np.arange(-15, 15).reshape([2, 3, 5]).astype(np.int32))
114 self._testAll(
116 self._testAll(np.empty((2, 0, 5)).astype(np.int32))
[all …]
|
| D | constant_op_test.py | 64 def _testAll(self, x): member in ConstantTest
76 self._testAll(np.arange(-15, 15).reshape([2, 3, 5]).astype(bfloat16))
77 self._testAll(
79 self._testAll(np.empty((2, 0, 5)).astype(bfloat16))
83 self._testAll(np.arange(-15, 15).reshape([2, 3, 5]).astype(np.float16))
84 self._testAll(
86 self._testAll(np.empty((2, 0, 5)).astype(np.float16))
90 self._testAll(np.arange(-15, 15).reshape([2, 3, 5]).astype(np.float32))
91 self._testAll(
93 self._testAll(np.empty((2, 0, 5)).astype(np.float32))
[all …]
|
| D | pad_op_test.py | 108 def _testAll(self, np_inputs, paddings, constant_values): member in PadOpTest
227 self._testAll(
230 self._testAll(
237 self._testAll(np.random.rand(2, 5).astype(t), [[1, 0], [2, 0]], 0.0)
238 self._testAll(np.random.rand(2, 3, 4).astype(t),
240 self._testAll(np.random.rand(0, 3, 4).astype(t),
246 self._testAll(x + 1j * x, [[1, 0], [2, 0]], 1234.0 - 1234.0j)
248 self._testAll(x + 1j * x, [[0, 0], [0, 0], [0, 0], [0, 0]], 0 + 0j)
|
| D | bias_op_test.py | 83 def _testAll(self, np_inputs, np_bias): member in BiasAddTest
117 self._testAll(
124 self._testAll(
130 self._testAll(
137 self._testAll(
242 self._testAll(np.random.randn(*shape), np.random.randn(shape[-1]))
|
| D | softmax_op_test.py | 77 def _testAll(self, features): member in SoftmaxTest
126 self._testAll(
138 self._testAll(data.astype(np.float32))
141 self._testAll(
153 self._testAll(data.astype(np.float16))
|
| D | shape_ops_test.py | 140 def _testAll(self, x): member in ShapeOpsTest
145 self._testAll(np.random.randn(2))
146 self._testAll(np.random.randn(2, 3))
147 self._testAll(np.random.randn(2, 3, 5))
148 self._testAll(np.random.randn(2, 3, 5, 7))
149 self._testAll(np.random.randn(2, 3, 5, 7, 11))
150 self._testAll(np.random.randn(2, 3, 5, 7, 11, 13))
153 self._testAll(np.random.choice((False, True), size=(2,)))
154 self._testAll(np.random.choice((False, True), size=(2, 3)))
155 self._testAll(np.random.choice((False, True), size=(2, 3, 5)))
[all …]
|
| D | cast_op_test.py | 89 def _testAll(self, x): member in CastOpTest
96 self._testAll(np.arange(-10, 10).reshape(2, 10))
97 self._testAll(np.linspace(-10, 10, 17))
103 self._testAll(
120 self._testAll(np.random.normal(0, 10, 210).reshape([2, 3, 5, 7]))
121 self._testAll(np.random.normal(0, 1e6, 210).reshape([2, 3, 5, 7]))
|
| D | xent_op_test.py | 74 def _testAll(self, features, labels): member in XentTest
172 self._testAll(
177 self._testAll(
182 self._testAll(
|