| /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 | 84 def _run_graph(self, device, input_shape, filter_shape, strides, padding, member in Conv2DBenchmark
159 self._run_graph("gpu", ishape, fshape, stride, padding, dtype, 80,
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_benchmark.py | 134 def _run_graph(self, device, input_shape, filter_shape, strides, padding, member in FusedConv2DBiasActivationBenchmark
231 self._run_graph("gpu", ishape, fshape, stride, padding, 80,
238 self._run_graph("gpu", ishape, fshape, stride, padding, 80,
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | masked_matmul_benchmark.py | 62 def _run_graph(self, a_shape, b_shape, nnz, num_iters, sort=False, member in MaskedmatmulBenchmark
139 self._run_graph(a_shape, b_shape, nnz, num_iters, sort, transpose_a,
|