diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/test_util.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/test_util.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/test_util.cc b/tensorflow/contrib/lite/kernels/test_util.cc index 9156917140..0fdb0a3935 100644 --- a/tensorflow/contrib/lite/kernels/test_util.cc +++ b/tensorflow/contrib/lite/kernels/test_util.cc @@ -74,8 +74,8 @@ void SingleOpModel::SetCustomOp( CustomOptionsFormat_FLEXBUFFERS)); } -void SingleOpModel::BuildInterpreter( - std::vector<std::vector<int>> input_shapes) { +void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, + bool allow_fp32_relax_to_fp16) { auto opcodes = builder_.CreateVector(opcodes_); auto operators = builder_.CreateVector(operators_); auto tensors = builder_.CreateVector(tensors_); @@ -113,6 +113,8 @@ void SingleOpModel::BuildInterpreter( CHECK(interpreter_->ResizeInputTensor(input_idx, shape) == kTfLiteOk); } + interpreter_->SetAllowFp16PrecisionForFp32(allow_fp32_relax_to_fp16); + // Modify delegate with function. if (apply_delegate_fn_) { apply_delegate_fn_(interpreter_.get()); |