diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-10 08:55:24 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-10 08:57:46 -0700 |
commit | d7e4458c3ca839fd8f5a86b4342905ce511a47eb (patch) | |
tree | 648d4e1814f47733fb08692685d4d7bd7c578585 /tensorflow/contrib/lite/kernels/register.cc | |
parent | 7a8fb5b32e797565d5ffa8d6f250a33e2210f423 (diff) |
Added minimum op, better type support in maximum.
PiperOrigin-RevId: 192292693
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index 384e1afaa4..67ba8d0f39 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -77,6 +77,7 @@ TfLiteRegistration* Register_CAST(); TfLiteRegistration* Register_DEQUANTIZE(); TfLiteRegistration* Register_PRELU(); TfLiteRegistration* Register_MAXIMUM(); +TfLiteRegistration* Register_MINIMUM(); TfLiteRegistration* Register_ARG_MAX(); BuiltinOpResolver::BuiltinOpResolver() { @@ -136,6 +137,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_DEQUANTIZE, Register_DEQUANTIZE()); AddBuiltin(BuiltinOperator_PRELU, Register_PRELU()); AddBuiltin(BuiltinOperator_MAXIMUM, Register_MAXIMUM()); + AddBuiltin(BuiltinOperator_MINIMUM, Register_MINIMUM()); AddBuiltin(BuiltinOperator_ARG_MAX, Register_ARG_MAX()); // TODO(andrewharp, ahentz): Move these somewhere more appropriate so that |