diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-21 11:54:42 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 12:03:30 -0700 |
| commit | 81d90de884ad6005e57e0d7d333e8476659d00c2 (patch) | |
| tree | 86944a1eb9122dc5758aa9ccb2a3d077c9788e71 /tensorflow/contrib/lite/kernels/register.cc | |
| parent | 9158b1b83a0128fc41bfccd80fe26d8231fe958b (diff) | |
Support reduce_min
PiperOrigin-RevId: 209634537
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 9681b900b7..10d1fcc5bc 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -94,6 +94,7 @@ TfLiteRegistration* Register_NEG(); TfLiteRegistration* Register_SUM(); TfLiteRegistration* Register_REDUCE_PROD(); TfLiteRegistration* Register_REDUCE_MAX(); +TfLiteRegistration* Register_REDUCE_MIN(); TfLiteRegistration* Register_SELECT(); TfLiteRegistration* Register_SLICE(); TfLiteRegistration* Register_SIN(); @@ -219,6 +220,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_SUM, Register_SUM()); AddBuiltin(BuiltinOperator_REDUCE_PROD, Register_REDUCE_PROD()); AddBuiltin(BuiltinOperator_REDUCE_MAX, Register_REDUCE_MAX()); + AddBuiltin(BuiltinOperator_REDUCE_MIN, Register_REDUCE_MIN()); AddBuiltin(BuiltinOperator_EXPAND_DIMS, Register_EXPAND_DIMS()); AddBuiltin(BuiltinOperator_SPARSE_TO_DENSE, Register_SPARSE_TO_DENSE()); AddBuiltin(BuiltinOperator_EQUAL, Register_EQUAL()); |