Searched defs:_run_graph (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ | ||
| H A D | transpose_benchmark.py | 61 def _run_graph(self, device, input_shape, perm, num_iters, datatype): member in TransposeBenchmark |
| H A D | split_benchmark.py | 54 def _run_graph(self, device, output_shape, variable, num_outputs, axis): member in SplitBenchmark |
| H A D | concat_benchmark.py | 77 def _run_graph(self, device, input_shape, variable, num_inputs, axis, grad, member in ConcatBenchmark |
| H A D | conv2d_benchmark.py | 93 def _run_graph(self, device, dtype, data_format, input_shape, filter_shape, member in Conv2DBenchmark |
| H A D | batch_norm_benchmark.py | 122 def _run_graph(self, device, input_shape, axes, num_layers, mode, scale, member in BatchNormBenchmark |