Home | History | Annotate | Download | only in kernels

Lines Matching refs:ksize_

54     OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
55 OP_REQUIRES(context, ksize_.size() == 4,
63 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
81 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
84 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
211 std::vector<int32> ksize_;
229 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
230 OP_REQUIRES(context, ksize_.size() == 4,
238 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
258 pool_params.Init(context, ksize_, stride_, padding_, data_format_,
425 std::vector<int32> ksize_;