Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "b2dc8ef4af2f798410a44376de6ff4ecd8e660e2"
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. ...@@ -18,6 +18,8 @@ limitations under the License.
#include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/tensor_shape.h"
using ::tensorflow::int32;
class PoolingOp : public tensorflow::OpKernel { class PoolingOp : public tensorflow::OpKernel {
public: public:
explicit PoolingOp(tensorflow::OpKernelConstruction* context) 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