Searched refs:stride_int (Results 1 – 1 of 1) sorted by relevance
48 std::vector<int32> stride_int; in PoolingOp() local54 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &stride_int)); in PoolingOp()55 OP_REQUIRES(ctx, stride_int.size() == num_dims(), in PoolingOp()61 stride_.push_back(stride_int[i]); in PoolingOp()