tools/benchmark/configuration.py

   1 """
   2 Copyright (C) 2018-2019 Intel Corporation
   3
   4 Licensed under the Apache License, Version 2.0 (the "License");
   5 you may not use this file except in compliance with the License.
   6 You may obtain a copy of the License at
   7
   8       http://www.apache.org/licenses/LICENSE-2.0
   9
  10 Unless required by applicable law or agreed to in writing, software
  11 distributed under the License is distributed on an "AS IS" BASIS,
  12 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 See the License for the specific language governing permissions and
  14 limitations under the License.
  15 """
  16
  17
  18 class Configuration:
  19     def __init__(
  20         self,
  21         config: str,
  22         model: str,
  23         weights: str,
  24         device: str,
  25         cpu_extension: str,
  26         gpu_extension: str,
  27         benchmark_iterations_count: int
  28     ):
  29
  30         self._config = config
  31         self._model = model
  32         self._weights = weights
  33         self._device = device
  34         self._cpu_extension = cpu_extension
  35         self._gpu_extension = gpu_extension
  36         self._benchmark_iterations_count = benchmark_iterations_count
  37
  38     @property
  39     def config(self) -> str:
  40         return self._config
  41
  42     @property
  43     def model(self) -> str:
  44         return self._model
  45
  46     @property
  47     def weights(self) -> str:
  48         return self._weights
  49
  50     @property
  51     def device(self) -> str:
  52         return self._device
  53
  54     @property
  55     def cpu_extension(self) -> str:
  56         return self._cpu_extension
  57
  58     @property
  59     def gpu_extension(self) -> str:
  60         return self._gpu_extension
  61
  62     @property
  63     def benchmark_iterations_count(self):
  64         return self._benchmark_iterations_count