diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-19 09:49:17 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-19 09:54:05 -0700 |
| commit | 5fc2bdd2d5f624a6bad9e83b992029e3799ab64e (patch) | |
| tree | 576404472bd26801fe2fce6f6385669aef1f42e4 /tensorflow/contrib/lite/model.cc | |
| parent | c532c3f319c72074e6fb8cb10c6d05a3839bcc0a (diff) | |
Implement TFLite sqrt/rsqrt unary operators
PiperOrigin-RevId: 201191877
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
| -rw-r--r-- | tensorflow/contrib/lite/model.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index b9d100b7c9..1f8e796bc7 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -704,10 +704,12 @@ TfLiteStatus ParseOpData(const Operator* op, BuiltinOperator op_type, case BuiltinOperator_RELU: case BuiltinOperator_RELU6: case BuiltinOperator_RELU_N1_TO_1: + case BuiltinOperator_RSQRT: case BuiltinOperator_SELECT: case BuiltinOperator_SIN: case BuiltinOperator_SLICE: case BuiltinOperator_SPACE_TO_BATCH_ND: + case BuiltinOperator_SQRT: case BuiltinOperator_TANH: case BuiltinOperator_TILE: case BuiltinOperator_TOPK_V2: |