diff options
| author |  Yoni Ben-Meshulam <yonib@google.com> | 2016-11-29 22:24:21 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-29 22:44:22 -0800 |
| commit | 9d6eacc6abfc6a9e41e4fd55fee87bff67d5f990 (patch) | |
| tree | 8ae47cee9e4764248cf576642842f3563932cbef /tensorflow/tools/benchmark | |
| parent | bada4a5339d4567419e993b7736eb23a6f3535c4 (diff) | |
Add support for benchmarking with uint8 input types.
Change: 140568567
Diffstat (limited to 'tensorflow/tools/benchmark')
| -rw-r--r-- | tensorflow/tools/benchmark/benchmark_model.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/tools/benchmark/benchmark_model.cc b/tensorflow/tools/benchmark/benchmark_model.cc index c3d9e865b1..c544829e5f 100644 --- a/tensorflow/tools/benchmark/benchmark_model.cc +++ b/tensorflow/tools/benchmark/benchmark_model.cc @@ -100,6 +100,11 @@ Status RunBenchmark(const std::vector<InputLayerInfo>& inputs, int_tensor = int_tensor.constant(0.0); break; } + case DT_UINT8: { + auto int_tensor = input_tensor.flat<uint8>(); + int_tensor = int_tensor.constant(0.0); + break; + } default: LOG(FATAL) << "Unsupported input type: " << input.data_type; } |