| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_norm_benchmark.py | 126 def _run_graph(self, device, input_shape, axes, num_layers, mode, scale, member in BatchNormBenchmark
181 t1 = self._run_graph("cpu", shape, axes, 10, "op", True, False, 5)
182 t2 = self._run_graph("cpu", shape, axes, 10, "py", True, False, 5)
183 t3 = self._run_graph("cpu", shape, axes, 10, "slow", True, False, 5)
187 t1 = self._run_graph("gpu", shape, axes, 10, "op", True, False, 50)
188 t2 = self._run_graph("gpu", shape, axes, 10, "py", True, False, 50)
189 t3 = self._run_graph("gpu", shape, axes, 10, "slow", True, False, 50)
193 t1 = self._run_graph("cpu", shape, axes, 10, "op", True, True, 5)
194 t2 = self._run_graph("cpu", shape, axes, 10, "py", True, True, 5)
195 t3 = self._run_graph("cpu", shape, axes, 10, "slow", True, True, 5)
[all …]
|
| D | transpose_benchmark.py | 65 def _run_graph(self, device, input_shape, perm, num_iters, datatype): member in TransposeBenchmark
132 self._run_graph("gpu", ishape, perm, num_iters, datatype)
137 self._run_graph("gpu", ishape, perm, num_iters, datatype)
149 self._run_graph("gpu", ishape, perm, num_iters, datatype)
154 self._run_graph("gpu", ishape, perm, num_iters, datatype)
|
| D | split_benchmark.py | 58 def _run_graph(self, device, output_shape, variable, num_outputs, axis): member in SplitBenchmark
119 self._run_graph("gpu", shape, v, num_outputs, axis)
|
| D | concat_benchmark.py | 81 def _run_graph(self, device, input_shape, variable, num_inputs, axis, grad, member in ConcatBenchmark
143 self._run_graph("cpu", shape, v, num_inputs, axis, True, iters)
|
| D | conv2d_benchmark.py | 97 def _run_graph(self, device, dtype, data_format, input_shape, filter_shape, member in Conv2DBenchmark
206 self._run_graph("gpu", dtype, data_format, ishape, fshape, stride,
|