Lines Matching refs:ksize
43 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_int));
45 errors::InvalidArgument("Sliding window ksize field must "
82 std::vector<int64> ksize = ksize_;
88 errors::InvalidArgument("ksize must be a vector, not shape ",
91 errors::InvalidArgument("Sliding window ksize field must "
94 ksize.clear();
95 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &ksize));
116 input, InitValue(ctx->builder(), type), *Reduction(ctx, type), ksize,
163 .CompileTimeConstInput("ksize")
180 const std::vector<int64>& ksize, const std::vector<int64>& stride,
186 int64 window_size = std::accumulate(ksize.begin(), ksize.end(), 1,
207 window_ksize[i] = ksize[dim];
280 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_));
292 errors::InvalidArgument("Must supply ksize and stride arguments."));
296 errors::InvalidArgument("ksize must be a vector, not shape ",
309 errors::InvalidArgument("Sliding window ksize field must "
374 .CompileTimeConstInput("ksize")
390 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_));
392 errors::InvalidArgument("Sliding window ksize field must "
