| /external/tensorflow/tensorflow/python/kernel_tests/ |
| reduction_ops_test.py | 139 def _compareAll(self, x, reduction_axes, feed_dict=None):
142 self._compareAll(x, reduction_axes[0])
147 self._compareAll(x, None)
149 self._compareAll(x, axes, feed_dict)
195 self._compareAll(np_arr, None)
295 self._compareAll(np_input, reduction_axes, {c_unknown: np_input})
305 self._compareAll(np_input, reduction_axes, {c_known_rank: np_input})
342 self._compareAll(data, axes)
349 self._compareAll(data, axes)
354 self._compareAll(x, [1]
[all...] |
| scan_ops_test.py | 86 def _compareAll(self, x, axis):
95 self._compareAll(x, axis)
109 self._compareAll(x, axis)
115 self._compareAll(x, axis)
121 self._compareAll(x, axis)
127 self._compareAll(x, axis)
192 def _compareAll(self, x, axis):
201 self._compareAll(x, axis)
215 self._compareAll(x, axis)
221 self._compareAll(x, axis
[all...] |
| constant_op_eager_test.py | 502 def _compareAll(self, dims, val, np_ans):
508 self._compareAll([2, 3], np_ans[0][0], np_ans)
512 self._compareAll([2, 3], np_ans[0][0], np_ans)
516 self._compareAll([2, 3], np_ans[0][0], np_ans)
520 self._compareAll([2, 3], np_ans[0][0], np_ans)
|
| constant_op_test.py | 629 def _compareAll(self, dims, val, np_ans):
635 self._compareAll([2, 3], np_ans[0][0], np_ans)
639 self._compareAll([2, 3], np_ans[0][0], np_ans)
643 self._compareAll([2, 3], np_ans[0][0], np_ans)
647 self._compareAll([2, 3], np_ans[0][0], np_ans)
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| scan_ops_test.py | 88 def _compareAll(self, x, axis):
97 self._compareAll(x, axis)
112 self._compareAll(x, axis)
118 self._compareAll(x, axis)
124 self._compareAll(x, axis)
130 self._compareAll(x, axis)
166 def _compareAll(self, x, axis):
175 self._compareAll(x, axis)
190 self._compareAll(x, axis)
196 self._compareAll(x, axis
[all...] |
