Searched refs:use_xla_jit (Results 1 – 3 of 3) 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 | stateless_random_ops_test.py | 200 def _benchmarkUniform(self, name, dtype, use_xla_jit): argument
211 xla_test.Benchmark(self, builder_fn, use_xla_jit=use_xla_jit, device='cpu')
215 'uniform_f32', dtype=dtypes.float32, use_xla_jit=False)
219 'uniform_f64', dtype=dtypes.float64, use_xla_jit=False)
223 'uniform_f32', dtype=dtypes.float32, use_xla_jit=True)
227 'uniform_f64', dtype=dtypes.float64, use_xla_jit=True)
|
| D | xla_test.py | 268 use_xla_jit, argument
297 compile_ops=use_xla_jit,
312 xla = 'xla_' if use_xla_jit else ''
|