Searched refs:_test_loop_fn (Results 1 – 4 of 4) sorted by relevance
53 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 20)62 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32, dtypes.int64])71 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32, dtypes.int64])80 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32])92 self._test_loop_fn(102 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 2)113 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 2)122 self._test_loop_fn(loop_fn, 3)131 self._test_loop_fn(loop_fn, 3)151 self._test_loop_fn(loop_fn, 1)[all …]
121 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)137 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=x.dtype)206 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)217 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.bool])228 self._test_loop_fn(loop_fn, 2)253 self._test_loop_fn(loop_fn, 2)278 self._test_loop_fn(loop_fn, 2)296 self._test_loop_fn(loop_fn, 2)312 self._test_loop_fn(loop_fn, 2)328 self._test_loop_fn(loop_fn, 2)[all …]
73 self._test_loop_fn(76 self._test_loop_fn(89 self._test_loop_fn(loop_fn, 8, parallel_iterations=parallel_iterations)90 self._test_loop_fn(loop_fn, 4 * constant_op.constant(2),202 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.int32])226 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)241 self._test_loop_fn(loop_fn, 3)258 self._test_loop_fn(loop_fn, 3)279 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 2)296 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 2)[all …]
52 def _test_loop_fn(self, loop_fn, iters, member in PForTestCase