Lines Matching refs:ksize_
54 ksize_.push_back(ksize_int[i]);
82 std::vector<int64> ksize = ksize_;
123 std::vector<int64> ksize_;
245 ksize_, stride_, num_spatial_dims_,
280 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_));
298 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_));
308 OP_REQUIRES(ctx, ksize_.size() == num_dims(),
348 input, select, ksize_, stride_, xla_padding, out_backprop, init_value,
356 std::vector<int64> ksize_;
390 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_));
391 OP_REQUIRES(ctx, ksize_.size() == num_dims(),
401 OP_REQUIRES(ctx, ksize_[0] == 1 && stride_[0] == 1,
444 filter_dims[i] = ksize_[dim];
472 ctx, out_backprop, dtype, gradients_shape, xla_padding, ksize_,
493 padded_gradients, zero, *ctx->GetOrCreateAdd(dtype), ksize_,
501 std::vector<int64> ksize_;
