Searched refs:use_xla_jit (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | gather_test.py | 161 def _benchmarkGather(self, name, axis, gather_indices, use_xla_jit): argument
173 xla_test.Benchmark(self, BuilderFn, use_xla_jit=use_xla_jit, device='cpu')
175 def _benchmarkSliceGather(self, axis, use_xla_jit): argument
177 self._benchmarkGather('slice_gather', axis, [1], use_xla_jit)
179 def _benchmarkNontrivialGather(self, axis, use_xla_jit): argument
181 use_xla_jit)
184 self._benchmarkSliceGather(axis=0, use_xla_jit=False)
187 self._benchmarkSliceGather(axis=0, use_xla_jit=True)
190 self._benchmarkSliceGather(axis=1, use_xla_jit=False)
193 self._benchmarkSliceGather(axis=1, use_xla_jit=True)
[all …]
|
| D | xla_test.py | 231 use_xla_jit, argument
260 compile_ops=use_xla_jit,
274 xla = 'xla_' if use_xla_jit else ''
|