aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/toco/model.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-04-09 09:10:31 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-09 09:13:01 -0700
commit90a3db9ff995634314227f0aacf4984d1eee752a (patch)
tree43805c45952558a73d2625cae7b70099d0e30329 /tensorflow/contrib/lite/toco/model.h
parentbbb644e69d38722fd398f18ef8f20b05810d97c6 (diff)
Adding support for the standalone log operator. Basic import/export only, No run time support.
PiperOrigin-RevId: 192135843
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
-rw-r--r--tensorflow/contrib/lite/toco/model.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h
index 9bd72e7de1..56ef9fe2a8 100644
--- a/tensorflow/contrib/lite/toco/model.h
+++ b/tensorflow/contrib/lite/toco/model.h
@@ -56,6 +56,7 @@ enum class OperatorType {
kL2Pool,
kLstmCell,
kLocalResponseNormalization,
+ kLog,
kLogistic,
kMaxPool,
kFakeQuant,
@@ -591,6 +592,17 @@ struct LogisticOperator : Operator {
LogisticOperator() : Operator(OperatorType::kLogistic) {}
};
+// Element-wise natural log operator:
+// x -> ln(x)
+//
+// Inputs:
+// inputs[0]: required: the input array
+//
+// TensorFlow equivalent: Log
+struct LogOperator : Operator {
+ LogOperator() : Operator(OperatorType::kLog) {}
+};
+
// Element-wise Tanh operator:
// x -> Tanh(x) = (exp(x) - exp(-x)) / (exp(x) + exp(-x))
//