diff options
Diffstat (limited to 'tensorflow/core/user_ops')
-rw-r--r-- | tensorflow/core/user_ops/fact.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/core/user_ops/fact.cc b/tensorflow/core/user_ops/fact.cc index 3a4fc8115a..2e8b22a49b 100644 --- a/tensorflow/core/user_ops/fact.cc +++ b/tensorflow/core/user_ops/fact.cc @@ -15,10 +15,13 @@ limitations under the License. // An example Op. +#include "tensorflow/core/framework/common_shape_fns.h" #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_kernel.h" -REGISTER_OP("Fact").Output("fact: string"); +REGISTER_OP("Fact") + .Output("fact: string") + .SetShapeFn(tensorflow::shape_inference::UnknownShape); class FactOp : public tensorflow::OpKernel { public: |