Searched refs:enable_gpu (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 174 enable_gpu: bool,
180 cls).__new__(cls, model_config, enable_gpu, output_names,
245 enable_gpu: bool = True) -> TestResult:
302 enable_gpu=True) -> TestResult: argument
305 if not enable_gpu:
325 enable_gpu=enable_gpu,
363 enable_gpu=True) -> TestResult: argument
366 device = "/device:gpu:0" if enable_gpu else "/device:cpu:0"
380 enable_gpu=enable_gpu,
499 self, inputs, warmup_iterations, benchmark_iterations, enable_gpu=True)
[all …]
|
| D | result_analyzer.py | 113 r = list(result.model_config) + [result.enable_gpu]
|
| /external/skqp/gn/ |
| D | gn_to_bp.py | 262 def generate_args(target_os, enable_gpu): argument
287 if enable_gpu:
|
| /external/skia/gn/ |
| D | gn_to_bp.py | 397 def generate_args(target_os, enable_gpu, renderengine = False): argument
425 if enable_gpu:
|
| /external/skia/tools/skiaserve/tester/ |
| D | tester.py | 74 def enable_gpu(self): member in SkiaServeTester
|
| /external/skqp/tools/skiaserve/tester/ |
| D | tester.py | 74 def enable_gpu(self): member in SkiaServeTester
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | BUILD | 492 name = "enable_gpu",
547 ":enable_gpu": ["//tensorflow/compiler/xla/service:gpu_plugin"],
|