Searched refs:ksize_int (Results 1 – 1 of 1) sorted by relevance
47 std::vector<int32> ksize_int; in PoolingOp() local49 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_int)); in PoolingOp()50 OP_REQUIRES(ctx, ksize_int.size() == num_dims(), in PoolingOp()60 ksize_.push_back(ksize_int[i]); in PoolingOp()