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