Skip to content

GitLab

Unverified Commit c97c6208 authored by pyoung2778's avatar pyoung2778 Committed by GitHub
Browse files

Update op handler (#9521)

parent f7b4c6de
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
research/seq_flow_lite/tf_ops/tf_custom_ops.cc
View file @ c97c6208
......@@ -18,6 +18,8 @@ limitations under the License.
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/framework/tensor_shape.h"
using ::tensorflow::int32;
class PoolingOp : public tensorflow::OpKernel {
public:
explicit PoolingOp(tensorflow::OpKernelConstruction* context)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment