提交 714618af 编写于 作者: V Vijay Vasudevan

TensorFlow: merge changes from internal

Change 110024345
	Removed the unary operator restriction on MaxPool so we can reuse it more flexibly in the future

Base CL: 110043747
上级 7fb82f18
......@@ -67,9 +67,9 @@ struct PoolParameters {
// An implementation of MaxPooling (forward).
template <typename Device, typename T>
class MaxPoolingOp : public UnaryOp<T> {
class MaxPoolingOp : public OpKernel {
public:
explicit MaxPoolingOp(OpKernelConstruction* context) : UnaryOp<T>(context) {
explicit MaxPoolingOp(OpKernelConstruction* context) : OpKernel(context) {
OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
OP_REQUIRES(context, ksize_.size() == 4,
errors::InvalidArgument("Sliding window ksize field must "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册